diff --git a/test/integration/component/test_ss_limits.py b/test/integration/component/test_ss_limits.py index c498302c003..d5ee0633080 100644 --- a/test/integration/component/test_ss_limits.py +++ b/test/integration/component/test_ss_limits.py @@ -126,7 +126,7 @@ class TestSecondaryStorageLimits(cloudstackTestCase): except Exception as e: return [FAIL, e] return [PASS, None] - + @data(ROOT_DOMAIN_ADMIN, CHILD_DOMAIN_ADMIN) @attr(tags = ["advanced"], required_hardware="true") def test_01_register_template(self, value): @@ -143,24 +143,28 @@ class TestSecondaryStorageLimits(cloudstackTestCase): response = self.setupAccount(value) self.assertEqual(response[0], PASS, response[1]) + apiclient = self.testClient.getUserApiClient( + UserName=self.account.name, + DomainName=self.account.domain) + builtin_info = get_builtin_template_info(self.apiclient, self.zone.id) self.services["template_2"]["url"] = builtin_info[0] self.services["template_2"]["hypervisor"] = builtin_info[1] self.services["template_2"]["format"] = builtin_info[2] try: - template = Template.register(self.apiclient, + template = Template.register(apiclient, self.services["template_2"], zoneid=self.zone.id, account=self.account.name, domainid=self.account.domainid, hypervisor=self.hypervisor) - template.download(self.apiclient) + template.download(apiclient) except Exception as e: self.fail("Failed to register template: %s" % e) - templates = Template.list(self.apiclient, + templates = Template.list(apiclient, templatefilter=\ self.services["template_2"]["templatefilter"], id=template.id) @@ -170,19 +174,19 @@ class TestSecondaryStorageLimits(cloudstackTestCase): templateSize = (templates[0].size / (1024**3)) expectedCount = templateSize response = matchResourceCount( - self.apiclient, expectedCount, + apiclient, expectedCount, RESOURCE_SECONDARY_STORAGE, accountid=self.account.id) self.assertEqual(response[0], PASS, response[1]) try: - template.delete(self.apiclient) + template.delete(apiclient) except Exception as e: self.fail("Failed to delete template: %s" % e) expectedCount = 0 response = matchResourceCount( - self.apiclient, expectedCount, + apiclient, expectedCount, RESOURCE_SECONDARY_STORAGE, accountid=self.account.id) self.assertEqual(response[0], PASS, response[1])