class TestContainerResourceUsage(unittest.TestCase):
def setUp(self):
self.spawner = PrimeHubSpawner(_mock=True)
self.pods = Fake()
self.spawner.reflectors['primehub_pods'] = self.pods
def test_resource_none_resources_container(self):
pods = [pod([none_resource_container()])]
self.pods.pod_list = pods
self.assertEqual({'cpu': 0, 'gpu': 0, 'memory': 0.0}, self.spawner.get_container_resource_usage(dict(name=_TEST_GROUP)))
def test_resource_normal_container(self):
pods = [pod([resource_container(1, 0, 1024 ** 3)])]
self.pods.pod_list = pods
self.assertEqual({'cpu': 1.0, 'gpu': 0, 'memory': 1.0}, self.spawner.get_container_resource_usage(dict(name=_TEST_GROUP)))
def test_resource_normal_with_kernel_gateway_container(self):
pods = [pod([none_resource_container(), resource_container(1, 1, 1024 ** 3)])]
self.pods.pod_list = pods
self.assertEqual({'cpu': 1.0, 'gpu': 1, 'memory': 1.0}, self.spawner.get_container_resource_usage(dict(name=_TEST_GROUP)))
def test_resource_with_literal(self):
pods = [pod([none_resource_container(), resource_container('1000m', 1, '2G')])]
self.pods.pod_list = pods
self.assertEqual({'cpu': 1.0, 'gpu': 1, 'memory': 2.0}, self.spawner.get_container_resource_usage(dict(name=_TEST_GROUP)))
def mock_spawner():
spawner = PrimeHubSpawner(_mock=True)
def mock_render_html(template_name, local_vars):
template = template_loader.get_template(template_name)
return template.render(local_vars)
spawner.render_html = mock_render_html
return spawner
def mock_spawner():
spawner = PrimeHubSpawner(_mock=True)
def mock_render_html(template_name, local_vars):
template = template_loader.get_template(template_name)
return template.render(local_vars)
def mock_resource_usage(group):
return {'cpu': 0, 'memory': 0, 'gpu': 0}
spawner.render_html = mock_render_html
spawner.get_container_resource_usage = mock_resource_usage
return spawner
def mock_spawner():
return PrimeHubSpawner(_mock=True)
def setUp(self):
self.spawner = PrimeHubSpawner(_mock=True)
self.pods = Fake()
self.spawner.reflectors['primehub_pods'] = self.pods