Mercurial > libervia-pubsub

comparison idavoll/test/test_storage.py @ 114:e7cfe05bc1d2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Initial revision.
author Ralph Meijer <ralphm@ik.nu>
date Fri, 08 Apr 2005 16:33:40 +0000
parents
children bc872c33e88c
comparison
equal deleted inserted replaced
113:dfef919aaf1b 114:e7cfe05bc1d2
1 from twisted.trial import unittest
2 from twisted.trial.assertions import *
3 from twisted.words.protocols.jabber import jid
4
5 from idavoll import storage
6
7 OWNER = jid.JID('owner@example.com')
8 SUBSCRIBER = jid.JID('subscriber@example.com/Home')
9
10 class StorageTests:
11
12 def testGetNode(self):
13 return self.s.get_node('pre-existing')
14
15 def testGetNonExistingNode(self):
16 d = self.s.get_node('non-existing')
17 assertFailure(d, storage.NodeNotFound)
18 return d
19
20 def testGetNodeIDs(self):
21 def cb(node_ids):
22 assertIn('pre-existing', node_ids)
23 assertNotIn('non-existing', node_ids)
24
25 return self.s.get_node_ids().addCallback(cb)
26
27 def testCreateExistingNode(self):
28 d = self.s.create_node('pre-existing', OWNER)
29 assertFailure(d, storage.NodeExists)
30 return d
31
32 def testCreateNode(self):
33 def cb(void):
34 d = self.s.get_node('new 1')
35 return d
36
37 d = self.s.create_node('new 1', OWNER)
38 d.addCallback(cb)
39 return d
40
41 def testDeleteNonExistingNode(self):
42 d = self.s.delete_node('non-existing')
43 assertFailure(d, storage.NodeNotFound)
44 return d
45
46 def testDeleteNode(self):
47 def cb(void):
48 d = self.s.get_node('to-be-deleted')
49 assertFailure(d, storage.NodeNotFound)
50 return d
51
52 d = self.s.delete_node('to-be-deleted')
53 d.addCallback(cb)
54 return d
55
56 def testGetAffiliations(self):
57 def cb(affiliations):
58 assertIn(('pre-existing', 'owner'), affiliations)
59
60 d = self.s.get_affiliations(OWNER)
61 d.addCallback(cb)
62 return d
63
64 def testGetSubscriptions(self):
65 def cb(subscriptions):
66 assertIn(('pre-existing', SUBSCRIBER, 'subscribed'), subscriptions)
67
68 d = self.s.get_subscriptions(SUBSCRIBER)
69 d.addCallback(cb)
70 return d
71
72 class MemoryStorageStorageTestCase(unittest.TestCase, StorageTests):
73
74 def setUpClass(self):
75 from idavoll.memory_storage import Storage, LeafNode, Subscription
76 self.s = Storage()
77 self.s._nodes['pre-existing'] = LeafNode('pre-existing', OWNER, None)
78 self.s._nodes['to-be-deleted'] = LeafNode('to-be-deleted', OWNER, None)
79 self.s._nodes['pre-existing']._subscriptions[SUBSCRIBER.full()] = \
80 Subscription('subscribed')
81
82 class PgsqlStorageStorageTestCase(unittest.TestCase, StorageTests):
83 def setUpClass(self):
84 from idavoll.pgsql_storage import Storage
85 self.s = Storage('ralphm', 'pubsub_test')
86 self.s._dbpool.start()
87 return self.s._dbpool.runInteraction(self.init)
88
89 def tearDownClass(self):
90 return self.s._dbpool.runInteraction(self.cleandb)
91
92 def init(self, cursor):
93 self.cleandb(cursor)
94 cursor.execute("""INSERT INTO nodes (node) VALUES ('pre-existing')""")
95 cursor.execute("""INSERT INTO nodes (node) VALUES ('to-be-deleted')""")
96 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""",
97 OWNER.userhost().encode('utf-8'))
98 cursor.execute("""INSERT INTO affiliations
99 (node_id, entity_id, affiliation)
100 SELECT nodes.id, entities.id, 'owner'
101 FROM nodes, entities
102 WHERE node='pre-existing' AND jid=%s""",
103 OWNER.userhost().encode('utf-8'))
104 cursor.execute("""INSERT INTO entities (jid) VALUES (%s)""",
105 SUBSCRIBER.userhost().encode('utf-8'))
106 cursor.execute("""INSERT INTO subscriptions
107 (node_id, entity_id, resource, subscription)
108 SELECT nodes.id, entities.id, %s, 'subscribed'
109 FROM nodes, entities
110 WHERE node='pre-existing' AND jid=%s""",
111 (SUBSCRIBER.resource.encode('utf-8'),
112 SUBSCRIBER.userhost().encode('utf-8')))
113
114 def cleandb(self, cursor):
115 cursor.execute("""DELETE FROM nodes WHERE node in
116 ('pre-existing', 'new 1', 'new 2', 'new 3')""")
117 cursor.execute("""DELETE FROM entities WHERE jid=%s""",
118 OWNER.userhost().encode('utf-8'))
119 cursor.execute("""DELETE FROM entities WHERE jid=%s""",
120 SUBSCRIBER.userhost().encode('utf-8'))