Skip to content

Commit e8bc0ca

Browse files
garabaisDagger Team
authored andcommitted
Refactor some Dagger tests to improve code clarity and modernize exception testing.
This change updates specific tests in the Dagger project to: - Prefer `assertThrows` to `ExpectedException`. - Refactor test cases for better readability by extracting variables. - Remove unnecessary warnings suppressions. RELNOTES=Refactor some Dagger tests for code clarity and modernize exception testing. PiperOrigin-RevId: 905251628
1 parent f15088d commit e8bc0ca

2 files changed

Lines changed: 20 additions & 29 deletions

File tree

dagger-runtime/test/javatests/dagger/internal/MapProviderFactoryTest.java

Lines changed: 16 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -17,35 +17,33 @@
1717
package dagger.internal;
1818

1919
import static com.google.common.truth.Truth.assertThat;
20+
import static org.junit.Assert.assertThrows;
2021

2122
import java.util.LinkedHashMap;
2223
import java.util.Map;
2324
import java.util.concurrent.atomic.AtomicInteger;
24-
import org.junit.Rule;
2525
import org.junit.Test;
26-
import org.junit.rules.ExpectedException;
2726
import org.junit.runner.RunWith;
2827
import org.junit.runners.JUnit4;
2928

3029
@RunWith(JUnit4.class)
31-
@SuppressWarnings("unchecked")
3230
public class MapProviderFactoryTest {
33-
@Rule
34-
public ExpectedException thrown = ExpectedException.none();
3531

3632
@Test
3733
public void nullKey() {
38-
thrown.expect(NullPointerException.class);
39-
MapProviderFactory.<String, Integer>builder(1).put(null, incrementingIntegerProvider(1));
34+
MapProviderFactory.Builder<String, Integer> builder =
35+
MapProviderFactory.<String, Integer>builder(1);
36+
Provider<Integer> providerOfValue = incrementingIntegerProvider(1);
37+
assertThrows(NullPointerException.class, () -> builder.put(null, providerOfValue));
4038
}
4139

4240
@Test
4341
public void nullValue() {
44-
thrown.expect(NullPointerException.class);
45-
MapProviderFactory.<String, Integer>builder(1).put("Hello", null);
42+
MapProviderFactory.Builder<String, Integer> builder =
43+
MapProviderFactory.<String, Integer>builder(1);
44+
assertThrows(NullPointerException.class, () -> builder.put("Hello", null));
4645
}
4746

48-
4947
@Test
5048
public void iterationOrder() {
5149
Provider<Integer> p1 = incrementingIntegerProvider(10);
@@ -54,14 +52,14 @@ public void iterationOrder() {
5452
Provider<Integer> p4 = incrementingIntegerProvider(40);
5553
Provider<Integer> p5 = incrementingIntegerProvider(50);
5654

57-
Factory<Map<String, Provider<Integer>>> factory = MapProviderFactory
58-
.<String, Integer>builder(4)
59-
.put("two", p2)
60-
.put("one", p1)
61-
.put("three", p3)
62-
.put("one", p5)
63-
.put("four", p4)
64-
.build();
55+
Factory<Map<String, Provider<Integer>>> factory =
56+
MapProviderFactory.<String, Integer>builder(4)
57+
.put("two", p2)
58+
.put("one", p1)
59+
.put("three", p3)
60+
.put("one", p5)
61+
.put("four", p4)
62+
.build();
6563

6664
Map<String, Provider<Integer>> expectedMap = new LinkedHashMap<>();
6765
expectedMap.put("two", p2);
@@ -74,7 +72,6 @@ public void iterationOrder() {
7472
.inOrder();
7573
}
7674

77-
7875
private static Provider<Integer> incrementingIntegerProvider(int seed) {
7976
return new AtomicInteger(seed)::getAndIncrement;
8077
}

dagger-runtime/test/javatests/dagger/internal/SetFactoryTest.java

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -17,46 +17,40 @@
1717
package dagger.internal;
1818

1919
import static com.google.common.truth.Truth.assertThat;
20+
import static org.junit.Assert.assertThrows;
2021

2122
import com.google.common.collect.ImmutableSet;
2223
import java.util.Arrays;
2324
import java.util.LinkedHashSet;
2425
import java.util.Set;
2526
import java.util.concurrent.atomic.AtomicInteger;
26-
import org.junit.Rule;
2727
import org.junit.Test;
28-
import org.junit.rules.ExpectedException;
2928
import org.junit.runner.RunWith;
3029
import org.junit.runners.JUnit4;
3130

3231
@RunWith(JUnit4.class)
33-
@SuppressWarnings("unchecked")
3432
public class SetFactoryTest {
35-
@Rule public ExpectedException thrown = ExpectedException.none();
3633

3734
@Test
3835
public void providerReturnsNull() {
3936
Factory<Set<Integer>> factory =
4037
SetFactory.<Integer>builder(0, 1).addCollectionProvider(() -> null).build();
41-
thrown.expect(NullPointerException.class);
42-
factory.get();
38+
assertThrows(NullPointerException.class, () -> factory.get());
4339
}
4440

4541
@Test
4642
public void providerReturnsNullSet() {
4743
Factory<Set<Integer>> factory =
4844
SetFactory.<Integer>builder(1, 0).addProvider(() -> null).build();
49-
thrown.expect(NullPointerException.class);
50-
factory.get();
45+
assertThrows(NullPointerException.class, () -> factory.get());
5146
}
5247

5348
@Test
5449
public void providerReturnsSetWithNullElement() {
5550
Set<Integer> set = new LinkedHashSet<>(Arrays.asList(1, null, 3));
5651
Factory<Set<Integer>> factory =
5752
SetFactory.<Integer>builder(0, 1).addCollectionProvider(() -> set).build();
58-
thrown.expect(NullPointerException.class);
59-
factory.get();
53+
assertThrows(NullPointerException.class, () -> factory.get());
6054
}
6155

6256
@Test

0 commit comments

Comments
 (0)