Author: btellier Date: Tue Sep 22 10:19:51 2015 New Revision: 1704530 URL: http://svn.apache.org/viewvc?rev=1704530&view=rev Log: MAILBOX-64 Store Quota Manager implementation
Added: james/mailbox/trunk/store/src/main/java/org/apache/james/mailbox/store/quota/StoreQuotaManager.java james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/StoreQuotaManagerTest.java Added: james/mailbox/trunk/store/src/main/java/org/apache/james/mailbox/store/quota/StoreQuotaManager.java URL: http://svn.apache.org/viewvc/james/mailbox/trunk/store/src/main/java/org/apache/james/mailbox/store/quota/StoreQuotaManager.java?rev=1704530&view=auto ============================================================================== --- james/mailbox/trunk/store/src/main/java/org/apache/james/mailbox/store/quota/StoreQuotaManager.java (added) +++ james/mailbox/trunk/store/src/main/java/org/apache/james/mailbox/store/quota/StoreQuotaManager.java Tue Sep 22 10:19:51 2015 @@ -0,0 +1,72 @@ +/**************************************************************** + * Licensed to the Apache Software Foundation (ASF) under one * + * or more contributor license agreements. See the NOTICE file * + * distributed with this work for additional information * + * regarding copyright ownership. The ASF licenses this file * + * to you under the Apache License, Version 2.0 (the * + * "License"); you may not use this file except in compliance * + * with the License. You may obtain a copy of the License at * + * * + * http://www.apache.org/licenses/LICENSE-2.0 * + * * + * Unless required by applicable law or agreed to in writing, * + * software distributed under the License is distributed on an * + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * + * KIND, either express or implied. See the License for the * + * specific language governing permissions and limitations * + * under the License. * + ****************************************************************/ + +package org.apache.james.mailbox.store.quota; + +import org.apache.james.mailbox.quota.CurrentQuotaManager; +import org.apache.james.mailbox.quota.MaxQuotaManager; +import org.apache.james.mailbox.quota.QuotaManager; +import org.apache.james.mailbox.exception.MailboxException; +import org.apache.james.mailbox.model.Quota; +import org.apache.james.mailbox.model.QuotaRoot; + +import javax.inject.Inject; + +/** + * Default implementation for the Quota Manager. + * + * Relies on the CurrentQuotaManager and MaxQuotaManager provided. + */ +public class StoreQuotaManager implements QuotaManager { + private CurrentQuotaManager currentQuotaManager; + private MaxQuotaManager maxQuotaManager; + private boolean calculateWhenUnlimited = false; + + public void setCalculateWhenUnlimited(boolean calculateWhenUnlimited) { + this.calculateWhenUnlimited = calculateWhenUnlimited; + } + + @Inject + public void setMaxQuotaManager(MaxQuotaManager maxQuotaManager) { + this.maxQuotaManager = maxQuotaManager; + } + + @Inject + public void setCurrentQuotaManager(CurrentQuotaManager currentQuotaManager) { + this.currentQuotaManager = currentQuotaManager; + } + + public Quota getMessageQuota(QuotaRoot quotaRoot) throws MailboxException { + long maxValue = maxQuotaManager.getMaxMessage(quotaRoot); + if(maxValue == Quota.UNLIMITED && !calculateWhenUnlimited) { + return QuotaImpl.quota(Quota.UNKNOWN, Quota.UNLIMITED); + } + return QuotaImpl.quota(currentQuotaManager.getCurrentMessageCount(quotaRoot), maxValue); + } + + + public Quota getStorageQuota(QuotaRoot quotaRoot) throws MailboxException { + long maxValue = maxQuotaManager.getMaxStorage(quotaRoot); + if(maxValue == Quota.UNLIMITED && !calculateWhenUnlimited) { + return QuotaImpl.quota(Quota.UNKNOWN, Quota.UNLIMITED); + } + return QuotaImpl.quota(currentQuotaManager.getCurrentStorage(quotaRoot), maxValue); + } + +} \ No newline at end of file Added: james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/StoreQuotaManagerTest.java URL: http://svn.apache.org/viewvc/james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/StoreQuotaManagerTest.java?rev=1704530&view=auto ============================================================================== --- james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/StoreQuotaManagerTest.java (added) +++ james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/StoreQuotaManagerTest.java Tue Sep 22 10:19:51 2015 @@ -0,0 +1,151 @@ +/**************************************************************** + * Licensed to the Apache Software Foundation (ASF) under one * + * or more contributor license agreements. See the NOTICE file * + * distributed with this work for additional information * + * regarding copyright ownership. The ASF licenses this file * + * to you under the Apache License, Version 2.0 (the * + * "License"); you may not use this file except in compliance * + * with the License. You may obtain a copy of the License at * + * * + * http://www.apache.org/licenses/LICENSE-2.0 * + * * + * Unless required by applicable law or agreed to in writing, * + * software distributed under the License is distributed on an * + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * + * KIND, either express or implied. See the License for the * + * specific language governing permissions and limitations * + * under the License. * + ****************************************************************/ + +package org.apache.james.mailbox.store.quota; + +import static org.assertj.core.api.Assertions.assertThat; + +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.never; +import static org.mockito.Mockito.verify; +import static org.mockito.Mockito.when; + +import org.apache.james.mailbox.model.Quota; +import org.apache.james.mailbox.model.QuotaRoot; +import org.apache.james.mailbox.quota.CurrentQuotaManager; +import org.apache.james.mailbox.quota.MaxQuotaManager; +import org.junit.Before; +import org.junit.Test; +import org.mockito.invocation.InvocationOnMock; +import org.mockito.stubbing.Answer; + +public class StoreQuotaManagerTest { + + private StoreQuotaManager testee; + private CurrentQuotaManager mockedCurrentQuotaManager; + private MaxQuotaManager mockedMaxQuotaManager; + private QuotaRoot quotaRoot; + + @Before + public void setUp() { + mockedCurrentQuotaManager = mock(CurrentQuotaManager.class); + mockedMaxQuotaManager = mock(MaxQuotaManager.class); + testee = new StoreQuotaManager(); + testee.setCurrentQuotaManager(mockedCurrentQuotaManager); + testee.setMaxQuotaManager(mockedMaxQuotaManager); + quotaRoot = QuotaRootImpl.quotaRoot("benwa"); + } + + @Test + public void test() throws Exception { + when(mockedMaxQuotaManager.getMaxMessage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 360L; + } + }); + when(mockedCurrentQuotaManager.getCurrentMessageCount(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 36L; + } + }); + assertThat(testee.getMessageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(36, 360)); + } + + @Test + public void test2() throws Exception { + when(mockedMaxQuotaManager.getMaxStorage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 360L; + } + }); + when(mockedCurrentQuotaManager.getCurrentStorage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 36L; + } + }); + assertThat(testee.getStorageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(36, 360)); + } + + @Test + public void test3() throws Exception { + testee.setCalculateWhenUnlimited(false); + when(mockedMaxQuotaManager.getMaxStorage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return Quota.UNLIMITED; + } + }); + assertThat(testee.getStorageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(Quota.UNKNOWN, Quota.UNLIMITED)); + verify(mockedCurrentQuotaManager, never()).getCurrentStorage(quotaRoot); + } + + @Test + public void test4() throws Exception { + testee.setCalculateWhenUnlimited(false); + when(mockedMaxQuotaManager.getMaxMessage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return Quota.UNLIMITED; + } + }); + assertThat(testee.getMessageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(Quota.UNKNOWN, Quota.UNLIMITED)); + verify(mockedCurrentQuotaManager, never()).getCurrentMessageCount(quotaRoot); + } + + @Test + public void test5() throws Exception { + testee.setCalculateWhenUnlimited(true); + when(mockedMaxQuotaManager.getMaxStorage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return Quota.UNLIMITED; + } + }); + when(mockedCurrentQuotaManager.getCurrentStorage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 36L; + } + }); + assertThat(testee.getStorageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(36, Quota.UNLIMITED)); + } + + @Test + public void test6() throws Exception { + testee.setCalculateWhenUnlimited(true); + when(mockedMaxQuotaManager.getMaxMessage(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return Quota.UNLIMITED; + } + }); + when(mockedCurrentQuotaManager.getCurrentMessageCount(quotaRoot)).then(new Answer<Long>() { + @Override + public Long answer(InvocationOnMock invocationOnMock) throws Throwable { + return 36L; + } + }); + assertThat(testee.getMessageQuota(quotaRoot)).isEqualTo(QuotaImpl.quota(36, Quota.UNLIMITED)); + } + +} --------------------------------------------------------------------- To unsubscribe, e-mail: server-dev-unsubscr...@james.apache.org For additional commands, e-mail: server-dev-h...@james.apache.org