Mercurial > libervia-pubsub
annotate idavoll/test/test_storage.py @ 146:b4490bdc77e5
Change semantics of Node.is_subscribed() to match all subscriptions for an
entity (all resources).
| author | Ralph Meijer <ralphm@ik.nu> |
|---|---|
| date | Fri, 29 Jul 2005 15:33:43 +0000 |
| parents | f393bccec4bc |
| children | 5191ba7c4df8 |
| 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) |
|
146
b4490bdc77e5
Change semantics of Node.is_subscribed() to match all subscriptions for an
Ralph Meijer <ralphm@ik.nu>
parents:
145
diff
changeset
|
214 assertEquals(subscribed[1][1], True) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
215 assertEquals(subscribed[2][1], False) |
|
146
b4490bdc77e5
Change semantics of Node.is_subscribed() to match all subscriptions for an
Ralph Meijer <ralphm@ik.nu>
parents:
145
diff
changeset
|
216 assertEquals(subscribed[3][1], False) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
217 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
218 d = defer.DeferredList([self.node.is_subscribed(SUBSCRIBER), |
|
146
b4490bdc77e5
Change semantics of Node.is_subscribed() to match all subscriptions for an
Ralph Meijer <ralphm@ik.nu>
parents:
145
diff
changeset
|
219 self.node.is_subscribed(SUBSCRIBER.userhostJID()), |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
220 self.node.is_subscribed(SUBSCRIBER_PENDING), |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
221 self.node.is_subscribed(OWNER)]) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
222 d.addCallback(cb) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
223 return d |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
224 |
| 131 | 225 def testStoreItems(self): |
|
138
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
226 def cb1(void): |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
227 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
|
228 |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
229 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
|
230 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
|
231 |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
232 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
|
233 d.addCallback(cb1) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
234 d.addCallback(cb2) |
|
635e433810ef
Actually test whether the item was indeed stored.
Ralph Meijer <ralphm@ik.nu>
parents:
137
diff
changeset
|
235 return d |
| 131 | 236 |
| 237 def testStoreUpdatedItems(self): | |
| 238 def cb1(void): | |
| 239 return self.node.get_items_by_id(['current']) | |
| 240 | |
| 241 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
|
242 assertEqual(result[0], decode(ITEM_UPDATED.toXml())) |
| 131 | 243 |
| 244 d = self.node.store_items([ITEM_UPDATED], PUBLISHER) | |
| 245 d.addCallback(cb1) | |
| 246 d.addCallback(cb2) | |
| 247 return d | |
| 248 | |
| 249 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
|
250 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
|
251 assertEqual(result, ['to-be-deleted']) |
| 131 | 252 return self.node.get_items_by_id(['to-be-deleted']) |
| 253 | |
| 254 def cb2(result): | |
| 255 assertEqual(len(result), 0) | |
| 256 | |
| 257 d = self.node.remove_items(['to-be-deleted']) | |
| 258 d.addCallback(cb1) | |
| 259 d.addCallback(cb2) | |
| 260 return d | |
| 261 | |
| 262 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
|
263 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
|
264 assertEqual(result, []) |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
265 |
|
812300cdbc22
Changed behaviour of retraction of items so that only the actually deleted
Ralph Meijer <ralphm@ik.nu>
parents:
139
diff
changeset
|
266 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
|
267 d.addCallback(cb) |
| 131 | 268 return d |
| 269 | |
| 270 def testGetItems(self): | |
| 271 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
|
272 assertIn(decode(ITEM.toXml()), result) |
| 131 | 273 |
| 274 d = self.node.get_items() | |
| 275 d.addCallback(cb) | |
| 276 return d | |
| 277 | |
| 278 def testLastItem(self): | |
| 279 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
|
280 assertEqual([decode(ITEM.toXml())], result) |
| 131 | 281 |
| 282 d = self.node.get_items(1) | |
| 283 d.addCallback(cb) | |
| 284 return d | |
| 285 | |
| 286 def testGetItemsById(self): | |
| 287 def cb(result): | |
| 288 assertEqual(len(result), 1) | |
| 289 | |
| 290 d = self.node.get_items_by_id(['current']) | |
| 291 d.addCallback(cb) | |
| 292 return d | |
| 293 | |
| 294 def testGetNonExistingItemsById(self): | |
| 295 def cb(result): | |
| 296 assertEqual(len(result), 0) | |
| 297 | |
| 298 d = self.node.get_items_by_id(['non-existing']) | |
| 299 d.addCallback(cb) | |
| 300 return d | |
| 301 | |
| 302 def testPurge(self): | |
| 303 def cb1(node): | |
| 304 d = node.purge() | |
| 305 d.addCallback(lambda _: node) | |
| 306 return d | |
| 307 | |
| 308 def cb2(node): | |
| 309 return node.get_items() | |
| 310 | |
| 311 def cb3(result): | |
| 312 assertEqual([], result) | |
| 313 | |
| 314 d = self.s.get_node('to-be-purged') | |
| 315 d.addCallback(cb1) | |
| 316 d.addCallback(cb2) | |
| 317 d.addCallback(cb3) | |
| 318 return d | |
| 319 | |
|
145
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
320 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
|
321 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
|
322 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
|
323 |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
324 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
|
325 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
|
326 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
|
327 |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
328 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
|
329 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
|
330 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
|
331 return d |
|
f393bccec4bc
Add get_affiliations to Node class in storage facilities in preparation of
Ralph Meijer <ralphm@ik.nu>
parents:
142
diff
changeset
|
332 |
| 114 | 333 class MemoryStorageStorageTestCase(unittest.TestCase, StorageTests): |
| 334 | |
| 335 def setUpClass(self): | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
336 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
|
337 default_config |
| 114 | 338 self.s = Storage() |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
339 self.s._nodes['pre-existing'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
340 LeafNode('pre-existing', OWNER, default_config) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
341 self.s._nodes['to-be-deleted'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
342 LeafNode('to-be-deleted', OWNER, None) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
343 self.s._nodes['to-be-reconfigured'] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
344 LeafNode('to-be-reconfigured', OWNER, default_config) |
| 131 | 345 self.s._nodes['to-be-purged'] = \ |
| 346 LeafNode('to-be-purged', OWNER, None) | |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
347 |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
348 subscriptions = self.s._nodes['pre-existing']._subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
349 subscriptions[SUBSCRIBER.full()] = Subscription('subscribed') |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
350 subscriptions[SUBSCRIBER_TO_BE_DELETED.full()] = \ |
| 114 | 351 Subscription('subscribed') |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
352 subscriptions[SUBSCRIBER_PENDING.full()] = \ |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
353 Subscription('pending') |
| 114 | 354 |
|
139
8f6956b9a688
Follow API change in Twisted Xish, while still being compatible with
Ralph Meijer <ralphm@ik.nu>
parents:
138
diff
changeset
|
355 item = (decode(ITEM_TO_BE_DELETED.toXml()), PUBLISHER) |
| 131 | 356 self.s._nodes['pre-existing']._items['to-be-deleted'] = item |
| 357 self.s._nodes['pre-existing']._itemlist.append(item) | |
| 358 self.s._nodes['to-be-purged']._items['to-be-deleted'] = item | |
| 359 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
|
360 item = (decode(ITEM.toXml()), PUBLISHER) |
| 131 | 361 self.s._nodes['pre-existing']._items['current'] = item |
| 362 self.s._nodes['pre-existing']._itemlist.append(item) | |
| 363 | |
|
117
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
364 return StorageTests.setUpClass(self) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
365 |
| 114 | 366 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
|
367 def _callSuperSetUpClass(self, void): |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
368 return StorageTests.setUpClass(self) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
369 |
| 114 | 370 def setUpClass(self): |
| 371 from idavoll.pgsql_storage import Storage | |
| 372 self.s = Storage('ralphm', 'pubsub_test') | |
| 373 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
|
374 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
|
375 d.addCallback(self._callSuperSetUpClass) |
|
bc872c33e88c
Added test cases for a few of the Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
114
diff
changeset
|
376 return d |
| 114 | 377 |
| 378 def tearDownClass(self): | |
| 131 | 379 #return self.s._dbpool.runInteraction(self.cleandb) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
380 pass |
| 114 | 381 |
| 382 def init(self, cursor): | |
| 383 self.cleandb(cursor) | |
| 384 cursor.execute("""INSERT INTO nodes (node) VALUES ('pre-existing')""") | |
| 385 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
|
386 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-reconfigured')""") |
| 131 | 387 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-purged')""") |
| 114 | 388 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
|
389 OWNER.userhost()) |
| 114 | 390 cursor.execute("""INSERT INTO affiliations |
| 391 (node_id, entity_id, affiliation) | |
| 392 SELECT nodes.id, entities.id, 'owner' | |
| 393 FROM nodes, entities | |
| 394 WHERE node='pre-existing' AND jid=%s""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
395 OWNER.userhost()) |
| 114 | 396 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
|
397 SUBSCRIBER.userhost()) |
| 114 | 398 cursor.execute("""INSERT INTO subscriptions |
| 399 (node_id, entity_id, resource, subscription) | |
| 400 SELECT nodes.id, entities.id, %s, 'subscribed' | |
| 401 FROM nodes, entities | |
| 402 WHERE node='pre-existing' AND jid=%s""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
403 (SUBSCRIBER.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
404 SUBSCRIBER.userhost())) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
405 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
|
406 SUBSCRIBER_TO_BE_DELETED.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
407 cursor.execute("""INSERT INTO subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
408 (node_id, entity_id, resource, subscription) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
409 SELECT nodes.id, entities.id, %s, 'subscribed' |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
410 FROM nodes, entities |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
411 WHERE node='pre-existing' AND jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
412 (SUBSCRIBER_TO_BE_DELETED.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
413 SUBSCRIBER_TO_BE_DELETED.userhost())) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
414 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
|
415 SUBSCRIBER_PENDING.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
416 cursor.execute("""INSERT INTO subscriptions |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
417 (node_id, entity_id, resource, subscription) |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
418 SELECT nodes.id, entities.id, %s, 'pending' |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
419 FROM nodes, entities |
|
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
420 WHERE node='pre-existing' AND jid=%s""", |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
421 (SUBSCRIBER_PENDING.resource, |
|
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
422 SUBSCRIBER_PENDING.userhost())) |
| 131 | 423 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
|
424 PUBLISHER.userhost()) |
| 131 | 425 cursor.execute("""INSERT INTO items |
| 426 (node_id, publisher, item, data, date) | |
| 427 SELECT nodes.id, %s, 'to-be-deleted', %s, | |
| 428 now() - interval '1 day' | |
| 429 FROM nodes | |
| 430 WHERE node='pre-existing'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
431 (PUBLISHER.userhost(), |
| 131 | 432 ITEM_TO_BE_DELETED.toXml())) |
| 433 cursor.execute("""INSERT INTO items (node_id, publisher, item, data) | |
| 434 SELECT nodes.id, %s, 'to-be-deleted', %s | |
| 435 FROM nodes | |
| 436 WHERE node='to-be-purged'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
437 (PUBLISHER.userhost(), |
| 131 | 438 ITEM_TO_BE_DELETED.toXml())) |
| 439 cursor.execute("""INSERT INTO items (node_id, publisher, item, data) | |
| 440 SELECT nodes.id, %s, 'current', %s | |
| 441 FROM nodes | |
| 442 WHERE node='pre-existing'""", | |
|
137
aca904a41a43
Discover client_encoding parameter to pyPgSQL.
Ralph Meijer <ralphm@ik.nu>
parents:
131
diff
changeset
|
443 (PUBLISHER.userhost(), |
| 131 | 444 ITEM.toXml())) |
| 114 | 445 |
| 446 def cleandb(self, cursor): | |
| 447 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
|
448 ('non-existing', 'pre-existing', 'to-be-deleted', |
| 131 | 449 'new 1', 'new 2', 'new 3', 'to-be-reconfigured', |
| 450 'to-be-purged')""") | |
| 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 OWNER.userhost()) |
| 114 | 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.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_TO_BE_DELETED.userhost()) |
|
123
8f99b4f7aea2
Add tests for all Node methods.
Ralph Meijer <ralphm@ik.nu>
parents:
117
diff
changeset
|
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 SUBSCRIBER_PENDING.userhost()) |
| 131 | 459 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
|
460 PUBLISHER.userhost()) |