def test_total_usage_for_cpu_gpu_ram_is_aggregated(self): allocation1 = fixtures.SlurmFixture().allocation allocation1.cpu_usage = 100 allocation1.gpu_usage = 200 allocation1.ram_usage = 2000 allocation1.save() allocation2 = fixtures.SlurmFixture().allocation allocation2.cpu_usage = 500 allocation2.gpu_usage = 1000 allocation2.ram_usage = 10000 allocation2.save() expected_points = [ { 'measurement': 'slurm_ram_usage', 'fields': { 'value': 12000, } }, { 'measurement': 'slurm_cpu_usage', 'fields': { 'value': 600, } }, { 'measurement': 'slurm_gpu_usage', 'fields': { 'value': 1200, } }, ] self.assertItemsEqual(slurm.get_usage(), expected_points)
def setUp(self): fixture = slurm_fixtures.SlurmFixture() service_settings = fixture.service.settings offering = marketplace_factories.OfferingFactory( type=PLUGIN_NAME, scope=service_settings) plan = marketplace_factories.PlanFactory(offering=offering) order = marketplace_factories.OrderFactory( project=fixture.project, state=marketplace_models.Order.States.EXECUTING) order_item = marketplace_factories.OrderItemFactory( order=order, offering=offering, limits={ component.type: 10 for component in manager.get_components(PLUGIN_NAME) }) for component in manager.get_components(PLUGIN_NAME): component = marketplace_models.OfferingComponent.objects.create( offering=offering, type=component.type, name=component.name, measured_unit=component.measured_unit, ) marketplace_models.PlanComponent.objects.create( plan=plan, component=component, ) # Create SPL fixture.spl self.fixture = fixture self.order_item = order_item self.offering = offering
def setUp(self): self.fixture = slurm_fixtures.SlurmFixture() self.package = slurm_factories.SlurmPackageFactory( service_settings=self.fixture.service.settings) self.invoice = factories.InvoiceFactory(customer=self.fixture.customer) self.scope = self.fixture.allocation self.item = models.GenericInvoiceItem.objects.filter( scope=self.scope).get()
def get_report(self, raw): self.fixture = fixtures.SlurmFixture() self.subprocess_patcher = mock.patch('subprocess.check_output') self.subprocess_mock = self.subprocess_patcher.start() self.subprocess_mock.return_value = raw backend = self.fixture.settings.get_backend() return backend.get_usage_report(VALID_ALLOCATION)
def setUp(self): self.fixture = slurm_fixtures.SlurmFixture() self.package = slurm_factories.SlurmPackageFactory(service_settings=self.fixture.service.settings) self.invoice = factories.InvoiceFactory(customer=self.fixture.customer) self.scope = self.fixture.allocation self.item = models.InvoiceItem.objects.filter(scope=self.scope).get() self.item.unit = models.InvoiceItem.Units.QUANTITY self.item.quantity = 10 self.item.unit_price = 10 self.item.save()
def setUp(self): self.fixture = slurm_fixtures.SlurmFixture() self.allocation = self.fixture.allocation self.offering = marketplace_factories.OfferingFactory(type=PLUGIN_NAME) self.resource = marketplace_factories.ResourceFactory(scope=self.allocation, offering=self.offering) self.order = marketplace_factories.OrderFactory( project=self.fixture.project, state=marketplace_models.Order.States.EXECUTING, ) self.order_item = marketplace_factories.OrderItemFactory( resource=self.resource, type=marketplace_models.RequestTypeMixin.Types.TERMINATE, )
def setUp(self): self.fixture = slurm_fixtures.SlurmFixture() self.invoice = factories.InvoiceFactory(customer=self.fixture.customer) self.url = structure_factories.CustomerFactory.get_url( self.fixture.customer)
def setUp(self): self.fixture = slurm_fixtures.SlurmFixture()