X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=src%2Femuvim%2Ftest%2Ftest_resourcemodel_api.py;h=a11bc8b96c7c9a0f7e0671a5cf86c6b25cc803a4;hb=3055e70eb50125372d5f7b8892b185f8e6e7349b;hp=0836c6831d757adf4c4783c6b8b58d231ab82f19;hpb=07bc6add985ec7c20bc3b0ac83030e5522dae07a;p=osm%2Fvim-emu.git diff --git a/src/emuvim/test/test_resourcemodel_api.py b/src/emuvim/test/test_resourcemodel_api.py index 0836c68..a11bc8b 100644 --- a/src/emuvim/test/test_resourcemodel_api.py +++ b/src/emuvim/test/test_resourcemodel_api.py @@ -1,4 +1,5 @@ import time +import os from emuvim.test.base import SimpleTestTopology from emuvim.dcemulator.resourcemodel import BaseResourceModel, ResourceFlavor from emuvim.dcemulator.resourcemodel.upb.simple import UpbSimpleCloudDcRM @@ -23,9 +24,6 @@ class testResourceModel(SimpleTestTopology): r.addFlavour(f) self.assertTrue("test" in r._flavors) self.assertTrue(r._flavors.get("test").get("testmetric") == 42) - # test if allocate and free runs through - self.assertTrue(len(r.allocate("testc", "tiny")) == 3) # expected: 3tuple - self.assertTrue(r.free("testc")) def testAddRmToDc(self): """ @@ -51,19 +49,44 @@ class testResourceModel(SimpleTestTopology): self.assertTrue(len(self.net.rm_registrar.resource_models) == 1) # check if alloc was called during startCompute - self.assertTrue(len(r.allocated_compute_instances) == 0) + self.assertTrue(len(r._allocated_compute_instances) == 0) self.dc[0].startCompute("tc1") time.sleep(1) - self.assertTrue(len(r.allocated_compute_instances) == 1) + self.assertTrue(len(r._allocated_compute_instances) == 1) # check if free was called during stopCompute self.dc[0].stopCompute("tc1") - self.assertTrue(len(r.allocated_compute_instances) == 0) + self.assertTrue(len(r._allocated_compute_instances) == 0) # check connectivity by using ping self.assertTrue(self.net.ping([self.h[0], self.h[1]]) <= 0.0) # stop Mininet network self.stopNet() +def createDummyContainerObject(name, flavor): + + class DummyContainer(object): + + def __init__(self): + self.cpu_period = -1 + self.cpu_quota = -1 + self.mem_limit = -1 + self.memswap_limit = -1 + + def updateCpuLimit(self, cpu_period, cpu_quota): + self.cpu_period = cpu_period + self.cpu_quota = cpu_quota + + def updateMemoryLimit(self, mem_limit): + self.mem_limit = mem_limit + + d = DummyContainer() + d.name = name + d.flavor_name = flavor + return d + + + + class testUpbSimpleCloudDcRM(SimpleTestTopology): """ Test the UpbSimpleCloudDc resource model. @@ -84,30 +107,34 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): rm = UpbSimpleCloudDcRM(max_cu=MAX_CU, max_mu=MAX_MU) reg.register("test_dc", rm) - res = rm.allocate("c1", "tiny") # calculate allocation - self.assertEqual(res[0], E_CPU / MAX_CU * 1) # validate compute result - self.assertEqual(res[1], float(E_MEM) / MAX_MU * 32) # validate memory result - self.assertTrue(res[2] < 0) # validate disk result + c1 = createDummyContainerObject("c1", flavor="tiny") + rm.allocate(c1) # calculate allocation + self.assertEqual(float(c1.cpu_quota) / c1.cpu_period, E_CPU / MAX_CU * 0.5) # validate compute result + self.assertEqual(float(c1.mem_limit/1024/1024), float(E_MEM) / MAX_MU * 32) # validate memory result + + c2 = createDummyContainerObject("c2", flavor="small") + rm.allocate(c2) # calculate allocation + self.assertEqual(float(c2.cpu_quota) / c2.cpu_period, E_CPU / MAX_CU * 1) # validate compute result + self.assertEqual(float(c2.mem_limit/1024/1024), float(E_MEM) / MAX_MU * 128) # validate memory result + - res = rm.allocate("c2", "small") # calculate allocation - self.assertEqual(res[0], E_CPU / MAX_CU * 4) # validate compute result - self.assertEqual(res[1], float(E_MEM) / MAX_MU * 128) # validate memory result - self.assertTrue(res[2] < 0) # validate disk result + c3 = createDummyContainerObject("c3", flavor="medium") + res = rm.allocate(c3) # calculate allocation + self.assertEqual(float(c3.cpu_quota) / c3.cpu_period, E_CPU / MAX_CU * 4) # validate compute result + self.assertEqual(float(c3.mem_limit/1024/1024), float(E_MEM) / MAX_MU * 256) # validate memory result - res = rm.allocate("c3", "medium") # calculate allocation - self.assertEqual(res[0], E_CPU / MAX_CU * 8) # validate compute result - self.assertEqual(res[1], float(E_MEM) / MAX_MU * 256) # validate memory result - self.assertTrue(res[2] < 0) # validate disk result - res = rm.allocate("c4", "large") # calculate allocation - self.assertEqual(res[0], E_CPU / MAX_CU * 16) # validate compute result - self.assertEqual(res[1], float(E_MEM) / MAX_MU * 512) # validate memory result - self.assertTrue(res[2] < 0) # validate disk result + c4 = createDummyContainerObject("c4", flavor="large") + rm.allocate(c4) # calculate allocation + self.assertEqual(float(c4.cpu_quota) / c4.cpu_period, E_CPU / MAX_CU * 8) # validate compute result + self.assertEqual(float(c4.mem_limit/1024/1024), float(E_MEM) / MAX_MU * 512) # validate memory result + + + c5 = createDummyContainerObject("c5", flavor="xlarge") + rm.allocate(c5) # calculate allocation + self.assertEqual(float(c5.cpu_quota) / c5.cpu_period, E_CPU / MAX_CU * 16) # validate compute result + self.assertEqual(float(c5.mem_limit/1024/1024), float(E_MEM) / MAX_MU * 1024) # validate memory result - res = rm.allocate("c5", "xlarge") # calculate allocation - self.assertEqual(res[0], E_CPU / MAX_CU * 32) # validate compute result - self.assertEqual(res[1], float(E_MEM) / MAX_MU * 1024) # validate memory result - self.assertTrue(res[2] < 0) # validate disk result def testAllocationCpuLimit(self): """ @@ -116,7 +143,7 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): """ # config E_CPU = 1.0 - MAX_CU = 100 + MAX_CU = 40 E_MEM = 512 MAX_MU = 4096 # create dummy resource model environment @@ -127,10 +154,14 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): # test over provisioning exeption exception = False try: - rm.allocate("c6", "xlarge") # calculate allocation - rm.allocate("c7", "xlarge") # calculate allocation - rm.allocate("c8", "xlarge") # calculate allocation - rm.allocate("c9", "xlarge") # calculate allocation + c6 = createDummyContainerObject("c6", flavor="xlarge") + c7 = createDummyContainerObject("c7", flavor="xlarge") + c8 = createDummyContainerObject("c8", flavor="xlarge") + c9 = createDummyContainerObject("c9", flavor="xlarge") + rm.allocate(c6) # calculate allocation + rm.allocate(c7) # calculate allocation + rm.allocate(c8) # calculate allocation + rm.allocate(c9) # calculate allocation except Exception as e: self.assertIn("Not enough compute", e.message) exception = True @@ -154,9 +185,12 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): # test over provisioning exeption exception = False try: - rm.allocate("c6", "xlarge") # calculate allocation - rm.allocate("c7", "xlarge") # calculate allocation - rm.allocate("c8", "xlarge") # calculate allocation + c6 = createDummyContainerObject("c6", flavor="xlarge") + c7 = createDummyContainerObject("c7", flavor="xlarge") + c8 = createDummyContainerObject("c8", flavor="xlarge") + rm.allocate(c6) # calculate allocation + rm.allocate(c7) # calculate allocation + rm.allocate(c8) # calculate allocation except Exception as e: self.assertIn("Not enough memory", e.message) exception = True @@ -174,9 +208,10 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): reg = ResourceModelRegistrar(dc_emulation_max_cpu=1.0, dc_emulation_max_mem=512) rm = UpbSimpleCloudDcRM(max_cu=100, max_mu=100) reg.register("test_dc", rm) - rm.allocate("c1", "tiny") # calculate allocation - self.assertTrue(rm.dc_alloc_cu == 1) - rm.free("c1") + c1 = createDummyContainerObject("c6", flavor="tiny") + rm.allocate(c1) # calculate allocation + self.assertTrue(rm.dc_alloc_cu == 0.5) + rm.free(c1) self.assertTrue(rm.dc_alloc_cu == 0) def testInRealTopo(self): @@ -184,6 +219,10 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): Start a real container and check if limitations are really passed down to Dockernet. :return: """ + # ATTENTION: This test should only be executed if emu runs not inside a Docker container, + # because it manipulates cgroups. + if os.environ.get("SON_EMU_IN_DOCKER") is not None: + return # create network self.createNet(nswitches=0, ndatacenter=1, nhosts=2, ndockers=0) # setup links @@ -203,17 +242,18 @@ class testUpbSimpleCloudDcRM(SimpleTestTopology): self.assertTrue(len(self.net.rm_registrar.resource_models) == 1) # check if alloc was called during startCompute - self.assertTrue(len(r.allocated_compute_instances) == 0) + self.assertTrue(len(r._allocated_compute_instances) == 0) tc1 = self.dc[0].startCompute("tc1", flavor_name="tiny") time.sleep(1) - self.assertTrue(len(r.allocated_compute_instances) == 1) + self.assertTrue(len(r._allocated_compute_instances) == 1) # check if there is a real limitation set for containers cgroup - self.assertEqual(tc1.cpu_period/tc1.cpu_quota, 100) + # deactivated for now, seems not to work in docker-in-docker setup used in CI + self.assertEqual(float(tc1.cpu_quota)/tc1.cpu_period, 0.005) # check if free was called during stopCompute self.dc[0].stopCompute("tc1") - self.assertTrue(len(r.allocated_compute_instances) == 0) + self.assertTrue(len(r._allocated_compute_instances) == 0) # check connectivity by using ping self.assertTrue(self.net.ping([self.h[0], self.h[1]]) <= 0.0) # stop Mininet network