Search in sources :

Example 6 with ClusterStatus

use of com.ctrip.xpipe.redis.console.migration.status.ClusterStatus in project x-pipe by ctripcorp.

the class DefaultMigrationClusterTest2 method testUpdateStat.

@Test(expected = ServerException.class)
public void testUpdateStat() throws Exception {
    AtomicInteger counter = new AtomicInteger(1);
    when(clusterService.find(anyString())).thenReturn(clusterTbl);
    doAnswer(new Answer() {

        @Override
        public Void answer(InvocationOnMock invocation) throws Throwable {
            int currentCounter = counter.getAndIncrement();
            if (currentCounter % 5 == 0) {
                clusterTbl.setStatus(((ClusterStatus) invocation.getArguments()[1]).toString());
            }
            return null;
        }
    }).when(clusterService).updateStatusById(anyLong(), any());
    migrationCluster.updateStat(new MigrationCheckingState(migrationCluster));
}
Also used : Answer(org.mockito.stubbing.Answer) Mockito.doAnswer(org.mockito.Mockito.doAnswer) MigrationCheckingState(com.ctrip.xpipe.redis.console.migration.status.migration.MigrationCheckingState) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) InvocationOnMock(org.mockito.invocation.InvocationOnMock) ClusterStatus(com.ctrip.xpipe.redis.console.migration.status.ClusterStatus) Test(org.junit.Test)

Aggregations

ClusterStatus (com.ctrip.xpipe.redis.console.migration.status.ClusterStatus)6 Test (org.junit.Test)6 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)4 Mockito.doAnswer (org.mockito.Mockito.doAnswer)4 InvocationOnMock (org.mockito.invocation.InvocationOnMock)4 Answer (org.mockito.stubbing.Answer)4 AbstractConsoleTest (com.ctrip.xpipe.redis.console.AbstractConsoleTest)1 MigrationCheckingState (com.ctrip.xpipe.redis.console.migration.status.migration.MigrationCheckingState)1 MigrationMigratingState (com.ctrip.xpipe.redis.console.migration.status.migration.MigrationMigratingState)1 MigrationPublishState (com.ctrip.xpipe.redis.console.migration.status.migration.MigrationPublishState)1 ClusterTbl (com.ctrip.xpipe.redis.console.model.ClusterTbl)1 DcTbl (com.ctrip.xpipe.redis.console.model.DcTbl)1 MigrationClusterTbl (com.ctrip.xpipe.redis.console.model.MigrationClusterTbl)1