Skip to content

Commit c5c881b

Browse files
fix(test): Clean up tests (#439)
* fix(test): Clean up tests * Add missing RunWith annotations * unnecessary downcasts * Make emulator path overridable * add missing annotation * lint
1 parent fb7bc5c commit c5c881b

File tree

10 files changed

+40
-7
lines changed

10 files changed

+40
-7
lines changed

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

+3-1
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

+3
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

+6-2
Original file line numberDiff line numberDiff line change
@@ -71,18 +71,22 @@
7171
import java.util.concurrent.ExecutionException;
7272
import java.util.concurrent.atomic.AtomicBoolean;
7373
import org.junit.Before;
74+
import org.junit.Rule;
7475
import org.junit.Test;
7576
import org.junit.runner.RunWith;
77+
import org.junit.runners.JUnit4;
7678
import org.mockito.Matchers;
7779
import org.mockito.Mock;
7880
import org.mockito.Mockito;
7981
import org.mockito.invocation.InvocationOnMock;
80-
import org.mockito.runners.MockitoJUnitRunner;
82+
import org.mockito.junit.MockitoJUnit;
83+
import org.mockito.junit.MockitoRule;
8184
import org.mockito.stubbing.Answer;
8285
import org.threeten.bp.Instant;
8386

84-
@RunWith(MockitoJUnitRunner.class)
87+
@RunWith(JUnit4.class)
8588
public class BigtableTableAdminClientTest {
89+
@Rule public final MockitoRule mockitoRule = MockitoJUnit.rule();
8690

8791
private static final String PROJECT_ID = "my-project";
8892
private static final String INSTANCE_ID = "my-instance";

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

+3
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

+3
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

+3
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

+3
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

+3
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

+3-3
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

+10-1
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,12 @@
2222
import com.google.cloud.bigtable.data.v2.BigtableDataClient;
2323
import com.google.cloud.bigtable.data.v2.BigtableDataSettings;
2424
import com.google.cloud.bigtable.emulator.v2.Emulator;
25+
import com.google.common.base.Strings;
26+
import java.nio.file.Paths;
2527

2628
public class EmulatorEnv extends AbstractTestEnv {
29+
private static final String EMULATOR_OVERRIDE_PROPERTY_NAME = "bigtable.emulator-path";
30+
2731
private static final String PROJECT_ID = "fake-project";
2832
private static final String INSTANCE_ID = "fake-instance";
2933
private static final String TABLE_ID = "default-table";
@@ -42,7 +46,12 @@ private EmulatorEnv() {}
4246

4347
@Override
4448
void start() throws Exception {
45-
emulator = Emulator.createBundled();
49+
String overridePath = System.getProperty(EMULATOR_OVERRIDE_PROPERTY_NAME);
50+
if (!Strings.isNullOrEmpty(overridePath)) {
51+
emulator = Emulator.createFromPath(Paths.get(overridePath));
52+
} else {
53+
emulator = Emulator.createBundled();
54+
}
4655
emulator.start();
4756

4857
dataSettings =

0 commit comments

Comments
 (0)