diff --git a/javatests/google/registry/rde/BUILD b/javatests/google/registry/rde/BUILD index b1853744b..80dd899a9 100644 --- a/javatests/google/registry/rde/BUILD +++ b/javatests/google/registry/rde/BUILD @@ -54,7 +54,7 @@ java_library( GenTestRules( name = "GeneratedTestRules", default_test_size = "medium", - shard_count = 3, + shard_count = 4, test_files = glob(["*Test.java"]), deps = [":rde"], ) diff --git a/javatests/google/registry/rde/BrdaCopyActionTest.java b/javatests/google/registry/rde/BrdaCopyActionTest.java index f53e5116f..c338fd46c 100644 --- a/javatests/google/registry/rde/BrdaCopyActionTest.java +++ b/javatests/google/registry/rde/BrdaCopyActionTest.java @@ -35,6 +35,7 @@ import google.registry.testing.BouncyCastleProviderRule; import google.registry.testing.GcsTestingUtils; import google.registry.testing.GpgSystemCommandRule; import google.registry.testing.Providers; +import google.registry.testing.ShardableTestCase; import java.io.File; import java.io.FileNotFoundException; import java.io.IOException; @@ -53,7 +54,7 @@ import org.junit.runners.JUnit4; /** Unit tests for {@link BrdaCopyAction}. */ @RunWith(JUnit4.class) -public class BrdaCopyActionTest { +public class BrdaCopyActionTest extends ShardableTestCase { private static final ByteSource DEPOSIT_XML = RdeTestData.get("deposit_full.xml"); // 2010-10-17 diff --git a/javatests/google/registry/rde/RdeMarshallerTest.java b/javatests/google/registry/rde/RdeMarshallerTest.java index c0ac45af5..510408ea3 100644 --- a/javatests/google/registry/rde/RdeMarshallerTest.java +++ b/javatests/google/registry/rde/RdeMarshallerTest.java @@ -18,6 +18,7 @@ import static com.google.common.truth.Truth.assertThat; import google.registry.model.registrar.Registrar; import google.registry.testing.AppEngineRule; +import google.registry.testing.ShardableTestCase; import google.registry.xml.XmlTestUtils; import org.joda.time.DateTime; import org.joda.time.format.ISODateTimeFormat; @@ -28,7 +29,7 @@ import org.junit.runners.JUnit4; /** Unit tests for {@link RdeMarshaller}. */ @RunWith(JUnit4.class) -public class RdeMarshallerTest { +public class RdeMarshallerTest extends ShardableTestCase { private static final String DECLARATION = "\n";