cms.test_utils.util.context_managers: 111 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/cms/test_utils/util/context_managers.py

Stats: 0 executed, 102 missed, 9 excluded, 77 ignored

1. # -*- coding: utf-8 -*-
2. from django.conf import settings
3. from django.core.signals import request_started
4. from django.db import reset_queries
5. from django.template import context
6. from django.utils.translation import get_language, activate
7. from shutil import rmtree as _rmtree
8. from tempfile import template, mkdtemp, _exists
9. import StringIO
10. import sys
12. class NULL:
13. pass
15. class SettingsOverride(object):
16. """
17. Overrides Django settings within a context and resets them to their inital
18. values on exit.
20. Example:
22. with SettingsOverride(DEBUG=True):
23. # do something
24. """
26. def __init__(self, **overrides):
27. self.overrides = overrides
28. self.special_handlers = {
29. 'TEMPLATE_CONTEXT_PROCESSORS': self.template_context_processors,
30. }
32. def __enter__(self):
33. self.old = {}
34. for key, value in self.overrides.items():
35. self.old[key] = getattr(settings, key, NULL)
36. setattr(settings, key, value)
38. def __exit__(self, type, value, traceback):
39. for key, value in self.old.items():
40. if value is not NULL:
41. setattr(settings, key, value)
42. else:
43. delattr(settings,key) # do not pollute the context!
44. self.special_handlers.get(key, lambda:None)()
46. def template_context_processors(self):
47. context._standard_context_processors = None
50. class StdOverride(object):
51. def __init__(self, std='out', buffer=None):
52. self.std = std
53. self.buffer = buffer or StringIO.StringIO()
55. def __enter__(self):
56. setattr(sys, 'std%s' % self.std, self.buffer)
57. return self.buffer
59. def __exit__(self, type, value, traceback):
60. setattr(sys, 'std%s' % self.std, getattr(sys, '__std%s__' % self.std))
62. class StdoutOverride(StdOverride):
63. """
64. This overrides Python's the standard output and redirects it to a StringIO
65. object, so that on can test the output of the program.
67. example:
68. lines = None
69. with StdoutOverride() as buffer:
70. # print stuff
71. lines = buffer.getvalue()
72. """
73. def __init__(self, buffer=None):
74. super(StdoutOverride, self).__init__('out', buffer)
77. class LanguageOverride(object):
78. def __init__(self, language):
79. self.newlang = language
81. def __enter__(self):
82. self.oldlang = get_language()
83. activate(self.newlang)
85. def __exit__(self, type, value, traceback):
86. activate(self.oldlang)
89. class TemporaryDirectory:
90. """Create and return a temporary directory. This has the same
91. behavior as mkdtemp but can be used as a context manager. For
92. example:
94. with TemporaryDirectory() as tmpdir:
95. ...
97. Upon exiting the context, the directory and everthing contained
98. in it are removed.
99. """
101. def __init__(self, suffix="", prefix=template, dir=None):
102. self.name = mkdtemp(suffix, prefix, dir)
104. def __enter__(self):
105. return self.name
107. def cleanup(self):
108. if _exists(self.name):
109. _rmtree(self.name)
111. def __exit__(self, exc, value, tb):
112. self.cleanup()
115. class UserLoginContext(object):
116. def __init__(self, testcase, user):
117. self.testcase = testcase
118. self.user = user
120. def __enter__(self):
121. loginok = self.testcase.client.login(username=self.user.username,
122. password=self.user.username)
123. self.old_user = getattr(self.testcase, 'user', None)
124. self.testcase.user = self.user
125. self.testcase.assertTrue(loginok)
127. def __exit__(self, exc, value, tb):
128. self.testcase.user = self.old_user
129. if not self.testcase.user:
130. delattr(self.testcase, 'user')
131. self.testcase.client.logout()
134. class ChangeModel(object):
135. """
136. Changes attributes on a model while within the context.
138. These changes *ARE* saved to the database for the context!
139. """
140. def __init__(self, instance, **overrides):
141. self.instance = instance
142. self.overrides = overrides
144. def __enter__(self):
145. self.old = {}
146. for key, value in self.overrides.items():
147. self.old[key] = getattr(self.instance, key, NULL)
148. setattr(self.instance, key, value)
149. self.instance.save()
151. def __exit__(self, exc, value, tb):
152. for key in self.overrides.keys():
153. old_value = self.old[key]
154. if old_value is NULL:
155. delattr(self.instance, key)
156. else:
157. setattr(self.instance, key, old_value)
158. self.instance.save()
160. class _AssertNumQueriesContext(object):
161. def __init__(self, test_case, num, connection):
162. self.test_case = test_case
163. self.num = num
164. self.connection = connection
166. def __enter__(self):
167. self.old_debug = settings.DEBUG
168. settings.DEBUG = True
169. self.starting_queries = len(self.connection.queries)
170. request_started.disconnect(reset_queries)
171. return self
173. def __exit__(self, exc_type, exc_value, traceback):
174. settings.DEBUG = self.old_debug
175. request_started.connect(reset_queries)
176. if exc_type is not None:
177. return
179. final_queries = len(self.connection.queries)
180. executed = final_queries - self.starting_queries
182. queries = '\n'.join([q['sql'] for q in self.connection.queries[self.starting_queries:]])
184. self.test_case.assertEqual(
185. executed, self.num, "%d queries executed, %d expected. Queries executed:\n%s" % (
186. executed, self.num, queries
187. )
188. )