is_over_quota test cases.

galaxyproject · Sep 11, 2020 · 30cbc41 · 30cbc41
1 parent 4975bb3
commit 30cbc41
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/unit/data/test_quota.py b/test/unit/data/test_quota.py
@@ -60,3 +60,15 @@ def _add_user_quota(self, user, quota):
 
     def _assert_user_quota_is(self, user, amount):
         assert amount == self.quota_agent.get_quota(user)
+        if amount is None:
+            user.total_disk_usage = 1000
+            job = self.model.Job()
+            job.user = user
+            assert not self.quota_agent.is_over_quota(job, None)
+        else:
+            job = self.model.Job()
+            job.user = user
+            user.total_disk_usage = amount - 1
+            assert not self.quota_agent.is_over_quota(job, None)
+            user.total_disk_usage = amount + 1
+            assert self.quota_agent.is_over_quota(job, None)