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)