def test_get_machine_count(self): user = factory.make_User() handler = ResourcePoolHandler(user, {}, None) pool = factory.make_ResourcePool() factory.make_Machine(pool=pool) result = handler.get({"id": pool.id}) self.assertEqual(1, result['machine_total_count'])
def test_get_machine_ready_count(self): user = factory.make_User() handler = ResourcePoolHandler(user, {}, None) pool = factory.make_ResourcePool() factory.make_Machine(pool=pool, status=NODE_STATUS.NEW) factory.make_Machine(pool=pool, status=NODE_STATUS.READY) result = handler.get({"id": pool.id}) self.assertEqual(2, result['machine_total_count']) self.assertEqual(1, result['machine_ready_count'])
def test_get(self): user = factory.make_User() handler = ResourcePoolHandler(user, {}, None) pool = factory.make_ResourcePool() result = handler.get({"id": pool.id}) self.assertEqual( {'id': pool.id, 'name': pool.name, 'description': pool.description, 'is_default': False, 'created': dehydrate_datetime(pool.created), 'updated': dehydrate_datetime(pool.updated), 'machine_total_count': 0, 'machine_ready_count': 0, 'permissions': []}, result)
def test_get(self): user = factory.make_User() handler = ResourcePoolHandler(user, {}, None) pool = factory.make_ResourcePool() result = handler.get({"id": pool.id}) self.assertEqual( { "id": pool.id, "name": pool.name, "description": pool.description, "is_default": False, "created": dehydrate_datetime(pool.created), "updated": dehydrate_datetime(pool.updated), "machine_total_count": 0, "machine_ready_count": 0, "permissions": [], }, result, )
def test_get_is_default(self): pool = ResourcePool.objects.get_default_resource_pool() handler = ResourcePoolHandler(factory.make_User(), {}, None) result = handler.get({"id": pool.id}) self.assertTrue(result['is_default'])