QuotaChecker { public Quota getDomainQuota(ObjectStoreConnection con, String domainName) { Quota quota = con.getQuota(domainName); return (quota == null) ? defaultQuota : quota; } QuotaChecker(); Quota getDomainQuota(ObjectStoreConnection con, String domainName);  }
@Test public void testGetDomainQuota() { QuotaChecker quotaCheck = new QuotaChecker(); Quota mockQuota = new Quota().setName("athenz") .setAssertion(10).setEntity(11) .setPolicy(12).setPublicKey(13) .setRole(14).setRoleMember(15) .setService(16).setServiceHost(17) .setSubdomain(18); ObjectStoreConnection con = Mockito.mock(ObjectStoreConnection.class); Mockito.when(con.getQuota("athenz")).thenReturn(mockQuota); Quota quota = quotaCheck.getDomainQuota(con, "athenz"); assertNotNull(quota); assertEquals(quota.getAssertion(), 10); assertEquals(quota.getRole(), 14); assertEquals(quota.getPolicy(), 12); }
@Test public void testGetDomainQuotaDefault() { QuotaChecker quotaCheck = new QuotaChecker(); ObjectStoreConnection con = Mockito.mock(ObjectStoreConnection.class); Mockito.when(con.getQuota("athenz")).thenReturn(null); Quota quota = quotaCheck.getDomainQuota(con, "athenz"); assertNotNull(quota); assertEquals(quota.getAssertion(), 100); assertEquals(quota.getRole(), 1000); assertEquals(quota.getPolicy(), 1000); }