def test_register_node(self):
site = 'site1'
node1 = {
'name': 'node1',
'site': site,
'url' : 'http://node1'
}
node2 = {
'name': 'node2',
'site': site,
'url' : 'http://node2'
}
dao = SiteDAO()
dao.register_node(node1)
dao.register_node(node2)
self.assertEqual(len(dao.nodes(site).keys()), 2)
node3 = {
'name': 'node3',
'site': site,
'url' : 'http://node3'
}
dao.register_node(node3)
self.assertEqual(len(dao.nodes(site).keys()), 3)
def test_all_site_keys(self):
dao = SiteDAO()
node = {
'name': 'node1',
'site': 'site1',
'url' : 'http://node1'
}
dao.register_node(node)
keys = dao._all_site_keys()
self.assertEqual(keys[0], 'site:site1')
def test_unregister_node(self):
node1 = {
'name': 'node1',
'site': 'site1',
'url' : 'http://node1'
}
dao = SiteDAO()
dao.register_node(node1)
self.assertEqual(len(dao.nodes('site1').keys()), 1)
dao.unregister_node(node1)
self.assertEqual(len(dao.nodes('site1').keys()), 0)
sites = dao.get_sites()
self.assertEqual(len(sites.keys()), 0)