You are viewing a plain text version of this content. The canonical link for it is here.
Posted to server-dev@james.apache.org by bt...@apache.org on 2015/09/22 12:25:04 UTC

svn commit: r1704540 - in /james/mailbox/trunk: cassandra/src/main/java/org/apache/james/mailbox/cassandra/ cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/ cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/ cassandra/sr...

Author: btellier
Date: Tue Sep 22 10:25:00 2015
New Revision: 1704540

URL: http://svn.apache.org/viewvc?rev=1704540&view=rev
Log:
MAILBOX-64 Cassandra stored MaxQuotaManager

Added:
    james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/
    james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManager.java
    james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraDefaultMaxQuota.java
    james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraMaxQuota.java
    james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/
    james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManagerTest.java
    james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/GenericMaxQuotaManagerTest.java
Modified:
    james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/CassandraTableManager.java
    james/mailbox/trunk/memory/src/test/java/org/apache/james/mailbox/inmemory/quota/InMemoryPerUserMaxQuotaManagerTest.java

Modified: james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/CassandraTableManager.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/CassandraTableManager.java?rev=1704540&r1=1704539&r2=1704540&view=diff
==============================================================================
--- james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/CassandraTableManager.java (original)
+++ james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/CassandraTableManager.java Tue Sep 22 10:25:00 2015
@@ -36,8 +36,10 @@ import com.datastax.driver.core.schemabu
 import com.datastax.driver.core.schemabuilder.SchemaStatement;
 
 import org.apache.james.mailbox.cassandra.table.CassandraACLTable;
+import org.apache.james.mailbox.cassandra.table.CassandraDefaultMaxQuota;
 import org.apache.james.mailbox.cassandra.table.CassandraMailboxCountersTable;
 import org.apache.james.mailbox.cassandra.table.CassandraMailboxTable;
+import org.apache.james.mailbox.cassandra.table.CassandraMaxQuota;
 import org.apache.james.mailbox.cassandra.table.CassandraMessageModseqTable;
 import org.apache.james.mailbox.cassandra.table.CassandraMessageTable;
 import org.apache.james.mailbox.cassandra.table.CassandraMessageUidTable;
@@ -109,7 +111,18 @@ public class CassandraTableManager {
             SchemaBuilder.createTable(CassandraMessageModseqTable.TABLE_NAME)
                 .ifNotExists()
                 .addPartitionKey(CassandraMessageModseqTable.MAILBOX_ID, timeuuid())
-                .addColumn(CassandraMessageModseqTable.NEXT_MODSEQ, bigint()))
+                .addColumn(CassandraMessageModseqTable.NEXT_MODSEQ, bigint())),
+        MaxQuota(CassandraMaxQuota.TABLE_NAME,
+            SchemaBuilder.createTable(CassandraMaxQuota.TABLE_NAME)
+                .ifNotExists()
+                .addPartitionKey(CassandraMaxQuota.QUOTA_ROOT, text())
+                .addColumn(CassandraMaxQuota.MESSAGE_COUNT, bigint())
+                .addColumn(CassandraMaxQuota.STORAGE, bigint())),
+        DefaultMaxQuota(CassandraDefaultMaxQuota.TABLE_NAME,
+            SchemaBuilder.createTable(CassandraDefaultMaxQuota.TABLE_NAME)
+                .ifNotExists()
+                .addPartitionKey(CassandraDefaultMaxQuota.TYPE, text())
+                .addColumn(CassandraDefaultMaxQuota.VALUE, bigint()))
         ;
 
         private Create createStatement;

