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))
