when(mockConfig.isRunRawDataMigration()).thenReturn(true);
when(mockConfig.isRun1HAggregateDataMigration()).thenReturn(false);
when(mockConfig.isRun6HAggregateDataMigration()).thenReturn(false);
when(mockConfig.isRun1DAggregateDataMigration()).thenReturn(false);
RawDataMigrator mockRawDataMigrator = mock(RawDataMigrator.class);
PowerMockito.whenNew(RawDataMigrator.class).withArguments(eq(mockConfig)).thenReturn(mockRawDataMigrator);
long estimateExpected = 1234L;
when(mockRawDataMigrator.estimate()).thenReturn(estimateExpected);
//create object to test and inject required dependencies
DataMigrator objectUnderTest = new DataMigrator(mockEntityManager, mockCassandraSession, databaseType);
//run code under test