def test_get_node(self): user = factory.make_User() handler = NodeResultHandler(user, {}, None) node = factory.make_Node() self.assertEqual(node, handler.get_node({"system_id": node.system_id})) self.assertDictEqual({node.system_id: node}, handler.cache["system_ids"])
def test_get_node_from_cache(self): user = factory.make_User() handler = NodeResultHandler(user, {}) fake_system_id = factory.make_name("system_id") fake_node = factory.make_name("node") handler.cache["system_ids"][fake_system_id] = fake_node self.assertEquals( fake_node, handler.get_node({"system_id": fake_system_id}))