Repository: ignite Updated Branches: refs/heads/master 19772109d -> 957b59c35
http://git-wip-us.apache.org/repos/asf/ignite/blob/957b59c3/modules/ml/src/test/java/org/apache/ignite/ml/inference/storage/model/DefaultModelStorageTest.java ---------------------------------------------------------------------- diff --git a/modules/ml/src/test/java/org/apache/ignite/ml/inference/storage/model/DefaultModelStorageTest.java b/modules/ml/src/test/java/org/apache/ignite/ml/inference/storage/model/DefaultModelStorageTest.java new file mode 100644 index 0000000..2c2a678 --- /dev/null +++ b/modules/ml/src/test/java/org/apache/ignite/ml/inference/storage/model/DefaultModelStorageTest.java @@ -0,0 +1,143 @@ +/* + * 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.ignite.ml.inference.storage.model; + +import java.util.concurrent.locks.Lock; +import org.junit.Test; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.fail; +import static org.mockito.Mockito.doThrow; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.times; +import static org.mockito.Mockito.verify; +import static org.mockito.Mockito.verifyNoMoreInteractions; + +/** + * Tests for {@link DefaultModelStorage}. + */ +public class DefaultModelStorageTest extends AbstractModelStorageTest { + /** {@inheritDoc} */ + @Override ModelStorage getModelStorage() { + ModelStorageProvider provider = new LocalModelStorageProvider(); + return new DefaultModelStorage(provider); + } + + /** */ + @Test + public void testSynchronize() { + Lock[] locks = new Lock[10]; + for (int i = 0; i < locks.length; i++) + locks[i] = mock(Lock.class); + + DefaultModelStorage.synchronize(() -> {}, locks); + + for (Lock lock : locks) { + verify(lock, times(1)).lock(); + verify(lock, times(1)).unlock(); + verifyNoMoreInteractions(lock); + } + } + + /** */ + @Test + public void testSynchronizeWithExceptionInTask() { + Lock[] locks = new Lock[10]; + for (int i = 0; i < locks.length; i++) + locks[i] = mock(Lock.class); + + RuntimeException ex = new RuntimeException(); + + try { + DefaultModelStorage.synchronize(() -> { throw ex; }, locks); + fail(); + } + catch (RuntimeException e) { + assertEquals(ex, e); + } + + for (Lock lock : locks) { + verify(lock, times(1)).lock(); + verify(lock, times(1)).unlock(); + verifyNoMoreInteractions(lock); + } + } + + /** */ + @Test + public void testSynchronizeWithExceptionInLock() { + Lock[] locks = new Lock[10]; + for (int i = 0; i < locks.length; i++) + locks[i] = mock(Lock.class); + + RuntimeException ex = new RuntimeException(); + + doThrow(ex).when(locks[5]).lock(); + + try { + DefaultModelStorage.synchronize(() -> {}, locks); + fail(); + } + catch (RuntimeException e) { + assertEquals(ex, e); + } + + for (int i = 0; i < locks.length; i++) { + if (i <= 4) { + verify(locks[i], times(1)).lock(); + verify(locks[i], times(1)).unlock(); + } + else if (i > 5) { + verify(locks[i], times(0)).lock(); + verify(locks[i], times(0)).unlock(); + } + else { + verify(locks[i], times(1)).lock(); + verify(locks[i], times(0)).unlock(); + } + + verifyNoMoreInteractions(locks[i]); + } + } + + /** */ + @Test + public void testSynchronizeWithExceptionInUnlock() { + Lock[] locks = new Lock[10]; + for (int i = 0; i < locks.length; i++) + locks[i] = mock(Lock.class); + + RuntimeException ex = new RuntimeException(); + + doThrow(ex).when(locks[5]).unlock(); + + try { + DefaultModelStorage.synchronize(() -> {}, locks); + fail(); + } + catch (RuntimeException e) { + assertEquals(ex, e); + } + + for (Lock lock : locks) { + verify(lock, times(1)).lock(); + verify(lock, times(1)).unlock(); + verifyNoMoreInteractions(lock); + } + } +}
