add test cases

Marcusk19 · Jan 9, 2024 · da2f964 · da2f964
1 parent 0ecb447
commit da2f964
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/test/test_api_usage.py b/test/test_api_usage.py
@@ -37,9 +37,11 @@
     OrganizationCard,
     OrganizationPlan,
     OrganizationRhSku,
+    OrganizationRhSkuBatchRemoval,
     OrganizationRhSkuSubscriptionField,
     UserCard,
     UserPlan,
+    UserSkuList,
 )
 from endpoints.api.build import (
     RepositoryBuildList,
@@ -5143,6 +5145,31 @@ def test_sku_stacking(self):
         json = self.getJsonResponse(OrgPrivateRepositories, params=dict(orgname=SUBSCRIPTION_ORG))
         self.assertEqual(True, json["privateAllowed"])
 
+    def test_batch_sku_remove(self):
+        self.login(SUBSCRIPTION_USER)
+        self.postResponse(
+            resource_name=OrganizationRhSku,
+            params=dict(orgname=SUBSCRIPTION_ORG),
+            data={"subscriptions": [{"subscription_id": 12345678}, {"subscription_id": 11223344}]},
+            expected_code=201,
+        )
+        self.postResponse(
+            resource_name=OrganizationRhSkuBatchRemoval,
+            params=dict(orgname=SUBSCRIPTION_ORG),
+            data={"subscriptions": [{"subscription_id": 12345678}, {"subscription_id": 11223344}]},
+        )
+        json = self.getJsonResponse(
+            resource_name=OrganizationRhSku, params=dict(orgname=SUBSCRIPTION_ORG)
+        )
+        self.assertEqual(len(json), 0)
+
+
+class TestUserSku(ApiTestCase):
+    def test_get_user_skus(self):
+        self.login(SUBSCRIPTION_USER)
+        json = self.getJsonResponse(UserSkuList)
+        self.assertEqual(len(json), 2)
+
 
 if __name__ == "__main__":
     unittest.main()