Skip to content

Commit 7cd4e92

Browse files
igorbernstein2ad548
authored andcommitted
fix(test): Clean up tests (googleapis#439)
* fix(test): Clean up tests * Add missing RunWith annotations * unnecessary downcasts * Make emulator path overridable * add missing annotation * lint
1 parent 75d1105 commit 7cd4e92

10 files changed

Lines changed: 40 additions & 7 deletions

File tree

google-cloud-bigtable-emulator/src/main/java/com/google/cloud/bigtable/emulator/v2/Emulator.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@
4242
*/
4343
@BetaApi("Surface for Bigtable emulator is not yet stable")
4444
public class Emulator {
45-
4645
private static final Logger LOGGER = Logger.getLogger(Emulator.class.getName());
4746

4847
private final Path executable;
@@ -54,6 +53,9 @@ public class Emulator {
5453
private ManagedChannel dataChannel;
5554
private ManagedChannel adminChannel;
5655

56+
public static Emulator createFromPath(Path path) {
57+
return new Emulator(path);
58+
}
5759
/**
5860
* Create a new instance of emulator. The emulator will use the bundled binaries in this jar.
5961
* Please note that the emulator is created in a stopped state, please use {@link #start()} after

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminSettingsTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,11 @@
2121
import com.google.api.gax.rpc.StatusCode.Code;
2222
import java.io.IOException;
2323
import org.junit.Test;
24+
import org.junit.runner.RunWith;
25+
import org.junit.runners.JUnit4;
2426
import org.mockito.Mockito;
2527

28+
@RunWith(JUnit4.class)
2629
public class BigtableInstanceAdminSettingsTest {
2730
@Test
2831
public void testProjectName() throws Exception {

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientTest.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,18 +75,22 @@
7575
import java.util.concurrent.ExecutionException;
7676
import java.util.concurrent.atomic.AtomicBoolean;
7777
import org.junit.Before;
78+
import org.junit.Rule;
7879
import org.junit.Test;
7980
import org.junit.runner.RunWith;
81+
import org.junit.runners.JUnit4;
8082
import org.mockito.Matchers;
8183
import org.mockito.Mock;
8284
import org.mockito.Mockito;
8385
import org.mockito.invocation.InvocationOnMock;
84-
import org.mockito.runners.MockitoJUnitRunner;
86+
import org.mockito.junit.MockitoJUnit;
87+
import org.mockito.junit.MockitoRule;
8588
import org.mockito.stubbing.Answer;
8689
import org.threeten.bp.Instant;
8790

88-
@RunWith(MockitoJUnitRunner.class)
91+
@RunWith(JUnit4.class)
8992
public class BigtableTableAdminClientTest {
93+
@Rule public final MockitoRule mockitoRule = MockitoJUnit.rule();
9094

9195
private static final String PROJECT_ID = "my-project";
9296
private static final String INSTANCE_ID = "my-instance";

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminSettingsTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,11 @@
2121
import com.google.api.gax.rpc.StatusCode.Code;
2222
import java.io.IOException;
2323
import org.junit.Test;
24+
import org.junit.runner.RunWith;
25+
import org.junit.runners.JUnit4;
2426
import org.mockito.Mockito;
2527

28+
@RunWith(JUnit4.class)
2629
public class BigtableTableAdminSettingsTest {
2730

2831
@Test

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableBackupIT.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,12 @@
4747
import java.util.concurrent.TimeoutException;
4848
import java.util.logging.Logger;
4949
import org.junit.*;
50+
import org.junit.runner.RunWith;
51+
import org.junit.runners.JUnit4;
5052
import org.threeten.bp.Duration;
5153
import org.threeten.bp.Instant;
5254

55+
@RunWith(JUnit4.class)
5356
public class BigtableBackupIT {
5457
private static final Logger LOGGER = Logger.getLogger(BigtableBackupIT.class.getName());
5558

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableInstanceAdminClientIT.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,11 @@
4040
import org.junit.BeforeClass;
4141
import org.junit.ClassRule;
4242
import org.junit.Test;
43+
import org.junit.runner.RunWith;
44+
import org.junit.runners.JUnit4;
4345
import org.threeten.bp.Instant;
4446

47+
@RunWith(JUnit4.class)
4548
public class BigtableInstanceAdminClientIT {
4649

4750
@ClassRule public static TestEnvRule testEnvRule = new TestEnvRule();

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableTableAdminClientIT.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,11 @@
4646
import org.junit.Rule;
4747
import org.junit.Test;
4848
import org.junit.rules.TestName;
49+
import org.junit.runner.RunWith;
50+
import org.junit.runners.JUnit4;
4951
import org.threeten.bp.Duration;
5052

53+
@RunWith(JUnit4.class)
5154
public class BigtableTableAdminClientIT {
5255
@ClassRule public static TestEnvRule testEnvRule = new TestEnvRule();
5356
@Rule public final TestName testNameRule = new TestName();

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/models/ReadModifyWriteRowTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,10 @@
2828
import java.io.ObjectInputStream;
2929
import java.io.ObjectOutputStream;
3030
import org.junit.Test;
31+
import org.junit.runner.RunWith;
32+
import org.junit.runners.JUnit4;
3133

34+
@RunWith(JUnit4.class)
3235
public class ReadModifyWriteRowTest {
3336
private static final String PROJECT_ID = "fake-project";
3437
private static final String INSTANCE_ID = "fake-instance";

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/EnhancedBigtableStubTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,11 +138,11 @@ public void testChannelPrimerConfigured() throws IOException {
138138
}
139139

140140
private static class FakeDataService extends BigtableGrpc.BigtableImplBase {
141-
final BlockingQueue<Object> requests = Queues.newLinkedBlockingDeque();
141+
final BlockingQueue<ReadRowsRequest> requests = Queues.newLinkedBlockingDeque();
142142

143143
@SuppressWarnings("unchecked")
144-
<T> T popLastRequest() throws InterruptedException {
145-
return (T) requests.poll(1, TimeUnit.SECONDS);
144+
ReadRowsRequest popLastRequest() throws InterruptedException {
145+
return requests.poll(1, TimeUnit.SECONDS);
146146
}
147147

148148
@Override

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/test_helpers/env/EmulatorEnv.java

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,12 @@
2323
import com.google.cloud.bigtable.data.v2.BigtableDataClient;
2424
import com.google.cloud.bigtable.data.v2.BigtableDataSettings;
2525
import com.google.cloud.bigtable.emulator.v2.Emulator;
26+
import com.google.common.base.Strings;
27+
import java.nio.file.Paths;
2628

2729
public class EmulatorEnv extends AbstractTestEnv {
30+
private static final String EMULATOR_OVERRIDE_PROPERTY_NAME = "bigtable.emulator-path";
31+
2832
private static final String PROJECT_ID = "fake-project";
2933
private static final String INSTANCE_ID = "fake-instance";
3034
private static final String TABLE_ID = "default-table";
@@ -44,7 +48,12 @@ private EmulatorEnv() {}
4448

4549
@Override
4650
void start() throws Exception {
47-
emulator = Emulator.createBundled();
51+
String overridePath = System.getProperty(EMULATOR_OVERRIDE_PROPERTY_NAME);
52+
if (!Strings.isNullOrEmpty(overridePath)) {
53+
emulator = Emulator.createFromPath(Paths.get(overridePath));
54+
} else {
55+
emulator = Emulator.createBundled();
56+
}
4857
emulator.start();
4958

5059
dataSettings =

0 commit comments

Comments
 (0)