1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
import unittest
class Test_get_root(unittest.TestCase):
def _callFUT(self, app, request=None):
from pyramid.scripting import get_root
return get_root(app, request)
def test_it_norequest(self):
app = DummyApp(registry=dummy_registry)
root, closer = self._callFUT(app)
self.assertEqual(len(app.threadlocal_manager.pushed), 1)
pushed = app.threadlocal_manager.pushed[0]
self.assertEqual(pushed['registry'], dummy_registry)
self.assertEqual(pushed['request'].registry, app.registry)
self.assertEqual(len(app.threadlocal_manager.popped), 0)
closer()
self.assertEqual(len(app.threadlocal_manager.popped), 1)
def test_it_withrequest(self):
app = DummyApp(registry=dummy_registry)
request = DummyRequest({})
root, closer = self._callFUT(app, request)
self.assertEqual(len(app.threadlocal_manager.pushed), 1)
pushed = app.threadlocal_manager.pushed[0]
self.assertEqual(pushed['registry'], dummy_registry)
self.assertEqual(pushed['request'], request)
self.assertEqual(len(app.threadlocal_manager.popped), 0)
closer()
self.assertEqual(len(app.threadlocal_manager.popped), 1)
def test_it_requestfactory_overridden(self):
app = DummyApp(registry=dummy_registry)
root, closer = self._callFUT(app)
self.assertEqual(len(app.threadlocal_manager.pushed), 1)
pushed = app.threadlocal_manager.pushed[0]
self.assertEqual(pushed['request'].environ['path'], '/')
class Test_prepare(unittest.TestCase):
def _callFUT(self, request=None, registry=None):
from pyramid.scripting import prepare
return prepare(request, registry)
def _makeRegistry(self):
return DummyRegistry(DummyFactory)
def setUp(self):
from pyramid.threadlocal import manager
self.manager = manager
self.default = manager.get()
def test_it_norequest(self):
registry = self._makeRegistry()
info = self._callFUT(registry=registry)
root, closer = info['root'], info['closer']
pushed = self.manager.get()
self.assertEqual(pushed['registry'], registry)
self.assertEqual(pushed['request'].registry, registry)
self.assertEqual(root.a, (pushed['request'],))
closer()
self.assertEqual(self.default, self.manager.get())
def test_it_withrequest(self):
request = DummyRequest({})
registry = request.registry = self._makeRegistry()
info = self._callFUT(request=request)
root, closer = info['root'], info['closer']
pushed = self.manager.get()
self.assertEqual(pushed['request'], request)
self.assertEqual(pushed['registry'], registry)
self.assertEqual(pushed['request'].registry, registry)
self.assertEqual(root.a, (request,))
closer()
self.assertEqual(self.default, self.manager.get())
def test_it_with_request_and_registry(self):
request = DummyRequest({})
registry = request.registry = self._makeRegistry()
info = self._callFUT(request=request, registry=registry)
root, closer = info['root'], info['closer']
pushed = self.manager.get()
self.assertEqual(pushed['request'], request)
self.assertEqual(pushed['registry'], registry)
self.assertEqual(pushed['request'].registry, registry)
self.assertEqual(root.a, (request,))
closer()
self.assertEqual(self.default, self.manager.get())
class TestMakeRequest(unittest.TestCase):
def _callFUT(self, path='/', registry=None):
from pyramid.scripting import make_request
return make_request(path, registry)
def test_it_with_registry(self):
request = self._callFUT('/', dummy_registry)
self.assertEqual(request.environ['path'], '/')
self.assertEqual(request.registry, dummy_registry)
def test_it_with_no_registry(self):
from pyramid.config import global_registries
# keep registry local so that global_registries is cleared after
registry = DummyRegistry(DummyFactory)
global_registries.add(registry)
request = self._callFUT('/hello')
self.assertEqual(request.environ['path'], '/hello')
self.assertEqual(request.registry, registry)
global_registries.empty()
class Dummy:
pass
dummy_root = Dummy()
class DummyFactory(object):
@classmethod
def blank(cls, path):
req = DummyRequest({'path': path})
return req
def __init__(self, *a, **kw):
self.a = a
self.kw = kw
class DummyRegistry(object):
def __init__(self, factory=None):
self.factory = factory
def queryUtility(self, iface, default=None):
return self.factory or default
dummy_registry = DummyRegistry(DummyFactory)
class DummyApp:
def __init__(self, registry=None):
self.threadlocal_manager = DummyThreadLocalManager()
if registry:
self.registry = registry
def root_factory(self, environ):
return dummy_root
class DummyThreadLocalManager:
def __init__(self):
self.pushed = []
self.popped = []
def push(self, item):
self.pushed.append(item)
def pop(self):
self.popped.append(True)
class DummyRequest:
def __init__(self, environ):
self.environ = environ
|
