Mercurial > libervia-pubsub
annotate idavoll/test/test_storage.py @ 145:f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
support for 'Modifying Entity Affiliations' in the JEP-0060 specification.
| author | Ralph Meijer <ralphm@ik.nu> |
|---|---|
| date | Thu, 14 Jul 2005 20:51:48 +0000 |
| parents | 812300cdbc22 |
| children | b4490bdc77e5 |
| rev | line source |
|---|---|
| 114 | 1 from twisted.trial import unittest |
| 2 from twisted.trial.assertions import * | |
| 3 from twisted.words.protocols.jabber import jid | |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
4 from twisted.internet import defer |
| 131 | 5 from twisted.xish import domish |
| 114 | 6 |
| 131 | 7 from idavoll import storage, pubsub |
| 114 | 8 |
| 9 OWNER = jid.JID('owner@example.com') | |
| 10 SUBSCRIBER = jid.JID('subscriber@example.com/Home') | |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
11 SUBSCRIBER_NEW = jid.JID('new@example.com/Home') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
12 SUBSCRIBER_TO_BE_DELETED = jid.JID('to_be_deleted@example.com/Home') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
13 SUBSCRIBER_PENDING = jid.JID('pending@example.com/Home') |
| 131 | 14 PUBLISHER = jid.JID('publisher@example.com') |
| 15 ITEM = domish.Element((pubsub.NS_PUBSUB, 'item'), pubsub.NS_PUBSUB) | |
| 16 ITEM['id'] = 'current' | |
| 17 ITEM.addElement(('testns', 'test'), content=u'Test \u2083 item') | |
| 18 ITEM_NEW = domish.Element((pubsub.NS_PUBSUB, 'item'), pubsub.NS_PUBSUB) | |
| 19 ITEM_NEW['id'] = 'new' | |
| 20 ITEM_NEW.addElement(('testns', 'test'), content=u'Test \u2083 item') | |
| 21 ITEM_UPDATED = domish.Element((pubsub.NS_PUBSUB, 'item'), pubsub.NS_PUBSUB) | |
| 22 ITEM_UPDATED['id'] = 'current' | |
| 23 ITEM_UPDATED.addElement(('testns', 'test'), content=u'Test \u2084 item') | |
| 24 ITEM_TO_BE_DELETED = domish.Element((pubsub.NS_PUBSUB, 'item'), | |
| 25 pubsub.NS_PUBSUB) | |
| 26 ITEM_TO_BE_DELETED['id'] = 'to-be-deleted' | |
| 27 ITEM_TO_BE_DELETED.addElement(('testns', 'test'), content=u'Test \u2083 item') | |
| 114 | 28 |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
29 def decode(object): |
|
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
30 if isinstance(object, str): |
|
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
31 object = object.decode('utf-8') |
|
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
32 return object |
|
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
33 |
| 114 | 34 class StorageTests: |
| 35 | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
36 def _assignTestNode(self, node): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
37 self.node = node |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
38 |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
39 def setUpClass(self): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
40 d = self.s.get_node('pre-existing') |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
41 d.addCallback(self._assignTestNode) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
42 return d |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
43 |
| 114 | 44 def testGetNode(self): |
| 45 return self.s.get_node('pre-existing') | |
| 46 | |
| 47 def testGetNonExistingNode(self): | |
| 48 d = self.s.get_node('non-existing') | |
| 49 assertFailure(d, storage.NodeNotFound) | |
| 50 return d | |
| 51 | |
| 52 def testGetNodeIDs(self): | |
| 53 def cb(node_ids): | |
| 54 assertIn('pre-existing', node_ids) | |
| 55 assertNotIn('non-existing', node_ids) | |
| 56 | |
| 57 return self.s.get_node_ids().addCallback(cb) | |
| 58 | |
| 59 def testCreateExistingNode(self): | |
| 60 d = self.s.create_node('pre-existing', OWNER) | |
| 61 assertFailure(d, storage.NodeExists) | |
| 62 return d | |
| 63 | |
| 64 def testCreateNode(self): | |
| 65 def cb(void): | |
| 66 d = self.s.get_node('new 1') | |
| 67 return d | |
| 68 | |
| 69 d = self.s.create_node('new 1', OWNER) | |
| 70 d.addCallback(cb) | |
| 71 return d | |
| 72 | |
| 73 def testDeleteNonExistingNode(self): | |
| 74 d = self.s.delete_node('non-existing') | |
| 75 assertFailure(d, storage.NodeNotFound) | |
| 76 return d | |
| 77 | |
| 78 def testDeleteNode(self): | |
| 79 def cb(void): | |
| 80 d = self.s.get_node('to-be-deleted') | |
| 81 assertFailure(d, storage.NodeNotFound) | |
| 82 return d | |
| 83 | |
| 84 d = self.s.delete_node('to-be-deleted') | |
| 85 d.addCallback(cb) | |
| 86 return d | |
| 87 | |
| 88 def testGetAffiliations(self): | |
| 89 def cb(affiliations): | |
| 90 assertIn(('pre-existing', 'owner'), affiliations) | |
| 91 | |
| 92 d = self.s.get_affiliations(OWNER) | |
| 93 d.addCallback(cb) | |
| 94 return d | |
| 95 | |
| 96 def testGetSubscriptions(self): | |
| 97 def cb(subscriptions): | |
| 98 assertIn(('pre-existing', SUBSCRIBER, 'subscribed'), subscriptions) | |
| 99 | |
| 100 d = self.s.get_subscriptions(SUBSCRIBER) | |
| 101 d.addCallback(cb) | |
| 102 return d | |
| 103 | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
104 # Node tests |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
105 |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
106 def testGetType(self): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
107 assertEqual(self.node.get_type(), 'leaf') |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
108 |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
109 def testGetConfiguration(self): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
110 config = self.node.get_configuration() |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
111 assertIn('pubsub#persist_items', config.iterkeys()) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
112 assertIn('pubsub#deliver_payloads', config.iterkeys()) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
113 assertEqual(config['pubsub#persist_items'], True) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
114 assertEqual(config['pubsub#deliver_payloads'], True) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
115 |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
116 def testSetConfiguration(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
117 def get_config(node): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
118 d = node.set_configuration({'pubsub#persist_items': False}) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
119 d.addCallback(lambda _: node) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
120 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
121 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
122 def check_object_config(node): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
123 config = node.get_configuration() |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
124 assertEqual(config['pubsub#persist_items'], False) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
125 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
126 def get_node(void): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
127 return self.s.get_node('to-be-reconfigured') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
128 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
129 def check_storage_config(node): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
130 config = node.get_configuration() |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
131 assertEqual(config['pubsub#persist_items'], False) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
132 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
133 d = self.s.get_node('to-be-reconfigured') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
134 d.addCallback(get_config) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
135 d.addCallback(check_object_config) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
136 d.addCallback(get_node) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
137 d.addCallback(check_storage_config) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
138 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
139 |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
140 def testGetMetaData(self): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
141 meta_data = self.node.get_meta_data() |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
142 for key, value in self.node.get_configuration().iteritems(): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
143 assertIn(key, meta_data.iterkeys()) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
144 assertEqual(value, meta_data[key]) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
145 assertIn('pubsub#node_type', meta_data.iterkeys()) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
146 assertEqual(meta_data['pubsub#node_type'], 'leaf') |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
147 |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
148 def testGetAffiliation(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
149 def cb(affiliation): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
150 assertEqual(affiliation, 'owner') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
151 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
152 d = self.node.get_affiliation(OWNER) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
153 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
154 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
155 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
156 def testGetNonExistingAffiliation(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
157 def cb(affiliation): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
158 assertEqual(affiliation, None) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
159 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
160 d = self.node.get_affiliation(SUBSCRIBER) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
161 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
162 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
163 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
164 def testAddSubscription(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
165 def cb1(void): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
166 return self.node.get_subscription(SUBSCRIBER_NEW) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
167 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
168 def cb2(state): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
169 assertEqual(state, 'pending') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
170 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
171 d = self.node.add_subscription(SUBSCRIBER_NEW, 'pending') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
172 d.addCallback(cb1) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
173 d.addCallback(cb2) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
174 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
175 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
176 def testAddExistingSubscription(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
177 d = self.node.add_subscription(SUBSCRIBER, 'pending') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
178 assertFailure(d, storage.SubscriptionExists) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
179 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
180 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
181 def testGetSubscription(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
182 def cb(subscriptions): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
183 assertEquals(subscriptions[0][1], 'subscribed') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
184 assertEquals(subscriptions[1][1], 'pending') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
185 assertEquals(subscriptions[2][1], None) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
186 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
187 d = defer.DeferredList([self.node.get_subscription(SUBSCRIBER), |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
188 self.node.get_subscription(SUBSCRIBER_PENDING), |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
189 self.node.get_subscription(OWNER)]) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
190 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
191 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
192 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
193 def testRemoveSubscription(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
194 return self.node.remove_subscription(SUBSCRIBER_TO_BE_DELETED) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
195 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
196 def testRemoveNonExistingSubscription(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
197 d = self.node.remove_subscription(OWNER) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
198 assertFailure(d, storage.SubscriptionNotFound) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
199 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
200 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
201 def testGetSubscribers(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
202 def cb(subscribers): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
203 assertIn(SUBSCRIBER, subscribers) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
204 assertNotIn(SUBSCRIBER_PENDING, subscribers) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
205 assertNotIn(OWNER, subscribers) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
206 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
207 d = self.node.get_subscribers() |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
208 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
209 return d |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
210 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
211 def testIsSubscriber(self): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
212 def cb(subscribed): |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
213 assertEquals(subscribed[0][1], True) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
214 assertEquals(subscribed[1][1], False) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
215 assertEquals(subscribed[2][1], False) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
216 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
217 d = defer.DeferredList([self.node.is_subscribed(SUBSCRIBER), |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
218 self.node.is_subscribed(SUBSCRIBER_PENDING), |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
219 self.node.is_subscribed(OWNER)]) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
220 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
221 return d |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
222 |
| 131 | 223 def testStoreItems(self): |
|
138
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
224 def cb1(void): |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
225 return self.node.get_items_by_id(['new']) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
226 |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
227 def cb2(result): |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
228 assertEqual(result[0], decode(ITEM_NEW.toXml())) |
|
138
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
229 |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
230 d = self.node.store_items([ITEM_NEW], PUBLISHER) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
231 d.addCallback(cb1) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
232 d.addCallback(cb2) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
233 return d |
| 131 | 234 |
| 235 def testStoreUpdatedItems(self): | |
| 236 def cb1(void): | |
| 237 return self.node.get_items_by_id(['current']) | |
| 238 | |
| 239 def cb2(result): | |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
240 assertEqual(result[0], decode(ITEM_UPDATED.toXml())) |
| 131 | 241 |
| 242 d = self.node.store_items([ITEM_UPDATED], PUBLISHER) | |
| 243 d.addCallback(cb1) | |
| 244 d.addCallback(cb2) | |
| 245 return d | |
| 246 | |
| 247 def testRemoveItems(self): | |
|
142
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
248 def cb1(result): |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
249 assertEqual(result, ['to-be-deleted']) |
| 131 | 250 return self.node.get_items_by_id(['to-be-deleted']) |
| 251 | |
| 252 def cb2(result): | |
| 253 assertEqual(len(result), 0) | |
| 254 | |
| 255 d = self.node.remove_items(['to-be-deleted']) | |
| 256 d.addCallback(cb1) | |
| 257 d.addCallback(cb2) | |
| 258 return d | |
| 259 | |
| 260 def testRemoveNonExistingItems(self): | |
|
142
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
261 def cb(result): |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
262 assertEqual(result, []) |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
263 |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
264 d = self.node.remove_items(['non-existing']) |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
265 d.addCallback(cb) |
| 131 | 266 return d |
| 267 | |
| 268 def testGetItems(self): | |
| 269 def cb(result): | |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
270 assertIn(decode(ITEM.toXml()), result) |
| 131 | 271 |
| 272 d = self.node.get_items() | |
| 273 d.addCallback(cb) | |
| 274 return d | |
| 275 | |
| 276 def testLastItem(self): | |
| 277 def cb(result): | |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
278 assertEqual([decode(ITEM.toXml())], result) |
| 131 | 279 |
| 280 d = self.node.get_items(1) | |
| 281 d.addCallback(cb) | |
| 282 return d | |
| 283 | |
| 284 def testGetItemsById(self): | |
| 285 def cb(result): | |
| 286 assertEqual(len(result), 1) | |
| 287 | |
| 288 d = self.node.get_items_by_id(['current']) | |
| 289 d.addCallback(cb) | |
| 290 return d | |
| 291 | |
| 292 def testGetNonExistingItemsById(self): | |
| 293 def cb(result): | |
| 294 assertEqual(len(result), 0) | |
| 295 | |
| 296 d = self.node.get_items_by_id(['non-existing']) | |
| 297 d.addCallback(cb) | |
| 298 return d | |
| 299 | |
| 300 def testPurge(self): | |
| 301 def cb1(node): | |
| 302 d = node.purge() | |
| 303 d.addCallback(lambda _: node) | |
| 304 return d | |
| 305 | |
| 306 def cb2(node): | |
| 307 return node.get_items() | |
| 308 | |
| 309 def cb3(result): | |
| 310 assertEqual([], result) | |
| 311 | |
| 312 d = self.s.get_node('to-be-purged') | |
| 313 d.addCallback(cb1) | |
| 314 d.addCallback(cb2) | |
| 315 d.addCallback(cb3) | |
| 316 return d | |
| 317 | |
|
145
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
318 def testGetNodeAffilatiations(self): |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
319 def cb1(node): |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
320 return node.get_affiliations() |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
321 |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
322 def cb2(affiliations): |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
323 affiliations = dict(((a[0].full(), a[1]) for a in affiliations)) |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
324 assertEquals(affiliations[OWNER.full()], 'owner') |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
325 |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
326 d = self.s.get_node('pre-existing') |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
327 d.addCallback(cb1) |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
328 d.addCallback(cb2) |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
329 return d |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
330 |
| 114 | 331 class MemoryStorageStorageTestCase(unittest.TestCase, StorageTests): |
| 332 | |
| 333 def setUpClass(self): | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
334 from idavoll.memory_storage import Storage, LeafNode, Subscription, \ |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
335 default_config |
| 114 | 336 self.s = Storage() |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
337 self.s._nodes['pre-existing'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
338 LeafNode('pre-existing', OWNER, default_config) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
339 self.s._nodes['to-be-deleted'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
340 LeafNode('to-be-deleted', OWNER, None) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
341 self.s._nodes['to-be-reconfigured'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
342 LeafNode('to-be-reconfigured', OWNER, default_config) |
| 131 | 343 self.s._nodes['to-be-purged'] = \ |
| 344 LeafNode('to-be-purged', OWNER, None) | |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
345 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
346 subscriptions = self.s._nodes['pre-existing']._subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
347 subscriptions[SUBSCRIBER.full()] = Subscription('subscribed') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
348 subscriptions[SUBSCRIBER_TO_BE_DELETED.full()] = \ |
| 114 | 349 Subscription('subscribed') |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
350 subscriptions[SUBSCRIBER_PENDING.full()] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
351 Subscription('pending') |
| 114 | 352 |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
353 item = (decode(ITEM_TO_BE_DELETED.toXml()), PUBLISHER) |
| 131 | 354 self.s._nodes['pre-existing']._items['to-be-deleted'] = item |
| 355 self.s._nodes['pre-existing']._itemlist.append(item) | |
| 356 self.s._nodes['to-be-purged']._items['to-be-deleted'] = item | |
| 357 self.s._nodes['to-be-purged']._itemlist.append(item) | |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
358 item = (decode(ITEM.toXml()), PUBLISHER) |
| 131 | 359 self.s._nodes['pre-existing']._items['current'] = item |
| 360 self.s._nodes['pre-existing']._itemlist.append(item) | |
| 361 | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
362 return StorageTests.setUpClass(self) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
363 |
| 114 | 364 class PgsqlStorageStorageTestCase(unittest.TestCase, StorageTests): |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
365 def _callSuperSetUpClass(self, void): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
366 return StorageTests.setUpClass(self) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
367 |
| 114 | 368 def setUpClass(self): |
| 369 from idavoll.pgsql_storage import Storage | |
| 370 self.s = Storage('ralphm', 'pubsub_test') | |
| 371 self.s._dbpool.start() | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
372 d = self.s._dbpool.runInteraction(self.init) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
373 d.addCallback(self._callSuperSetUpClass) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
374 return d |
| 114 | 375 |
| 376 def tearDownClass(self): | |
| 131 | 377 #return self.s._dbpool.runInteraction(self.cleandb) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
378 pass |
| 114 | 379 |
| 380 def init(self, cursor): | |
| 381 self.cleandb(cursor) | |
| 382 cursor.execute("""INSERT INTO nodes (node) VALUES ('pre-existing')""") | |
| 383 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-deleted')""") | |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
384 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-reconfigured')""") |
| 131 | 385 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-purged')""") |
| 114 | 386 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
387 OWNER.userhost()) |
| 114 | 388 cursor.execute("""INSERT INTO affiliations |
| 389 (node_id, entity_id, affiliation) | |
| 390 SELECT nodes.id, entities.id, 'owner' | |
| 391 FROM nodes, entities | |
| 392 WHERE node='pre-existing' AND jid=%s""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
393 OWNER.userhost()) |
| 114 | 394 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
395 SUBSCRIBER.userhost()) |
| 114 | 396 cursor.execute("""INSERT INTO subscriptions |
| 397 (node_id, entity_id, resource, subscription) | |
| 398 SELECT nodes.id, entities.id, %s, 'subscribed' | |
| 399 FROM nodes, entities | |
| 400 WHERE node='pre-existing' AND jid=%s""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
401 (SUBSCRIBER.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
402 SUBSCRIBER.userhost())) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
403 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
404 SUBSCRIBER_TO_BE_DELETED.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
405 cursor.execute("""INSERT INTO subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
406 (node_id, entity_id, resource, subscription) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
407 SELECT nodes.id, entities.id, %s, 'subscribed' |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
408 FROM nodes, entities |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
409 WHERE node='pre-existing' AND jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
410 (SUBSCRIBER_TO_BE_DELETED.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
411 SUBSCRIBER_TO_BE_DELETED.userhost())) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
412 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
413 SUBSCRIBER_PENDING.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
414 cursor.execute("""INSERT INTO subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
415 (node_id, entity_id, resource, subscription) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
416 SELECT nodes.id, entities.id, %s, 'pending' |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
417 FROM nodes, entities |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
418 WHERE node='pre-existing' AND jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
419 (SUBSCRIBER_PENDING.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
420 SUBSCRIBER_PENDING.userhost())) |
| 131 | 421 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
422 PUBLISHER.userhost()) |
| 131 | 423 cursor.execute("""INSERT INTO items |
| 424 (node_id, publisher, item, data, date) | |
| 425 SELECT nodes.id, %s, 'to-be-deleted', %s, | |
| 426 now() - interval '1 day' | |
| 427 FROM nodes | |
| 428 WHERE node='pre-existing'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
429 (PUBLISHER.userhost(), |
| 131 | 430 ITEM_TO_BE_DELETED.toXml())) |
| 431 cursor.execute("""INSERT INTO items (node_id, publisher, item, data) | |
| 432 SELECT nodes.id, %s, 'to-be-deleted', %s | |
| 433 FROM nodes | |
| 434 WHERE node='to-be-purged'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
435 (PUBLISHER.userhost(), |
| 131 | 436 ITEM_TO_BE_DELETED.toXml())) |
| 437 cursor.execute("""INSERT INTO items (node_id, publisher, item, data) | |
| 438 SELECT nodes.id, %s, 'current', %s | |
| 439 FROM nodes | |
| 440 WHERE node='pre-existing'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
441 (PUBLISHER.userhost(), |
| 131 | 442 ITEM.toXml())) |
| 114 | 443 |
| 444 def cleandb(self, cursor): | |
| 445 cursor.execute("""DELETE FROM nodes WHERE node in | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
446 ('non-existing', 'pre-existing', 'to-be-deleted', |
| 131 | 447 'new 1', 'new 2', 'new 3', 'to-be-reconfigured', |
| 448 'to-be-purged')""") | |
| 114 | 449 cursor.execute("""DELETE FROM entities WHERE jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
450 OWNER.userhost()) |
| 114 | 451 cursor.execute("""DELETE FROM entities WHERE jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
452 SUBSCRIBER.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
453 cursor.execute("""DELETE FROM entities WHERE jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
454 SUBSCRIBER_TO_BE_DELETED.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
455 cursor.execute("""DELETE FROM entities WHERE jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
456 SUBSCRIBER_PENDING.userhost()) |
| 131 | 457 cursor.execute("""DELETE FROM entities WHERE jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
458 PUBLISHER.userhost()) |