def test_update_uses_handler_queryset(self): # test for lp:1927292 user = factory.make_admin() handler = ControllerHandler(user, {}, None) controller = factory.make_RackController(power_type="manual") data = handler.get({"system_id": controller.system_id}) updated = handler.update(data) self.assertIn("vlans_ha", updated)
def test_last_image_sync_returns_none_for_none(self): owner = factory.make_admin() handler = ControllerHandler(owner, {}, None) node = factory.make_RackController(owner=owner, last_image_sync=None) result = handler.list({}) self.assertEqual(1, len(result)) self.assertEqual(NODE_TYPE.RACK_CONTROLLER, result[0].get('node_type')) self.assertIsNone(result[0].get("last_image_sync")) data = handler.get({"system_id": node.system_id}) self.assertIsNone(data.get("last_image_sync"))
def test_last_image_sync(self): owner = factory.make_admin() handler = ControllerHandler(owner, {}, None) node = factory.make_RackController(owner=owner) result = handler.list({}) self.assertEqual(1, len(result)) self.assertEqual(NODE_TYPE.RACK_CONTROLLER, result[0].get('node_type')) self.assertEqual(result[0].get('last_image_sync'), dehydrate_datetime(node.last_image_sync)) data = handler.get({"system_id": node.system_id}) self.assertEqual(data.get("last_image_sync"), dehydrate_datetime(node.last_image_sync))