diff --git a/core/src/test/java/google/registry/rdap/RdapDomainSearchActionTest.java b/core/src/test/java/google/registry/rdap/RdapDomainSearchActionTest.java index e727bb8c84a..0885d461c87 100644 --- a/core/src/test/java/google/registry/rdap/RdapDomainSearchActionTest.java +++ b/core/src/test/java/google/registry/rdap/RdapDomainSearchActionTest.java @@ -283,15 +283,15 @@ void beforeEach() { private JsonObject generateExpectedJsonForTwoDomainsNsReply() { return jsonFileBuilder() - .addDomain("cat.example", "F-EXAMPLE") - .addDomain("cat.lol", "6-LOL") + .addDomain("cat.example", domainCatExample.getRepoId()) + .addDomain("cat.lol", domainCatLol.getRepoId()) .load("rdap_domains_two.json"); } private JsonObject generateExpectedJsonForTwoDomainsCatStarReplySql() { return jsonFileBuilder() - .addDomain("cat.lol", "6-LOL") - .addDomain("cat2.lol", "B-LOL") + .addDomain("cat.lol", domainCatLol.getRepoId()) + .addDomain("cat2.lol", domainCatLol2.getRepoId()) .load("rdap_domains_two.json"); } @@ -306,7 +306,7 @@ private void deleteCatLol() { minusMonths(clock.now(), 6))); } - private void createManyDomainsAndHosts( + private ImmutableList createManyDomainsAndHosts( int numActiveDomains, int numTotalDomainsPerActiveDomain, int numHosts) { ImmutableSet.Builder> hostKeysBuilder = new ImmutableSet.Builder<>(); ImmutableSet.Builder subordinateHostnamesBuilder = new ImmutableSet.Builder<>(); @@ -340,7 +340,7 @@ private void createManyDomainsAndHosts( } domainsBuilder.add(builder.build()); } - persistResources(domainsBuilder.build()); + return persistResources(domainsBuilder.build()); } private void checkNumberOfDomainsInResult(JsonObject obj, int expected) { @@ -353,10 +353,10 @@ private void runSuccessfulTestWithCatLol( requestType, queryString, jsonFileBuilder() - .addDomain("cat.lol", "6-LOL") + .addDomain("cat.lol", domainCatLol.getRepoId()) .addRegistrar("Yes Virginia