Added: james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManager.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManager.java?rev=1704540&view=auto
==============================================================================
--- james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManager.java (added)
+++ james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManager.java Tue Sep 22 10:25:00 2015
@@ -0,0 +1,133 @@
+/****************************************************************
+ * 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.cassandra.quota;
+
+import static com.datastax.driver.core.querybuilder.QueryBuilder.bindMarker;
+import static com.datastax.driver.core.querybuilder.QueryBuilder.eq;
+import static com.datastax.driver.core.querybuilder.QueryBuilder.insertInto;
+import static com.datastax.driver.core.querybuilder.QueryBuilder.select;
+
+import com.datastax.driver.core.PreparedStatement;
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Session;
+import com.datastax.driver.core.Statement;
+import org.apache.james.mailbox.cassandra.table.CassandraDefaultMaxQuota;
+import org.apache.james.mailbox.cassandra.table.CassandraMaxQuota;
+import org.apache.james.mailbox.exception.MailboxException;
+import org.apache.james.mailbox.model.Quota;
+import org.apache.james.mailbox.model.QuotaRoot;
+import org.apache.james.mailbox.quota.MaxQuotaManager;
+
+public class CassandraPerUserMaxQuotaManager implements MaxQuotaManager {
+
+    private final Session session;
+    private final PreparedStatement setMaxStorageStatement;
+    private final PreparedStatement setMaxMessageStatement;
+    private final PreparedStatement getMaxStorageStatement;
+    private final PreparedStatement getMaxMessageStatement;
+    private final PreparedStatement setDefaultMaxStorageStatement;
+    private final PreparedStatement setDefaultMaxMessageStatement;
+    private final Statement getDefaultMaxStorageStatement;
+    private final Statement getDefaultMaxMessageStatement;
+
+    public CassandraPerUserMaxQuotaManager(Session session) {
+        this.session = session;
+        this.setMaxStorageStatement = session.prepare(insertInto(CassandraMaxQuota.TABLE_NAME)
+            .value(CassandraMaxQuota.QUOTA_ROOT, bindMarker())
+            .value(CassandraMaxQuota.STORAGE, bindMarker()));
+        this.setMaxMessageStatement = session.prepare(insertInto(CassandraMaxQuota.TABLE_NAME)
+            .value(CassandraMaxQuota.QUOTA_ROOT, bindMarker())
+            .value(CassandraMaxQuota.MESSAGE_COUNT, bindMarker()));
+        this.getMaxStorageStatement = session.prepare(select(CassandraMaxQuota.STORAGE)
+            .from(CassandraMaxQuota.TABLE_NAME)
+            .where(eq(CassandraMaxQuota.QUOTA_ROOT, bindMarker())));
+        this.getMaxMessageStatement = session.prepare(select(CassandraMaxQuota.MESSAGE_COUNT)
+            .from(CassandraMaxQuota.TABLE_NAME)
+            .where(eq(CassandraMaxQuota.QUOTA_ROOT, bindMarker())));
+        this.getDefaultMaxMessageStatement = select(CassandraDefaultMaxQuota.VALUE)
+            .from(CassandraDefaultMaxQuota.TABLE_NAME)
+            .where(eq(CassandraDefaultMaxQuota.TYPE, CassandraDefaultMaxQuota.MESSAGE));
+        this.getDefaultMaxStorageStatement = select(CassandraDefaultMaxQuota.VALUE)
+            .from(CassandraDefaultMaxQuota.TABLE_NAME)
+            .where(eq(CassandraDefaultMaxQuota.TYPE, CassandraDefaultMaxQuota.STORAGE));
+        this.setDefaultMaxMessageStatement = session.prepare(insertInto(CassandraDefaultMaxQuota.TABLE_NAME)
+            .value(CassandraDefaultMaxQuota.TYPE, CassandraDefaultMaxQuota.MESSAGE)
+            .value(CassandraDefaultMaxQuota.VALUE, bindMarker()));
+        this.setDefaultMaxStorageStatement = session.prepare(insertInto(CassandraDefaultMaxQuota.TABLE_NAME)
+            .value(CassandraDefaultMaxQuota.TYPE, CassandraDefaultMaxQuota.STORAGE)
+            .value(CassandraDefaultMaxQuota.VALUE, bindMarker()));
+    }
+
+    @Override
+    public void setMaxStorage(QuotaRoot quotaRoot, long maxStorageQuota) throws MailboxException {
+        session.execute(setMaxStorageStatement.bind(quotaRoot.getValue(), maxStorageQuota));
+    }
+
+    @Override
+    public void setMaxMessage(QuotaRoot quotaRoot, long maxMessageCount) throws MailboxException {
+        session.execute(setMaxMessageStatement.bind(quotaRoot.getValue(), maxMessageCount));
+    }
+
+    @Override
+    public void setDefaultMaxStorage(long defaultMaxStorage) throws MailboxException {
+        session.execute(setDefaultMaxStorageStatement.bind(defaultMaxStorage));
+    }
+
+    @Override
+    public void setDefaultMaxMessage(long defaultMaxMessageCount) throws MailboxException {
+        session.execute(setDefaultMaxMessageStatement.bind(defaultMaxMessageCount));
+    }
+
+    @Override
+    public long getDefaultMaxStorage() throws MailboxException {
+        ResultSet resultSet = session.execute(getDefaultMaxStorageStatement);
+        if (resultSet.isExhausted()) {
+            return Quota.UNLIMITED;
+        }
+        return resultSet.one().getLong(CassandraDefaultMaxQuota.VALUE);
+    }
+
+    @Override
+    public long getDefaultMaxMessage() throws MailboxException {
+        ResultSet resultSet = session.execute(getDefaultMaxMessageStatement);
+        if (resultSet.isExhausted()) {
+            return Quota.UNLIMITED;
+        }
+        return resultSet.one().getLong(CassandraDefaultMaxQuota.VALUE);
+    }
+
+    @Override
+    public long getMaxStorage(QuotaRoot quotaRoot) throws MailboxException {
+        ResultSet resultSet = session.execute(getMaxStorageStatement.bind(quotaRoot.getValue()));
+        if (resultSet.isExhausted()) {
+            return getDefaultMaxStorage();
+        }
+        return resultSet.one().getLong(CassandraMaxQuota.STORAGE);
+    }
+
+    @Override
+    public long getMaxMessage(QuotaRoot quotaRoot) throws MailboxException {
+        ResultSet resultSet = session.execute(getMaxMessageStatement.bind(quotaRoot.getValue()));
+        if (resultSet.isExhausted()) {
+            return getDefaultMaxMessage();
+        }
+        return resultSet.one().getLong(CassandraMaxQuota.MESSAGE_COUNT);
+    }
+}

Added: james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraDefaultMaxQuota.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraDefaultMaxQuota.java?rev=1704540&view=auto
==============================================================================
--- james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraDefaultMaxQuota.java (added)
+++ james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraDefaultMaxQuota.java Tue Sep 22 10:25:00 2015
@@ -0,0 +1,30 @@
+/****************************************************************
+ * 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.cassandra.table;
+
+public interface CassandraDefaultMaxQuota {
+    String TABLE_NAME = "defaultMaxQuota";
+
+    String TYPE = "type";
+    String VALUE = "value";
+
+    String MESSAGE = "message";
+    String STORAGE = "storage";
+}

Added: james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraMaxQuota.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraMaxQuota.java?rev=1704540&view=auto
==============================================================================
--- james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraMaxQuota.java (added)
+++ james/mailbox/trunk/cassandra/src/main/java/org/apache/james/mailbox/cassandra/table/CassandraMaxQuota.java Tue Sep 22 10:25:00 2015
@@ -0,0 +1,28 @@
+/****************************************************************
+ * 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.cassandra.table;
+
+public interface CassandraMaxQuota {
+    String TABLE_NAME = "maxQuota";
+
+    String QUOTA_ROOT = "quotaRoot";
+    String MESSAGE_COUNT = "maxMessageCount";
+    String STORAGE = "maxStorage";
+}

Added: james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManagerTest.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManagerTest.java?rev=1704540&view=auto
==============================================================================
--- james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManagerTest.java (added)
+++ james/mailbox/trunk/cassandra/src/test/java/org/apache/james/mailbox/cassandra/quota/CassandraPerUserMaxQuotaManagerTest.java Tue Sep 22 10:25:00 2015
@@ -0,0 +1,43 @@
+/****************************************************************
+ * 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.cassandra.quota;
+
+import org.apache.james.mailbox.cassandra.CassandraClusterSingleton;
+import org.apache.james.mailbox.quota.MaxQuotaManager;
+import org.apache.james.mailbox.store.quota.GenericMaxQuotaManagerTest;
+import org.junit.After;
+
+public class CassandraPerUserMaxQuotaManagerTest extends GenericMaxQuotaManagerTest {
+
+    private CassandraClusterSingleton cassandra;
+
+    @Override
+    protected MaxQuotaManager provideMaxQuotaManager() {
+        cassandra = CassandraClusterSingleton.build();
+        cassandra.ensureAllTables();
+        return new CassandraPerUserMaxQuotaManager(cassandra.getConf());
+    }
+
+    @After
+    public void cleanUp() {
+        cassandra.clearAllTables();
+    }
+
+}

Modified: james/mailbox/trunk/memory/src/test/java/org/apache/james/mailbox/inmemory/quota/InMemoryPerUserMaxQuotaManagerTest.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/memory/src/test/java/org/apache/james/mailbox/inmemory/quota/InMemoryPerUserMaxQuotaManagerTest.java?rev=1704540&r1=1704539&r2=1704540&view=diff
==============================================================================
--- james/mailbox/trunk/memory/src/test/java/org/apache/james/mailbox/inmemory/quota/InMemoryPerUserMaxQuotaManagerTest.java (original)
+++ james/mailbox/trunk/memory/src/test/java/org/apache/james/mailbox/inmemory/quota/InMemoryPerUserMaxQuotaManagerTest.java Tue Sep 22 10:25:00 2015
@@ -19,57 +19,13 @@
 
 package org.apache.james.mailbox.inmemory.quota;
 
-import static org.assertj.core.api.Assertions.assertThat;
+import org.apache.james.mailbox.quota.MaxQuotaManager;
+import org.apache.james.mailbox.store.quota.GenericMaxQuotaManagerTest;
 
-import org.apache.james.mailbox.model.Quota;
-import org.apache.james.mailbox.model.QuotaRoot;
-import org.apache.james.mailbox.store.quota.QuotaRootImpl;
-import org.junit.Before;
-import org.junit.Test;
+public class InMemoryPerUserMaxQuotaManagerTest extends GenericMaxQuotaManagerTest {
 
-public class InMemoryPerUserMaxQuotaManagerTest {
-
-    private InMemoryPerUserMaxQuotaManager perUserMaxQuotaManager;
-    private QuotaRoot quotaRoot;
-
-    @Before
-    public void setUp() {
-        perUserMaxQuotaManager = new InMemoryPerUserMaxQuotaManager();
-        quotaRoot = QuotaRootImpl.quotaRoot("benwa");
-    }
-
-    @Test
-    public void getMaxMessageShouldReturnUnlimitedWhenNoDefaultValue() throws Exception {
-        assertThat(perUserMaxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(Quota.UNLIMITED);
-    }
-
-    @Test
-    public void getMaxStorageShouldReturnUnlimitedWhenNoDefaultValue() throws Exception {
-        assertThat(perUserMaxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(Quota.UNLIMITED);
-    }
-
-    @Test
-    public void getMaxMessageShouldReturnDefaultWhenNoValue() throws Exception {
-        perUserMaxQuotaManager.setDefaultMaxMessage(36);
-        assertThat(perUserMaxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(36);
-    }
-
-    @Test
-    public void getMaxStorageShouldReturnDefaultWhenNoValue() throws Exception {
-        perUserMaxQuotaManager.setDefaultMaxStorage(36);
-        assertThat(perUserMaxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(36);
-    }
-
-    @Test
-    public void getMaxMessageShouldReturnProvidedValue() throws Exception {
-        perUserMaxQuotaManager.setMaxMessage(quotaRoot, 36);
-        assertThat(perUserMaxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(36);
-    }
-
-    @Test
-    public void getMaxStorageShouldReturnProvidedValue() throws Exception {
-        perUserMaxQuotaManager.setMaxStorage(quotaRoot, 36);
-        assertThat(perUserMaxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(36);
+    protected MaxQuotaManager provideMaxQuotaManager() {
+        return new InMemoryPerUserMaxQuotaManager();
     }
 
 }

Added: james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/GenericMaxQuotaManagerTest.java
URL: http://svn.apache.org/viewvc/james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/GenericMaxQuotaManagerTest.java?rev=1704540&view=auto
==============================================================================
--- james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/GenericMaxQuotaManagerTest.java (added)
+++ james/mailbox/trunk/store/src/test/java/org/apache/james/mailbox/store/quota/GenericMaxQuotaManagerTest.java Tue Sep 22 10:25:00 2015
@@ -0,0 +1,77 @@
+/****************************************************************
+ * 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.model.Quota;
+import org.apache.james.mailbox.model.QuotaRoot;
+import org.apache.james.mailbox.quota.MaxQuotaManager;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public abstract class GenericMaxQuotaManagerTest {
+
+    private QuotaRoot quotaRoot;
+    private MaxQuotaManager maxQuotaManager;
+
+    protected abstract MaxQuotaManager provideMaxQuotaManager();
+
+    @Before
+    public void setUp() {
+        maxQuotaManager = provideMaxQuotaManager();
+        quotaRoot = QuotaRootImpl.quotaRoot("benwa");
+    }
+
+    @Test
+    public void getMaxMessageShouldReturnUnlimitedWhenNoDefaultValue() throws Exception {
+        assertThat(maxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(Quota.UNLIMITED);
+    }
+
+    @Test
+    public void getMaxStorageShouldReturnUnlimitedWhenNoDefaultValue() throws Exception {
+        assertThat(maxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(Quota.UNLIMITED);
+    }
+
+    @Test
+    public void getMaxMessageShouldReturnDefaultWhenNoValue() throws Exception {
+        maxQuotaManager.setDefaultMaxMessage(36);
+        assertThat(maxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(36);
+    }
+
+    @Test
+    public void getMaxStorageShouldReturnDefaultWhenNoValue() throws Exception {
+        maxQuotaManager.setDefaultMaxStorage(36);
+        assertThat(maxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(36);
+    }
+
+    @Test
+    public void getMaxMessageShouldReturnProvidedValue() throws Exception {
+        maxQuotaManager.setMaxMessage(quotaRoot, 36);
+        assertThat(maxQuotaManager.getMaxMessage(quotaRoot)).isEqualTo(36);
+    }
+
+    @Test
+    public void getMaxStorageShouldReturnProvidedValue() throws Exception {
+        maxQuotaManager.setMaxStorage(quotaRoot, 36);
+        assertThat(maxQuotaManager.getMaxStorage(quotaRoot)).isEqualTo(36);
+    }
+
+}



---------------------------------------------------------------------
To unsubscribe, e-mail: server-dev-unsubscribe@james.apache.org
For additional commands, e-mail: server-dev-help@james.apache.org