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_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)
def test_register_node_two(self):
# Make sure we don't return an unknown site
dao = SiteDAO()
nodes = dao.nodes('unknown site')
self.assertEqual(len(nodes), 0)
