zydronium/google-nomulus
1
0
Fork 0
mirror of https://github.com/google/nomulus.git synced 2025-05-20 19:29:35 +02:00
Code Issues Projects Releases Packages Wiki Activity

Use smaller shard size in ClaimsListShardTest

Browse source 
The default production value of 10,000 was unnecessarily large for testing
purposes.

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=132441792
This commit is contained in:
mcilwain 2016-09-07 08:58:15 -07:00 • committed by Ben McIlwain
parent 4652688585
commit cadf9d4af2
9 changed files with 17 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
javatests/google/registry/model/tmch/ClaimsListShardTest.java
View file

@ -26,10 +26,12 @@ import google.registry.model.tmch.ClaimsListShard.ClaimsListRevision;
import google.registry.model.tmch.ClaimsListShard.UnshardedSaveException;
import google.registry.testing.AppEngineRule;
import google.registry.testing.ExceptionRule;
import google.registry.testing.InjectRule;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.joda.time.DateTime;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
@ -47,7 +49,13 @@ public class ClaimsListShardTest {
@Rule
public final ExceptionRule thrown = new ExceptionRule();
protected final DateTime now = DateTime.now(UTC);
@Rule
public final InjectRule inject = new InjectRule();
@Before
public void before() throws Exception {
inject.setStaticField(ClaimsListShard.class, "shardSize", 10);
}
@Test
public void test_unshardedSaveFails() throws Exception {
@ -73,9 +81,10 @@ public class ClaimsListShardTest {
public void test_savesAndGets_withSharding() throws Exception {
// Create a ClaimsList that will need 4 shards to save.
Map<String, String> labelsToKeys = new HashMap<>();
for (int i = 0; i <= ClaimsListShard.SHARD_SIZE * 3; i++) {
for (int i = 0; i <= ClaimsListShard.shardSize * 3; i++) {
labelsToKeys.put(Integer.toString(i), Integer.toString(i));
}
DateTime now = DateTime.now(UTC);
// Save it with sharding, and make sure that reloading it works.
ClaimsListShard unsharded = ClaimsListShard.create(now, ImmutableMap.copyOf(labelsToKeys));
unsharded.save();
@ -88,7 +97,7 @@ public class ClaimsListShardTest {
// Create a smaller ClaimsList that will need only 2 shards to save.
labelsToKeys = new HashMap<>();
for (int i = 0; i <= ClaimsListShard.SHARD_SIZE; i++) {
for (int i = 0; i <= ClaimsListShard.shardSize; i++) {
labelsToKeys.put(Integer.toString(i), Integer.toString(i));
}
unsharded = ClaimsListShard.create(now.plusDays(1), ImmutableMap.copyOf(labelsToKeys));