Remove PremiumList from Datastore schema (#1256)

* Remove PremiumList from Datastore schema

* Remove commented out code

* Change lastUpdateTime to creationTimestamp

* Remove extra file

* Remove currency unit from input data to parse

* Revert extra file

* Check currency in parse

* Create all PremiumEntries before saving them in bulk

* small fixes

* Fix merge conflict
This commit is contained in:
sarahcaseybot 2021-08-10 13:26:13 -04:00 committed by GitHub
parent 7d907870ec
commit f82fe3b182
58 changed files with 399 additions and 713 deletions

4
.gitignore vendored
View file

@ -112,3 +112,7 @@ core/**/registrar_bin*.js
core/**/registrar_dbg*.js core/**/registrar_dbg*.js
core/**/registrar_bin*.css core/**/registrar_bin*.css
core/**/registrar_dbg*.css core/**/registrar_dbg*.css
# Appengine generated files
core/WEB-INF/appengine-generated/*.bin
core/WEB-INF/appengine-generated/*.xml

View file

@ -31,7 +31,7 @@ import com.google.common.flogger.FluentLogger;
import com.google.common.net.MediaType; import com.google.common.net.MediaType;
import google.registry.config.RegistryConfig.Config; import google.registry.config.RegistryConfig.Config;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.request.Action; import google.registry.request.Action;
import google.registry.request.Parameter; import google.registry.request.Parameter;
import google.registry.request.RequestParameters; import google.registry.request.RequestParameters;
@ -143,8 +143,8 @@ public class ExportPremiumTermsAction implements Runnable {
PremiumListDao.getLatestRevision(premiumListName).isPresent(), PremiumListDao.getLatestRevision(premiumListName).isPresent(),
"Could not load premium list for " + tld); "Could not load premium list for " + tld);
SortedSet<String> premiumTerms = SortedSet<String> premiumTerms =
Streams.stream(PremiumListDao.loadAllPremiumListEntries(premiumListName)) Streams.stream(PremiumListDao.loadAllPremiumEntries(premiumListName))
.map(PremiumListEntry::toString) .map(PremiumEntry::toString)
.collect(ImmutableSortedSet.toImmutableSortedSet(String::compareTo)); .collect(ImmutableSortedSet.toImmutableSortedSet(String::compareTo));
return Joiner.on("\n") return Joiner.on("\n")

View file

@ -48,7 +48,7 @@ public final class ExportUtils {
String.format("Reserved list %s does not exist", reservedListName))); String.format("Reserved list %s does not exist", reservedListName)));
if (reservedList.getShouldPublish()) { if (reservedList.getShouldPublish()) {
for (ReservedListEntry entry : reservedList.getReservedListEntries().values()) { for (ReservedListEntry entry : reservedList.getReservedListEntries().values()) {
reservedTerms.add(entry.getLabel()); reservedTerms.add(entry.getDomainLabel());
} }
} }
} }

View file

@ -40,7 +40,6 @@ import google.registry.model.rde.RdeRevision;
import google.registry.model.registrar.Registrar; import google.registry.model.registrar.Registrar;
import google.registry.model.registrar.RegistrarContact; import google.registry.model.registrar.RegistrarContact;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.ReservedList; import google.registry.model.registry.label.ReservedList;
import google.registry.model.reporting.HistoryEntry; import google.registry.model.reporting.HistoryEntry;
import google.registry.model.server.Lock; import google.registry.model.server.Lock;
@ -91,9 +90,6 @@ public final class EntityClasses {
PollMessage.class, PollMessage.class,
PollMessage.Autorenew.class, PollMessage.Autorenew.class,
PollMessage.OneTime.class, PollMessage.OneTime.class,
PremiumList.class,
PremiumList.PremiumListEntry.class,
PremiumList.PremiumListRevision.class,
RdeRevision.class, RdeRevision.class,
Registrar.class, Registrar.class,
RegistrarContact.class, RegistrarContact.class,

View file

@ -74,14 +74,12 @@ public abstract class BaseDomainLabelList<T extends Comparable<?>, R extends Dom
@Parent @Transient Key<EntityGroupRoot> parent = getCrossTldKey(); @Parent @Transient Key<EntityGroupRoot> parent = getCrossTldKey();
@Transient DateTime creationTime;
// The list in Cloud SQL is immutable, we only have a creation_timestamp field and it should be // The list in Cloud SQL is immutable, we only have a creation_timestamp field and it should be
// set to the timestamp when the list is created. In Datastore, we have two fields and the // set to the timestamp when the list is created. In Datastore, we have two fields and the
// lastUpdateTime is set to the current timestamp when creating and updating a list. So, we use // lastUpdateTime is set to the current timestamp when creating and updating a list. So, we use
// lastUpdateTime as the creation_timestamp column during the dual-write phase for compatibility. // lastUpdateTime as the creation_timestamp column during the dual-write phase for compatibility.
@Column(name = "creation_timestamp") @Column(name = "creation_timestamp")
DateTime lastUpdateTime; DateTime creationTimestamp;
/** Returns the ID of this revision, or throws if null. */ /** Returns the ID of this revision, or throws if null. */
public long getRevisionId() { public long getRevisionId() {
@ -97,12 +95,8 @@ public abstract class BaseDomainLabelList<T extends Comparable<?>, R extends Dom
} }
/** Returns the creation time of this revision of the reserved list. */ /** Returns the creation time of this revision of the reserved list. */
public DateTime getCreationTime() { public DateTime getCreationTimestamp() {
return creationTime; return creationTimestamp;
}
public DateTime getLastUpdateTime() {
return lastUpdateTime;
} }
/** /**
@ -118,7 +112,7 @@ public abstract class BaseDomainLabelList<T extends Comparable<?>, R extends Dom
if (entry == null) { if (entry == null) {
continue; continue;
} }
String label = entry.getLabel(); String label = entry.getDomainLabel();
// Check if the label was already processed for this list (which is an error), and if so, // Check if the label was already processed for this list (which is an error), and if so,
// accumulate it so that a list of all duplicates can be thrown. // accumulate it so that a list of all duplicates can be thrown.
if (labelsToEntries.containsKey(label)) { if (labelsToEntries.containsKey(label)) {
@ -196,13 +190,8 @@ public abstract class BaseDomainLabelList<T extends Comparable<?>, R extends Dom
return thisCastToDerived(); return thisCastToDerived();
} }
public B setCreationTime(DateTime creationTime) { public B setCreationTimestamp(DateTime creationTime) {
getInstance().creationTime = creationTime; getInstance().creationTimestamp = creationTime;
return thisCastToDerived();
}
public B setLastUpdateTime(DateTime lastUpdateTime) {
getInstance().lastUpdateTime = lastUpdateTime;
return thisCastToDerived(); return thisCastToDerived();
} }

View file

@ -37,17 +37,15 @@ public abstract class DomainLabelEntry<T extends Comparable<?>, D extends Domain
@Id @Id
@javax.persistence.Id @javax.persistence.Id
@Column(name = "domain_label", nullable = false) @Column(name = "domainLabel", nullable = false)
String label; String domainLabel;
String comment;
/** /**
* Returns the label of the field, which also happens to be used as the key for the Map object * Returns the label of the field, which also happens to be used as the key for the Map object
* that is serialized from Datastore. * that is serialized from Datastore.
*/ */
public String getLabel() { public String getDomainLabel() {
return label; return domainLabel;
} }
/** /**
@ -72,30 +70,25 @@ public abstract class DomainLabelEntry<T extends Comparable<?>, D extends Domain
} }
public B setLabel(String label) { public B setLabel(String label) {
getInstance().label = label; getInstance().domainLabel = label;
return thisCastToDerived();
}
public B setComment(String comment) {
getInstance().comment = comment;
return thisCastToDerived(); return thisCastToDerived();
} }
@Override @Override
public T build() { public T build() {
checkArgumentNotNull(emptyToNull(getInstance().label), "Label must be specified"); checkArgumentNotNull(emptyToNull(getInstance().domainLabel), "Label must be specified");
checkArgument( checkArgument(
getInstance().label.equals(canonicalizeDomainName(getInstance().label)), getInstance().domainLabel.equals(canonicalizeDomainName(getInstance().domainLabel)),
"Label '%s' must be in puny-coded, lower-case form", "Label '%s' must be in puny-coded, lower-case form",
getInstance().label); getInstance().domainLabel);
checkArgumentNotNull(getInstance().getValue(), "Value must be specified"); checkArgumentNotNull(getInstance().getValue(), "Value must be specified");
// Verify that the label creates a valid SLD if we add a TLD to the end of it. // Verify that the label creates a valid SLD if we add a TLD to the end of it.
// We require that the label is not already a full domain name including a dot. // We require that the label is not already a full domain name including a dot.
// Domain name validation is tricky, so let InternetDomainName handle it for us. // Domain name validation is tricky, so let InternetDomainName handle it for us.
checkArgument( checkArgument(
InternetDomainName.from(getInstance().label + ".tld").parts().size() == 2, InternetDomainName.from(getInstance().domainLabel + ".tld").parts().size() == 2,
"Label %s must not be a multi-level domain name", "Label %s must not be a multi-level domain name",
getInstance().label); getInstance().domainLabel);
return super.build(); return super.build();
} }
} }

View file

@ -18,45 +18,27 @@ import static com.google.common.base.Charsets.US_ASCII;
import static com.google.common.base.Preconditions.checkArgument; import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.collect.ImmutableMap.toImmutableMap; import static com.google.common.collect.ImmutableMap.toImmutableMap;
import static com.google.common.hash.Funnels.stringFunnel; import static com.google.common.hash.Funnels.stringFunnel;
import static com.google.common.hash.Funnels.unencodedCharsFunnel;
import static google.registry.model.IdService.allocateId;
import static google.registry.persistence.transaction.QueryComposer.Comparator.EQ;
import static google.registry.persistence.transaction.TransactionManagerFactory.jpaTm;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Splitter; import com.google.common.base.Splitter;
import com.google.common.collect.ImmutableMap; import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Streams;
import com.google.common.hash.BloomFilter; import com.google.common.hash.BloomFilter;
import com.googlecode.objectify.Key;
import com.googlecode.objectify.annotation.Entity;
import com.googlecode.objectify.annotation.Id;
import com.googlecode.objectify.annotation.Ignore;
import com.googlecode.objectify.annotation.Parent;
import google.registry.model.Buildable; import google.registry.model.Buildable;
import google.registry.model.ImmutableObject; import google.registry.model.ImmutableObject;
import google.registry.model.annotations.InCrossTld;
import google.registry.model.annotations.ReportedOn; import google.registry.model.annotations.ReportedOn;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.schema.replay.DatastoreOnlyEntity; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.schema.replay.NonReplicatedEntity; import google.registry.schema.replay.SqlOnlyEntity;
import google.registry.schema.tld.PremiumEntry;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import java.io.ByteArrayOutputStream; import java.io.Serializable;
import java.io.IOException;
import java.math.BigDecimal; import java.math.BigDecimal;
import java.math.RoundingMode; import java.math.RoundingMode;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Objects; import java.util.Objects;
import java.util.Set;
import javax.annotation.Nullable; import javax.annotation.Nullable;
import javax.persistence.Column; import javax.persistence.Column;
import javax.persistence.Index; import javax.persistence.Index;
import javax.persistence.PostLoad;
import javax.persistence.PostPersist;
import javax.persistence.PostUpdate;
import javax.persistence.PrePersist;
import javax.persistence.PreRemove;
import javax.persistence.Table; import javax.persistence.Table;
import javax.persistence.Transient; import javax.persistence.Transient;
import org.joda.money.CurrencyUnit; import org.joda.money.CurrencyUnit;
@ -71,16 +53,11 @@ import org.joda.money.Money;
* This is fine though, because we only use the list with the highest revisionId. * This is fine though, because we only use the list with the highest revisionId.
*/ */
@ReportedOn @ReportedOn
@Entity
@javax.persistence.Entity @javax.persistence.Entity
@Table(indexes = {@Index(columnList = "name", name = "premiumlist_name_idx")}) @Table(indexes = {@Index(columnList = "name", name = "premiumlist_name_idx")})
public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.PremiumListEntry> public final class PremiumList extends BaseDomainLabelList<BigDecimal, PremiumEntry>
implements NonReplicatedEntity { implements SqlOnlyEntity {
/** Stores the revision key for the set of currently used premium list entry entities. */
@Transient Key<PremiumListRevision> revisionKey;
@Ignore
@Column(nullable = false) @Column(nullable = false)
CurrencyUnit currency; CurrencyUnit currency;
@ -91,80 +68,11 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
* from the immutability contract so we can modify it after construction and we have to handle the * from the immutability contract so we can modify it after construction and we have to handle the
* database processing on our own so we can detach it after load. * database processing on our own so we can detach it after load.
*/ */
@Ignore @ImmutableObject.Insignificant @Transient ImmutableMap<String, BigDecimal> labelsToPrices; @ImmutableObject.Insignificant @Transient ImmutableMap<String, BigDecimal> labelsToPrices;
@Ignore
@Column(nullable = false) @Column(nullable = false)
BloomFilter<String> bloomFilter; BloomFilter<String> bloomFilter;
/** Virtual parent entity for premium list entry entities associated with a single revision. */
@ReportedOn
@Entity
@InCrossTld
public static class PremiumListRevision extends ImmutableObject implements DatastoreOnlyEntity {
@Parent Key<PremiumList> parent;
@Id long revisionId;
/**
* A Bloom filter that is used to determine efficiently and quickly whether a label might be
* premium.
*
* <p>If the label might be premium, then the premium list entry must be loaded by key and
* checked for existence. Otherwise, we know it's not premium, and no Datastore load is
* required.
*/
private BloomFilter<String> probablePremiumLabels;
/**
* Get the Bloom filter.
*
* <p>Note that this is not a copy, but the mutable object itself, because copying would be
* expensive. You probably should not modify the filter unless you know what you're doing.
*/
public BloomFilter<String> getProbablePremiumLabels() {
return probablePremiumLabels;
}
/**
* The maximum size of the Bloom filter.
*
* <p>Trying to set it any larger will throw an error, as we know it won't fit into a Datastore
* entity. We use 90% of the 1 MB Datastore limit to leave some wriggle room for the other
* fields and miscellaneous entity serialization overhead.
*/
private static final int MAX_BLOOM_FILTER_BYTES = 900000;
/** Returns a new PremiumListRevision for the given key and premium list map. */
@VisibleForTesting
public static PremiumListRevision create(PremiumList parent, Set<String> premiumLabels) {
PremiumListRevision revision = new PremiumListRevision();
revision.parent = Key.create(parent);
revision.revisionId = allocateId();
// All premium list labels are already punycoded, so don't perform any further character
// encoding on them.
revision.probablePremiumLabels =
BloomFilter.create(unencodedCharsFunnel(), premiumLabels.size());
premiumLabels.forEach(revision.probablePremiumLabels::put);
try {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
revision.probablePremiumLabels.writeTo(bos);
checkArgument(
bos.size() <= MAX_BLOOM_FILTER_BYTES,
"Too many premium labels were specified; Bloom filter exceeds max entity size");
} catch (IOException e) {
throw new IllegalStateException("Could not serialize premium labels Bloom filter", e);
}
return revision;
}
}
@VisibleForTesting
public Key<PremiumListRevision> getRevisionKey() {
return revisionKey;
}
/** Returns the {@link CurrencyUnit} used for this list. */ /** Returns the {@link CurrencyUnit} used for this list. */
public CurrencyUnit getCurrency() { public CurrencyUnit getCurrency() {
return currency; return currency;
@ -179,16 +87,14 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
*/ */
public synchronized ImmutableMap<String, BigDecimal> getLabelsToPrices() { public synchronized ImmutableMap<String, BigDecimal> getLabelsToPrices() {
if (labelsToPrices == null) { if (labelsToPrices == null) {
Iterable<PremiumEntry> entries = PremiumListDao.loadAllPremiumEntries(name);
labelsToPrices = labelsToPrices =
jpaTm() Streams.stream(entries)
.createQueryComposer(PremiumEntry.class)
.where("revisionId", EQ, revisionId)
.stream()
.collect( .collect(
toImmutableMap( toImmutableMap(
PremiumEntry::getDomainLabel, PremiumEntry::getDomainLabel,
// Set the correct amount of precision for the premium list's currency. // Set the correct amount of precision for the premium list's currency.
entry -> convertAmountToMoney(entry.getPrice()).getAmount())); premiumEntry -> convertAmountToMoney(premiumEntry.getValue()).getAmount()));
} }
return labelsToPrices; return labelsToPrices;
} }
@ -212,29 +118,38 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
} }
/** /**
* A premium list entry entity, persisted to Datastore. Each instance represents the price of a * A premium list entry entity, persisted to Cloud SQL. Each instance represents the price of a
* single label on a given TLD. * single label on a given TLD.
*/ */
@ReportedOn @javax.persistence.Entity(name = "PremiumEntry")
@Entity public static class PremiumEntry extends DomainLabelEntry<BigDecimal, PremiumList.PremiumEntry>
@InCrossTld implements Buildable, SqlOnlyEntity, Serializable {
public static class PremiumListEntry extends DomainLabelEntry<Money, PremiumListEntry>
implements Buildable, DatastoreOnlyEntity {
@Parent @ImmutableObject.Insignificant @javax.persistence.Id Long revisionId;
Key<PremiumListRevision> parent;
Money price; @Column(nullable = false)
BigDecimal price;
@Override @Override
public Money getValue() { public BigDecimal getValue() {
return price; return price;
} }
public static PremiumEntry create(Long revisionId, BigDecimal price, String label) {
return new PremiumEntry.Builder()
.setRevisionId(revisionId)
.setPrice(price)
.setLabel(label)
.build();
}
@Override @Override
public String toString() { public String toString() {
// Don't include the comment so that we can use this when exporting the premium list return String.format("%s, %s", domainLabel, price);
return String.format("%s,%s", label, price); }
public String toString(CurrencyUnit currencyUnit) {
return String.format("%s,%s %s", domainLabel, currencyUnit, price);
} }
@Override @Override
@ -242,22 +157,22 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
return new Builder(clone(this)); return new Builder(clone(this));
} }
/** A builder for constructing {@link PremiumListEntry} objects, since they are immutable. */ /** A builder for constructing {@link PremiumEntry} objects, since they are immutable. */
public static class Builder extends DomainLabelEntry.Builder<PremiumListEntry, Builder> { public static class Builder extends DomainLabelEntry.Builder<PremiumEntry, Builder> {
public Builder() {} public Builder() {}
private Builder(PremiumListEntry instance) { private Builder(PremiumEntry instance) {
super(instance); super(instance);
} }
public Builder setParent(Key<PremiumListRevision> parentKey) { public Builder setPrice(BigDecimal price) {
getInstance().parent = parentKey; getInstance().price = price;
return this; return this;
} }
public Builder setPrice(Money price) { public Builder setRevisionId(Long revisionId) {
getInstance().price = price; getInstance().revisionId = revisionId;
return this; return this;
} }
} }
@ -265,19 +180,29 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
@Override @Override
@Nullable @Nullable
PremiumListEntry createFromLine(String originalLine) { PremiumEntry createFromLine(String originalLine) {
List<String> lineAndComment = splitOnComment(originalLine); List<String> lineAndComment = splitOnComment(originalLine);
if (lineAndComment.isEmpty()) { if (lineAndComment.isEmpty()) {
return null; return null;
} }
String line = lineAndComment.get(0); String line = lineAndComment.get(0);
String comment = lineAndComment.get(1);
List<String> parts = Splitter.on(',').trimResults().splitToList(line); List<String> parts = Splitter.on(',').trimResults().splitToList(line);
checkArgument(parts.size() == 2, "Could not parse line in premium list: %s", originalLine); checkArgument(parts.size() == 2, "Could not parse line in premium list: %s", originalLine);
return new PremiumListEntry.Builder() List<String> moneyParts = Splitter.on(' ').trimResults().splitToList(parts.get(1));
if (moneyParts.size() == 2 && this.currency != null) {
if (!Money.parse(parts.get(1)).getCurrencyUnit().equals(this.currency)) {
throw new IllegalArgumentException(
String.format("The currency unit must be %s", this.currency.getCode()));
}
}
BigDecimal price =
moneyParts.size() == 2
? Money.parse(parts.get(1)).getAmount()
: new BigDecimal(parts.get(1));
return new PremiumEntry.Builder()
.setLabel(parts.get(0)) .setLabel(parts.get(0))
.setPrice(Money.parse(parts.get(1))) .setPrice(price)
.setComment(comment) .setRevisionId(revisionId)
.build(); .build();
} }
@ -300,11 +225,6 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
super(instance); super(instance);
} }
public Builder setRevision(Key<PremiumListRevision> revision) {
getInstance().revisionKey = revision;
return this;
}
public Builder setCurrency(CurrencyUnit currency) { public Builder setCurrency(CurrencyUnit currency) {
getInstance().currency = currency; getInstance().currency = currency;
return this; return this;
@ -330,42 +250,4 @@ public final class PremiumList extends BaseDomainLabelList<Money, PremiumList.Pr
return super.build(); return super.build();
} }
} }
@PrePersist
void prePersist() {
lastUpdateTime = creationTime;
}
@PostLoad
void postLoad() {
creationTime = lastUpdateTime;
}
@PreRemove
void preRemove() {
jpaTm()
.query("DELETE FROM PremiumEntry WHERE revision_id = :revisionId")
.setParameter("revisionId", revisionId)
.executeUpdate();
}
/**
* Hibernate hook called on the insert of a new PremiumList. Stores the associated {@link
* PremiumEntry}'s.
*
* <p>We need to persist the list entries, but only on the initial insert (not on update) since
* the entries themselves never get changed, so we only annotate it with {@link PostPersist}, not
* {@link PostUpdate}.
*/
@PostPersist
void postPersist() {
// If the price map is loaded, persist it too.
if (labelsToPrices != null) {
labelsToPrices.entrySet().stream()
.forEach(
entry ->
jpaTm()
.insert(PremiumEntry.create(revisionId, entry.getValue(), entry.getKey())));
}
}
} }

View file

@ -131,15 +131,21 @@ public final class ReservedList
@Column(nullable = false) @Column(nullable = false)
ReservationType reservationType; ReservationType reservationType;
String comment;
/** Mapper for use with @Mapify */ /** Mapper for use with @Mapify */
static class LabelMapper implements Mapper<String, ReservedListEntry> { static class LabelMapper implements Mapper<String, ReservedListEntry> {
@Override @Override
public String getKey(ReservedListEntry entry) { public String getKey(ReservedListEntry entry) {
return entry.getLabel(); return entry.getDomainLabel();
} }
} }
public String getComment(String comment) {
return comment;
}
/** Creates a {@link ReservedListEntry} from a label, reservation type, and optional comment. */ /** Creates a {@link ReservedListEntry} from a label, reservation type, and optional comment. */
public static ReservedListEntry create( public static ReservedListEntry create(
String label, ReservationType reservationType, @Nullable String comment) { String label, ReservationType reservationType, @Nullable String comment) {
@ -163,7 +169,7 @@ public final class ReservedList
@Override @Override
public String toString() { public String toString() {
return String.format( return String.format(
"%s,%s%s", label, reservationType, isNullOrEmpty(comment) ? "" : " # " + comment); "%s,%s%s", domainLabel, reservationType, isNullOrEmpty(comment) ? "" : " # " + comment);
} }
/** A builder for constructing {@link ReservedListEntry} objects, since they are immutable. */ /** A builder for constructing {@link ReservedListEntry} objects, since they are immutable. */
@ -180,6 +186,11 @@ public final class ReservedList
getInstance().reservationType = reservationType; getInstance().reservationType = reservationType;
return this; return this;
} }
ReservedListEntry.Builder setComment(String comment) {
getInstance().comment = comment;
return this;
}
} }
} }
@ -217,7 +228,7 @@ public final class ReservedList
.createQueryComposer(ReservedListEntry.class) .createQueryComposer(ReservedListEntry.class)
.where("revisionId", EQ, revisionId) .where("revisionId", EQ, revisionId)
.stream() .stream()
.collect(toImmutableMap(ReservedListEntry::getLabel, e -> e))); .collect(toImmutableMap(ReservedListEntry::getDomainLabel, e -> e)));
} }
return ImmutableMap.copyOf(nullToEmpty(reservedListMap)); return ImmutableMap.copyOf(nullToEmpty(reservedListMap));
} }

View file

@ -1,37 +0,0 @@
// Copyright 2020 The Nomulus Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package google.registry.persistence.converter;
import static google.registry.model.common.EntityGroupRoot.getCrossTldKey;
import com.googlecode.objectify.Key;
import google.registry.model.registry.label.PremiumList;
import javax.persistence.AttributeConverter;
import javax.persistence.Converter;
/** JPA converter for a {@link Key} containing a {@link PremiumList} */
@Converter(autoApply = true)
public class PremiumListKeyConverter implements AttributeConverter<Key<PremiumList>, String> {
@Override
public String convertToDatabaseColumn(Key<PremiumList> attribute) {
return (attribute == null) ? null : attribute.getName();
}
@Override
public Key<PremiumList> convertToEntityAttribute(String dbData) {
return (dbData == null) ? null : Key.create(getCrossTldKey(), PremiumList.class, dbData);
}
}

View file

@ -1,62 +0,0 @@
// Copyright 2019 The Nomulus Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package google.registry.schema.tld;
import google.registry.model.ImmutableObject;
import google.registry.model.registry.label.PremiumList;
import google.registry.schema.replay.SqlOnlyEntity;
import java.io.Serializable;
import java.math.BigDecimal;
import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.Id;
/**
* Entity class for the premium price of an individual domain label.
*
* <p>These are not persisted directly, but rather, using {@link PremiumList#getLabelsToPrices()}.
*/
@Entity
public class PremiumEntry extends ImmutableObject implements Serializable, SqlOnlyEntity {
@Id
@Column(nullable = false)
Long revisionId;
@Column(nullable = false)
BigDecimal price;
@Id
@Column(nullable = false)
String domainLabel;
private PremiumEntry() {}
public BigDecimal getPrice() {
return price;
}
public String getDomainLabel() {
return domainLabel;
}
public static PremiumEntry create(long revisionId, BigDecimal price, String domainLabel) {
PremiumEntry result = new PremiumEntry();
result.revisionId = revisionId;
result.price = price;
result.domainLabel = domainLabel;
return result;
}
}

View file

@ -26,15 +26,15 @@ import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader; import com.google.common.cache.CacheLoader;
import com.google.common.cache.CacheLoader.InvalidCacheLoadException; import com.google.common.cache.CacheLoader.InvalidCacheLoadException;
import com.google.common.cache.LoadingCache; import com.google.common.cache.LoadingCache;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Streams; import com.google.common.collect.Streams;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.util.NonFinalForTesting; import google.registry.util.NonFinalForTesting;
import java.math.BigDecimal; import java.math.BigDecimal;
import java.util.List; import java.util.List;
import java.util.Optional; import java.util.Optional;
import java.util.concurrent.ExecutionException; import java.util.concurrent.ExecutionException;
import org.joda.money.BigMoney;
import org.joda.money.CurrencyUnit; import org.joda.money.CurrencyUnit;
import org.joda.money.Money; import org.joda.money.Money;
import org.joda.time.Duration; import org.joda.time.Duration;
@ -153,18 +153,48 @@ public class PremiumListDao {
} }
} }
public static PremiumList save(String name, List<String> inputData) { public static PremiumList save(String name, CurrencyUnit currencyUnit, List<String> inputData) {
return save(PremiumListUtils.parseToPremiumList(name, inputData)); return save(PremiumListUtils.parseToPremiumList(name, currencyUnit, inputData));
} }
public static PremiumList save(PremiumList premiumList) { public static PremiumList save(PremiumList premiumList) {
jpaTm().transact(() -> jpaTm().insert(premiumList)); jpaTm().transact(() -> jpaTm().insert(premiumList));
premiumListCache.invalidate(premiumList.getName()); premiumListCache.invalidate(premiumList.getName());
jpaTm()
.transact(
() -> {
if (premiumList.getLabelsToPrices() != null) {
Optional<PremiumList> savedPremiumList =
PremiumListDao.getLatestRevision(premiumList.getName());
ImmutableSet.Builder<PremiumEntry> entries = new ImmutableSet.Builder<>();
premiumList.getLabelsToPrices().entrySet().stream()
.forEach(
entry ->
entries.add(
PremiumEntry.create(
savedPremiumList.get().getRevisionId(),
entry.getValue(),
entry.getKey())));
jpaTm().insertAll(entries.build());
}
});
premiumListCache.invalidate(premiumList.getName());
return premiumList; return premiumList;
} }
public static void delete(PremiumList premiumList) { public static void delete(PremiumList premiumList) {
jpaTm().transact(() -> getLatestRevision(premiumList.getName()).ifPresent(jpaTm()::delete)); jpaTm()
.transact(
() -> {
Optional<PremiumList> persistedList = getLatestRevision(premiumList.getName());
if (persistedList.isPresent()) {
jpaTm()
.query("DELETE FROM PremiumEntry WHERE revisionId = :revisionId")
.setParameter("revisionId", persistedList.get().getRevisionId())
.executeUpdate();
jpaTm().delete(persistedList.get());
}
});
premiumListCache.invalidate(premiumList.getName()); premiumListCache.invalidate(premiumList.getName());
} }
@ -183,11 +213,11 @@ public class PremiumListDao {
} }
/** /**
* Returns all {@link PremiumListEntry PremiumListEntries} in the given {@code premiumList}. * Returns all {@link PremiumEntry PremiumEntries} in the given {@code premiumList}.
* *
* <p>This is an expensive operation and should only be used when the entire list is required. * <p>This is an expensive operation and should only be used when the entire list is required.
*/ */
public static Iterable<PremiumEntry> loadPremiumListEntries(PremiumList premiumList) { public static Iterable<PremiumEntry> loadPremiumEntries(PremiumList premiumList) {
return jpaTm() return jpaTm()
.transact( .transact(
() -> () ->
@ -220,24 +250,25 @@ public class PremiumListDao {
} }
/** /**
* Returns all {@link PremiumListEntry PremiumListEntries} in the list with the given name. * Returns all {@link PremiumEntry PremiumEntries} in the list with the given name.
* *
* <p>This is an expensive operation and should only be used when the entire list is required. * <p>This is an expensive operation and should only be used when the entire list is required.
*/ */
public static Iterable<PremiumListEntry> loadAllPremiumListEntries(String premiumListName) { public static Iterable<PremiumEntry> loadAllPremiumEntries(String premiumListName) {
PremiumList premiumList = PremiumList premiumList =
getLatestRevision(premiumListName) getLatestRevision(premiumListName)
.orElseThrow( .orElseThrow(
() -> () ->
new IllegalArgumentException( new IllegalArgumentException(
String.format("No premium list with name %s.", premiumListName))); String.format("No premium list with name %s.", premiumListName)));
CurrencyUnit currencyUnit = premiumList.getCurrency(); Iterable<PremiumEntry> entries = loadPremiumEntries(premiumList);
return Streams.stream(loadPremiumListEntries(premiumList)) return Streams.stream(entries)
.map( .map(
premiumEntry -> premiumEntry ->
new PremiumListEntry.Builder() new PremiumEntry.Builder()
.setPrice(BigMoney.of(currencyUnit, premiumEntry.getPrice()).toMoney()) .setPrice(premiumEntry.getValue())
.setLabel(premiumEntry.getDomainLabel()) .setLabel(premiumEntry.getDomainLabel())
.setRevisionId(premiumList.getRevisionId())
.build()) .build())
.collect(toImmutableList()); .collect(toImmutableList());
} }

View file

@ -15,16 +15,12 @@
package google.registry.schema.tld; package google.registry.schema.tld;
import static com.google.common.base.Preconditions.checkArgument; import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.collect.ImmutableSet.toImmutableSet;
import static org.joda.time.DateTimeZone.UTC; import static org.joda.time.DateTimeZone.UTC;
import com.google.common.collect.ImmutableMap; import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSortedSet;
import com.google.common.collect.Iterables;
import com.google.common.collect.Maps; import com.google.common.collect.Maps;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import java.math.BigDecimal; import java.math.BigDecimal;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@ -34,28 +30,18 @@ import org.joda.time.DateTime;
/** Static utility methods for {@link PremiumList}. */ /** Static utility methods for {@link PremiumList}. */
public class PremiumListUtils { public class PremiumListUtils {
public static PremiumList parseToPremiumList(String name, List<String> inputData) { public static PremiumList parseToPremiumList(
ImmutableMap<String, PremiumListEntry> prices = String name, CurrencyUnit currencyUnit, List<String> inputData) {
new PremiumList.Builder().setName(name).build().parse(inputData); PremiumList partialPremiumList =
ImmutableSet<CurrencyUnit> currencies = new PremiumList.Builder()
prices.values().stream() .setName(name)
.map(e -> e.getValue().getCurrencyUnit()) .setCurrency(currencyUnit)
.distinct() .setCreationTimestamp(DateTime.now(UTC))
.collect(toImmutableSet()); .build();
checkArgument( ImmutableMap<String, PremiumEntry> prices = partialPremiumList.parse(inputData);
currencies.size() == 1, checkArgument(inputData.size() > 0, "Input cannot be empty");
"The Cloud SQL schema requires exactly one currency, but got: %s", Map<String, BigDecimal> priceAmounts = Maps.transformValues(prices, PremiumEntry::getValue);
ImmutableSortedSet.copyOf(currencies)); return partialPremiumList.asBuilder().setLabelsToPrices(priceAmounts).build();
CurrencyUnit currency = Iterables.getOnlyElement(currencies);
Map<String, BigDecimal> priceAmounts =
Maps.transformValues(prices, ple -> ple.getValue().getAmount());
return new PremiumList.Builder()
.setName(name)
.setCurrency(currency)
.setLabelsToPrices(priceAmounts)
.setCreationTime(DateTime.now(UTC))
.build();
} }
private PremiumListUtils() {} private PremiumListUtils() {}

View file

@ -21,15 +21,18 @@ import google.registry.tools.params.PathParameter;
import java.nio.file.Path; import java.nio.file.Path;
import java.util.List; import java.util.List;
import javax.annotation.Nullable; import javax.annotation.Nullable;
import org.joda.money.CurrencyUnit;
/** /**
* Base class for specification of command line parameters common to creating and updating premium * Base class for specification of command line parameters common to creating and updating premium
* lists. * lists.
*/ */
abstract class CreateOrUpdatePremiumListCommand extends MutatingCommand { abstract class CreateOrUpdatePremiumListCommand extends ConfirmingCommand
implements CommandWithRemoteApi {
private static final FluentLogger logger = FluentLogger.forEnclosingClass(); private static final FluentLogger logger = FluentLogger.forEnclosingClass();
protected List<String> inputData; protected List<String> inputData;
protected CurrencyUnit currency;
@Nullable @Nullable
@Parameter( @Parameter(
@ -51,7 +54,7 @@ abstract class CreateOrUpdatePremiumListCommand extends MutatingCommand {
String message = String.format("Saved premium list %s with %d entries", name, inputData.size()); String message = String.format("Saved premium list %s with %d entries", name, inputData.size());
try { try {
logger.atInfo().log("Saving premium list for TLD %s", name); logger.atInfo().log("Saving premium list for TLD %s", name);
PremiumListDao.save(name, inputData); PremiumListDao.save(name, currency, inputData);
logger.atInfo().log(message); logger.atInfo().log(message);
} catch (Throwable e) { } catch (Throwable e) {
message = "Unexpected error saving premium list from nomulus tool command"; message = "Unexpected error saving premium list from nomulus tool command";

View file

@ -22,12 +22,10 @@ import static java.nio.charset.StandardCharsets.UTF_8;
import com.beust.jcommander.Parameter; import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters; import com.beust.jcommander.Parameters;
import com.google.common.base.Strings; import com.google.common.base.Strings;
import com.googlecode.objectify.Key;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.persistence.VKey;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import google.registry.schema.tld.PremiumListUtils;
import java.nio.file.Files; import java.nio.file.Files;
import org.joda.money.CurrencyUnit;
/** Command to create a {@link PremiumList} on Database. */ /** Command to create a {@link PremiumList} on Database. */
@Parameters(separators = " =", commandDescription = "Create a PremiumList in Database.") @Parameters(separators = " =", commandDescription = "Create a PremiumList in Database.")
@ -38,9 +36,16 @@ public class CreatePremiumListCommand extends CreateOrUpdatePremiumListCommand {
description = "Override restrictions on premium list naming") description = "Override restrictions on premium list naming")
boolean override; boolean override;
@Parameter(
names = {"-c", "--currency"},
description = "CurrencyUnit for the list",
required = true)
String currencyUnit;
@Override @Override
// Using CreatePremiumListAction.java as reference; // Using CreatePremiumListAction.java as reference;
protected void init() throws Exception { protected String prompt() throws Exception {
currency = CurrencyUnit.of(currencyUnit);
name = Strings.isNullOrEmpty(name) ? convertFilePathToName(inputFile) : name; name = Strings.isNullOrEmpty(name) ? convertFilePathToName(inputFile) : name;
checkArgument( checkArgument(
!PremiumListDao.getLatestRevision(name).isPresent(), !PremiumListDao.getLatestRevision(name).isPresent(),
@ -53,9 +58,6 @@ public class CreatePremiumListCommand extends CreateOrUpdatePremiumListCommand {
+ " (unless --override is specified), yet TLD %s does not exist"); + " (unless --override is specified), yet TLD %s does not exist");
} }
inputData = Files.readAllLines(inputFile, UTF_8); inputData = Files.readAllLines(inputFile, UTF_8);
// create a premium list with only input data and store as the first version of the entity return String.format("Create new premium list for %s?", name);
PremiumList newPremiumList = PremiumListUtils.parseToPremiumList(name, inputData);
stageEntityChange(
null, newPremiumList, VKey.createOfy(PremiumList.class, Key.create(newPremiumList)));
} }
} }

View file

@ -63,8 +63,7 @@ final class CreateReservedListCommand extends CreateOrUpdateReservedListCommand
.setName(name) .setName(name)
.setReservedListMapFromLines(allLines) .setReservedListMapFromLines(allLines)
.setShouldPublish(shouldPublish) .setShouldPublish(shouldPublish)
.setCreationTime(now) .setCreationTimestamp(now)
.setLastUpdateTime(now)
.build(); .build();
// calls the stageEntityChange method that takes old entity, new entity and a new vkey; // calls the stageEntityChange method that takes old entity, new entity and a new vkey;

View file

@ -17,10 +17,12 @@ package google.registry.tools;
import com.beust.jcommander.Parameter; import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters; import com.beust.jcommander.Parameters;
import com.google.common.collect.Streams; import com.google.common.collect.Streams;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import java.util.Comparator; import java.util.Comparator;
import java.util.List; import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors; import java.util.stream.Collectors;
/** Retrieves and prints one or more premium lists. */ /** Retrieves and prints one or more premium lists. */
@ -33,13 +35,14 @@ public class GetPremiumListCommand implements CommandWithRemoteApi {
@Override @Override
public void run() { public void run() {
for (String premiumListName : mainParameters) { for (String premiumListName : mainParameters) {
if (PremiumListDao.getLatestRevision(premiumListName).isPresent()) { Optional<PremiumList> premiumList = PremiumListDao.getLatestRevision(premiumListName);
if (premiumList.isPresent()) {
System.out.printf( System.out.printf(
"%s:\n%s\n", "%s:\n%s\n",
premiumListName, premiumListName,
Streams.stream(PremiumListDao.loadAllPremiumListEntries(premiumListName)) Streams.stream(PremiumListDao.loadAllPremiumEntries(premiumListName))
.sorted(Comparator.comparing(PremiumListEntry::getLabel)) .sorted(Comparator.comparing(PremiumEntry::getDomainLabel))
.map(PremiumListEntry::toString) .map(premiumEntry -> premiumEntry.toString(premiumList.get().getCurrency()))
.collect(Collectors.joining("\n"))); .collect(Collectors.joining("\n")));
} else { } else {
System.out.printf("No list found with name %s.%n", premiumListName); System.out.printf("No list found with name %s.%n", premiumListName);

View file

@ -42,7 +42,7 @@ public class GetReservedListCommand implements CommandWithRemoteApi {
.get() .get()
.getReservedListEntries() .getReservedListEntries()
.values()) .values())
.sorted(Comparator.comparing(ReservedListEntry::getLabel)) .sorted(Comparator.comparing(ReservedListEntry::getDomainLabel))
.map(ReservedListEntry::toString) .map(ReservedListEntry::toString)
.collect(Collectors.joining("\n"))); .collect(Collectors.joining("\n")));
} else { } else {

View file

@ -24,17 +24,13 @@ import com.beust.jcommander.Parameters;
import com.google.common.base.Strings; import com.google.common.base.Strings;
import com.google.common.collect.ImmutableSet; import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Streams; import com.google.common.collect.Streams;
import com.googlecode.objectify.Key;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.persistence.VKey;
import google.registry.schema.tld.PremiumEntry;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import google.registry.schema.tld.PremiumListUtils; import google.registry.schema.tld.PremiumListUtils;
import java.nio.file.Files; import java.nio.file.Files;
import java.util.List; import java.util.List;
import java.util.Optional; import java.util.Optional;
import org.joda.money.BigMoney;
/** Command to safely update {@link PremiumList} in Database for a given TLD. */ /** Command to safely update {@link PremiumList} in Database for a given TLD. */
@Parameters(separators = " =", commandDescription = "Update a PremiumList in Database.") @Parameters(separators = " =", commandDescription = "Update a PremiumList in Database.")
@ -42,23 +38,23 @@ class UpdatePremiumListCommand extends CreateOrUpdatePremiumListCommand {
@Override @Override
// Using UpdatePremiumListAction.java as reference; // Using UpdatePremiumListAction.java as reference;
protected void init() throws Exception { protected String prompt() throws Exception {
name = Strings.isNullOrEmpty(name) ? convertFilePathToName(inputFile) : name; name = Strings.isNullOrEmpty(name) ? convertFilePathToName(inputFile) : name;
List<String> existingEntry = getExistingPremiumListEntry(name).asList(); Optional<PremiumList> list = PremiumListDao.getLatestRevision(name);
checkArgument(
list.isPresent(),
String.format("Could not update premium list %s because it doesn't exist.", name));
List<String> existingEntry = getExistingPremiumEntry(list.get()).asList();
inputData = Files.readAllLines(inputFile, UTF_8); inputData = Files.readAllLines(inputFile, UTF_8);
currency = list.get().getCurrency();
// reconstructing existing premium list to bypass Hibernate lazy initialization exception // reconstructing existing premium list to bypass Hibernate lazy initialization exception
PremiumList existingPremiumList = PremiumListUtils.parseToPremiumList(name, existingEntry); PremiumList existingPremiumList =
PremiumList updatedPremiumList = PremiumListUtils.parseToPremiumList(name, inputData); PremiumListUtils.parseToPremiumList(name, currency, existingEntry);
PremiumList updatedPremiumList = PremiumListUtils.parseToPremiumList(name, currency, inputData);
// use LabelsToPrices() for comparison between old and new premium lists since they have return String.format(
// different creation date, updated date even if they have same content; "Update premium list for %s?\n Old List: %s\n New List: %s",
if (!existingPremiumList.getLabelsToPrices().equals(updatedPremiumList.getLabelsToPrices())) { name, existingPremiumList, updatedPremiumList);
stageEntityChange(
existingPremiumList,
updatedPremiumList,
VKey.createOfy(PremiumList.class, Key.create(existingPremiumList)));
}
} }
/* /*
@ -78,22 +74,16 @@ class UpdatePremiumListCommand extends CreateOrUpdatePremiumListCommand {
assertThat(persistedList.getLabelsToPrices()).containsEntry("foo", new BigDecimal("9000.00")); assertThat(persistedList.getLabelsToPrices()).containsEntry("foo", new BigDecimal("9000.00"));
assertThat(persistedList.size()).isEqualTo(1); assertThat(persistedList.size()).isEqualTo(1);
*/ */
protected ImmutableSet<String> getExistingPremiumListEntry(String name) { protected ImmutableSet<String> getExistingPremiumEntry(PremiumList list) {
Optional<PremiumList> list = PremiumListDao.getLatestRevision(name);
checkArgument(
list.isPresent(),
String.format("Could not update premium list %s because it doesn't exist.", name));
Iterable<PremiumEntry> sqlListEntries = Iterable<PremiumEntry> sqlListEntries =
jpaTm().transact(() -> PremiumListDao.loadPremiumListEntries(list.get())); jpaTm().transact(() -> PremiumListDao.loadPremiumEntries(list));
return Streams.stream(sqlListEntries) return Streams.stream(sqlListEntries)
.map( .map(
premiumEntry -> premiumEntry ->
new PremiumListEntry.Builder() String.format(
.setPrice( "%s,%s %s",
BigMoney.of(list.get().getCurrency(), premiumEntry.getPrice()).toMoney()) premiumEntry.getDomainLabel(), list.getCurrency(), premiumEntry.getValue()))
.setLabel(premiumEntry.getDomainLabel())
.build()
.toString())
.collect(toImmutableSet()); .collect(toImmutableSet());
} }
} }

View file

@ -27,6 +27,7 @@ import google.registry.request.auth.Auth;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import java.util.List; import java.util.List;
import javax.inject.Inject; import javax.inject.Inject;
import org.joda.money.CurrencyUnit;
/** /**
* An action that creates a premium list, for use by the {@code nomulus create_premium_list} * An action that creates a premium list, for use by the {@code nomulus create_premium_list}
@ -43,9 +44,14 @@ public class CreatePremiumListAction extends CreateOrUpdatePremiumListAction {
public static final String OVERRIDE_PARAM = "override"; public static final String OVERRIDE_PARAM = "override";
public static final String PATH = "/_dr/admin/createPremiumList"; public static final String PATH = "/_dr/admin/createPremiumList";
public static final String CURRENCY = "currency";
@Inject @Parameter(OVERRIDE_PARAM) boolean override; @Inject @Parameter(OVERRIDE_PARAM) boolean override;
@Inject
@Parameter("currency")
CurrencyUnit currency;
@Inject CreatePremiumListAction() {} @Inject CreatePremiumListAction() {}
@Override @Override
@ -64,7 +70,7 @@ public class CreatePremiumListAction extends CreateOrUpdatePremiumListAction {
logInputData(); logInputData();
List<String> inputDataPreProcessed = List<String> inputDataPreProcessed =
Splitter.on('\n').omitEmptyStrings().splitToList(inputData); Splitter.on('\n').omitEmptyStrings().splitToList(inputData);
PremiumListDao.save(name, inputDataPreProcessed); PremiumListDao.save(name, currency, inputDataPreProcessed);
String message = String message =
String.format("Saved premium list %s with %d entries", name, inputDataPreProcessed.size()); String.format("Saved premium list %s with %d entries", name, inputDataPreProcessed.size());
logger.atInfo().log(message); logger.atInfo().log(message);

View file

@ -28,6 +28,7 @@ import google.registry.request.Parameter;
import google.registry.request.RequestParameters; import google.registry.request.RequestParameters;
import java.util.Optional; import java.util.Optional;
import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequest;
import org.joda.money.CurrencyUnit;
/** /**
* Dagger module for the tools package. * Dagger module for the tools package.
@ -66,6 +67,12 @@ public class ToolsServerModule {
return extractRequiredParameter(req, CreatePremiumListAction.NAME_PARAM); return extractRequiredParameter(req, CreatePremiumListAction.NAME_PARAM);
} }
@Provides
@Parameter("currency")
static CurrencyUnit provideCurrency(HttpServletRequest req) {
return CurrencyUnit.of(extractRequiredParameter(req, CreatePremiumListAction.CURRENCY));
}
@Provides @Provides
@Parameter("override") @Parameter("override")
static boolean provideOverride(HttpServletRequest req) { static boolean provideOverride(HttpServletRequest req) {

View file

@ -25,6 +25,7 @@ import google.registry.request.Action;
import google.registry.request.auth.Auth; import google.registry.request.auth.Auth;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import java.util.List; import java.util.List;
import java.util.Optional;
import javax.inject.Inject; import javax.inject.Inject;
/** /**
@ -46,8 +47,9 @@ public class UpdatePremiumListAction extends CreateOrUpdatePremiumListAction {
@Override @Override
protected void save() { protected void save() {
Optional<PremiumList> existingList = PremiumListDao.getLatestRevision(name);
checkArgument( checkArgument(
PremiumListDao.getLatestRevision(name).isPresent(), existingList.isPresent(),
"Could not update premium list %s because it doesn't exist.", "Could not update premium list %s because it doesn't exist.",
name); name);
@ -55,7 +57,8 @@ public class UpdatePremiumListAction extends CreateOrUpdatePremiumListAction {
logInputData(); logInputData();
List<String> inputDataPreProcessed = List<String> inputDataPreProcessed =
Splitter.on('\n').omitEmptyStrings().splitToList(inputData); Splitter.on('\n').omitEmptyStrings().splitToList(inputData);
PremiumList newPremiumList = PremiumListDao.save(name, inputDataPreProcessed); PremiumList newPremiumList =
PremiumListDao.save(name, existingList.get().getCurrency(), inputDataPreProcessed);
String message = String message =
String.format( String.format(

View file

@ -60,6 +60,7 @@
<class>google.registry.model.registrar.Registrar</class> <class>google.registry.model.registrar.Registrar</class>
<class>google.registry.model.registrar.RegistrarContact</class> <class>google.registry.model.registrar.RegistrarContact</class>
<class>google.registry.model.registry.label.PremiumList</class> <class>google.registry.model.registry.label.PremiumList</class>
<class>google.registry.model.registry.label.PremiumList$PremiumEntry</class>
<class>google.registry.model.registry.label.ReservedList</class> <class>google.registry.model.registry.label.ReservedList</class>
<class>google.registry.model.registry.label.ReservedList$ReservedListEntry</class> <class>google.registry.model.registry.label.ReservedList$ReservedListEntry</class>
<class>google.registry.model.registry.Registry</class> <class>google.registry.model.registry.Registry</class>
@ -74,7 +75,6 @@
<class>google.registry.persistence.transaction.TransactionEntity</class> <class>google.registry.persistence.transaction.TransactionEntity</class>
<class>google.registry.schema.domain.RegistryLock</class> <class>google.registry.schema.domain.RegistryLock</class>
<class>google.registry.schema.replay.SqlReplayCheckpoint</class> <class>google.registry.schema.replay.SqlReplayCheckpoint</class>
<class>google.registry.schema.tld.PremiumEntry</class>
<!-- Customized type converters --> <!-- Customized type converters -->
<class>google.registry.persistence.converter.AllocationTokenStatusTransitionConverter</class> <class>google.registry.persistence.converter.AllocationTokenStatusTransitionConverter</class>
@ -90,7 +90,6 @@
<class>google.registry.persistence.converter.InetAddressSetConverter</class> <class>google.registry.persistence.converter.InetAddressSetConverter</class>
<class>google.registry.persistence.converter.LocalDateConverter</class> <class>google.registry.persistence.converter.LocalDateConverter</class>
<class>google.registry.persistence.converter.PostalInfoChoiceListConverter</class> <class>google.registry.persistence.converter.PostalInfoChoiceListConverter</class>
<class>google.registry.persistence.converter.PremiumListKeyConverter</class>
<class>google.registry.persistence.converter.RegistrarPocSetConverter</class> <class>google.registry.persistence.converter.RegistrarPocSetConverter</class>
<class>google.registry.persistence.converter.ReservedListKeySetConverter</class> <class>google.registry.persistence.converter.ReservedListKeySetConverter</class>
<class>google.registry.persistence.converter.Spec11ThreatMatchThreatTypeSetConverter</class> <class>google.registry.persistence.converter.Spec11ThreatMatchThreatTypeSetConverter</class>

View file

@ -57,6 +57,7 @@ import google.registry.model.ofy.CommitLogManifest;
import google.registry.model.ofy.CommitLogMutation; import google.registry.model.ofy.CommitLogMutation;
import google.registry.model.registrar.RegistrarContact; import google.registry.model.registrar.RegistrarContact;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.model.registry.label.ReservedList; import google.registry.model.registry.label.ReservedList;
import google.registry.model.server.Lock; import google.registry.model.server.Lock;
import google.registry.model.tmch.ClaimsList; import google.registry.model.tmch.ClaimsList;
@ -65,7 +66,6 @@ import google.registry.persistence.VKey;
import google.registry.persistence.transaction.JpaTransactionManager; import google.registry.persistence.transaction.JpaTransactionManager;
import google.registry.persistence.transaction.TransactionManagerFactory; import google.registry.persistence.transaction.TransactionManagerFactory;
import google.registry.schema.replay.SqlReplayCheckpoint; import google.registry.schema.replay.SqlReplayCheckpoint;
import google.registry.schema.tld.PremiumEntry;
import google.registry.testing.AppEngineExtension; import google.registry.testing.AppEngineExtension;
import google.registry.testing.DatabaseHelper; import google.registry.testing.DatabaseHelper;
import google.registry.testing.FakeClock; import google.registry.testing.FakeClock;

View file

@ -691,7 +691,7 @@ public class ExpandRecurringBillingEventsActionTest
persistResource( persistResource(
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setPremiumList(persistPremiumList("tld2", "example,USD 100")) .setPremiumList(persistPremiumList("tld2", USD, "example,USD 100"))
.build()); .build());
persistResource(recurring); persistResource(recurring);
action.cursorTimeParam = Optional.of(START_OF_TIME); action.cursorTimeParam = Optional.of(START_OF_TIME);

View file

@ -23,6 +23,7 @@ import static google.registry.testing.DatabaseHelper.persistResource;
import static java.nio.charset.StandardCharsets.UTF_8; import static java.nio.charset.StandardCharsets.UTF_8;
import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR; import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR;
import static javax.servlet.http.HttpServletResponse.SC_OK; import static javax.servlet.http.HttpServletResponse.SC_OK;
import static org.joda.money.CurrencyUnit.USD;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.ArgumentMatchers.anyString;
@ -54,7 +55,7 @@ public class ExportPremiumTermsActionTest {
private static final ImmutableList<String> PREMIUM_NAMES = private static final ImmutableList<String> PREMIUM_NAMES =
ImmutableList.of("2048,USD 549", "0,USD 549"); ImmutableList.of("2048,USD 549", "0,USD 549");
private static final String EXPECTED_FILE_CONTENT = private static final String EXPECTED_FILE_CONTENT =
DISCLAIMER_WITH_NEWLINE + "0,USD 549.00\n" + "2048,USD 549.00\n"; DISCLAIMER_WITH_NEWLINE + "0, 549.00\n" + "2048, 549.00\n";
@RegisterExtension @RegisterExtension
public final AppEngineExtension appEngine = public final AppEngineExtension appEngine =
@ -75,7 +76,7 @@ public class ExportPremiumTermsActionTest {
@BeforeEach @BeforeEach
void beforeEach() throws Exception { void beforeEach() throws Exception {
createTld("tld"); createTld("tld");
PremiumList pl = PremiumListDao.save("pl-name", PREMIUM_NAMES); PremiumList pl = PremiumListDao.save("pl-name", USD, PREMIUM_NAMES);
persistResource( persistResource(
Registry.get("tld").asBuilder().setPremiumList(pl).setDriveFolderId("folder_id").build()); Registry.get("tld").asBuilder().setPremiumList(pl).setDriveFolderId("folder_id").build());
when(driveConnection.createOrUpdateFile( when(driveConnection.createOrUpdateFile(

View file

@ -895,7 +895,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v06.xml"); setEppInput("domain_check_fee_reserved_v06.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v06.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v06.xml"));
@ -907,7 +907,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
// The domain needs to exist in order for it to be loaded to check for restore fee. // The domain needs to exist in order for it to be loaded to check for restore fee.
persistActiveDomain("allowedinsunrise.tld"); persistActiveDomain("allowedinsunrise.tld");
@ -922,7 +922,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_create.xml"); setEppInput("domain_check_fee_reserved_v11_create.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_create.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_create.xml"));
@ -934,7 +934,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_renew.xml"); setEppInput("domain_check_fee_reserved_v11_renew.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_renew.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_renew.xml"));
@ -946,7 +946,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_transfer.xml"); setEppInput("domain_check_fee_reserved_v11_transfer.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_transfer.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_transfer.xml"));
@ -958,7 +958,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_restore.xml"); setEppInput("domain_check_fee_reserved_v11_restore.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_restore.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v11_restore.xml"));
@ -970,7 +970,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
persistPendingDeleteDomain("reserved.tld"); persistPendingDeleteDomain("reserved.tld");
persistPendingDeleteDomain("allowedinsunrise.tld"); persistPendingDeleteDomain("allowedinsunrise.tld");
@ -987,7 +987,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v12.xml"); setEppInput("domain_check_fee_reserved_v12.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v12.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_response_v12.xml"));
@ -999,7 +999,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
// The domain needs to exist in order for it to be loaded to check for restore fee. // The domain needs to exist in order for it to be loaded to check for restore fee.
persistActiveDomain("allowedinsunrise.tld"); persistActiveDomain("allowedinsunrise.tld");
@ -1014,7 +1014,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v06.xml"); setEppInput("domain_check_fee_reserved_v06.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v06.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v06.xml"));
@ -1028,7 +1028,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
persistPendingDeleteDomain("reserved.tld"); persistPendingDeleteDomain("reserved.tld");
persistPendingDeleteDomain("allowedinsunrise.tld"); persistPendingDeleteDomain("allowedinsunrise.tld");
@ -1046,7 +1046,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_create.xml"); setEppInput("domain_check_fee_reserved_v11_create.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_create.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_create.xml"));
@ -1059,7 +1059,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_renew.xml"); setEppInput("domain_check_fee_reserved_v11_renew.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_renew.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_renew.xml"));
@ -1072,7 +1072,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_transfer.xml"); setEppInput("domain_check_fee_reserved_v11_transfer.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_transfer.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_transfer.xml"));
@ -1085,7 +1085,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v11_restore.xml"); setEppInput("domain_check_fee_reserved_v11_restore.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_restore.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v11_restore.xml"));
@ -1098,7 +1098,7 @@ class DomainCheckFlowTest extends ResourceCheckFlowTestCase<DomainCheckFlow, Dom
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setReservedLists(createReservedList()) .setReservedLists(createReservedList())
.setPremiumList(persistPremiumList("tld", "premiumcollision,USD 70")) .setPremiumList(persistPremiumList("tld", USD, "premiumcollision,USD 70"))
.build()); .build());
setEppInput("domain_check_fee_reserved_v12.xml"); setEppInput("domain_check_fee_reserved_v12.xml");
runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v12.xml")); runFlowAssertResponse(loadFile("domain_check_fee_reserved_sunrise_response_v12.xml"));

View file

@ -64,7 +64,7 @@ public final class OteAccountBuilderTest {
@BeforeEach @BeforeEach
void beforeEach() { void beforeEach() {
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
} }
private void assertTldExists(String tld, TldState tldState, Money eapFee) { private void assertTldExists(String tld, TldState tldState, Money eapFee) {

View file

@ -23,6 +23,7 @@ import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static google.registry.testing.TestDataHelper.loadBytes; import static google.registry.testing.TestDataHelper.loadBytes;
import static google.registry.util.DateTimeUtils.END_OF_TIME; import static google.registry.util.DateTimeUtils.END_OF_TIME;
import static org.joda.money.CurrencyUnit.USD;
import google.registry.model.domain.DomainHistory; import google.registry.model.domain.DomainHistory;
import google.registry.model.eppcommon.Trid; import google.registry.model.eppcommon.Trid;
@ -77,7 +78,7 @@ public final class OteStatsTestHelper {
*/ */
public static void setupIncompleteOte(String baseClientId) throws IOException { public static void setupIncompleteOte(String baseClientId) throws IOException {
createTld("tld"); createTld("tld");
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
OteAccountBuilder.forClientId(baseClientId).addContact("email@example.com").buildAndPersist(); OteAccountBuilder.forClientId(baseClientId).addContact("email@example.com").buildAndPersist();
String oteAccount1 = String.format("%s-1", baseClientId); String oteAccount1 = String.format("%s-1", baseClientId);
DateTime now = DateTime.now(DateTimeZone.UTC); DateTime now = DateTime.now(DateTimeZone.UTC);

View file

@ -71,7 +71,7 @@ public final class RegistryTest extends EntityTestCase {
@TestOfyAndSql @TestOfyAndSql
void testPersistence_updateReservedAndPremiumListSuccessfully() { void testPersistence_updateReservedAndPremiumListSuccessfully() {
ReservedList rl15 = persistReservedList("tld-reserved15", "potato,FULLY_BLOCKED"); ReservedList rl15 = persistReservedList("tld-reserved15", "potato,FULLY_BLOCKED");
PremiumList pl = persistPremiumList("tld2", "lol,USD 50", "cat,USD 700"); PremiumList pl = persistPremiumList("tld2", USD, "lol,USD 50", "cat,USD 700");
Registry registry = Registry registry =
Registry.get("tld").asBuilder().setReservedLists(rl15).setPremiumList(pl).build(); Registry.get("tld").asBuilder().setReservedLists(rl15).setPremiumList(pl).build();
tm().transact(() -> tm().put(registry)); tm().transact(() -> tm().put(registry));
@ -145,7 +145,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED")) ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
ReservedList rl16 = ReservedList rl16 =
persistReservedList( persistReservedList(
@ -154,7 +154,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED")) ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
Registry registry1 = Registry registry1 =
newRegistry("propter", "PROPTER") newRegistry("propter", "PROPTER")
@ -197,7 +197,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED")) ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
ReservedList rl6 = ReservedList rl6 =
persistReservedList( persistReservedList(
@ -206,7 +206,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED")) ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
Registry r = Registry r =
Registry.get("tld").asBuilder().setReservedLists(ImmutableSet.of(rl5, rl6)).build(); Registry.get("tld").asBuilder().setReservedLists(ImmutableSet.of(rl5, rl6)).build();
@ -223,7 +223,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED")) ImmutableList.of("potato,FULLY_BLOCKED", "phone,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
persistReservedList( persistReservedList(
new ReservedList.Builder() new ReservedList.Builder()
@ -231,7 +231,7 @@ public final class RegistryTest extends EntityTestCase {
.setReservedListMapFromLines( .setReservedListMapFromLines(
ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED")) ImmutableList.of("port,FULLY_BLOCKED", "manteau,FULLY_BLOCKED"))
.setShouldPublish(true) .setShouldPublish(true)
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
Registry r = Registry r =
Registry.get("tld") Registry.get("tld")
@ -245,7 +245,7 @@ public final class RegistryTest extends EntityTestCase {
@TestOfyAndSql @TestOfyAndSql
void testSetPremiumList() { void testSetPremiumList() {
PremiumList pl2 = persistPremiumList("tld2", "lol,USD 50", "cat,USD 700"); PremiumList pl2 = persistPremiumList("tld2", USD, "lol,USD 50", "cat,USD 700");
Registry registry = Registry.get("tld").asBuilder().setPremiumList(pl2).build(); Registry registry = Registry.get("tld").asBuilder().setPremiumList(pl2).build();
Optional<String> pl = registry.getPremiumListName(); Optional<String> pl = registry.getPremiumListName();
assertThat(pl).hasValue("tld2"); assertThat(pl).hasValue("tld2");

View file

@ -27,7 +27,7 @@ import static org.junit.jupiter.api.Assertions.assertThrows;
import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableList;
import com.google.common.hash.BloomFilter; import com.google.common.hash.BloomFilter;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import google.registry.testing.AppEngineExtension; import google.registry.testing.AppEngineExtension;
import java.math.BigDecimal; import java.math.BigDecimal;
@ -48,11 +48,7 @@ public class PremiumListTest {
createTld("tld"); createTld("tld");
PremiumList pl = PremiumList pl =
persistPremiumList( persistPremiumList(
"tld", "tld", USD, "lol, 999", "rich, 1999", "icann, 100", "johnny-be-goode, 20.50");
"lol,USD 999 # yup",
"rich,USD 1999 #tada",
"icann,JPY 100",
"johnny-be-goode,USD 20.50");
persistResource(Registry.get("tld").asBuilder().setPremiumList(pl).build()); persistResource(Registry.get("tld").asBuilder().setPremiumList(pl).build());
} }
@ -60,7 +56,7 @@ public class PremiumListTest {
void testSave_badSyntax() { void testSave_badSyntax() {
assertThrows( assertThrows(
IllegalArgumentException.class, IllegalArgumentException.class,
() -> persistPremiumList("gtld1", "lol,nonsense USD,e,e # yup")); () -> persistPremiumList("gtld1", USD, "lol,nonsense USD,e,e # yup"));
} }
@Test @Test
@ -102,14 +98,21 @@ public class PremiumListTest {
"List 'tld' cannot contain duplicate labels. Dupes (with counts) were: [lol x 2]"); "List 'tld' cannot contain duplicate labels. Dupes (with counts) were: [lol x 2]");
} }
@Test
void testParse_canIncludeOrNotIncludeCurrencyUnit() {
PremiumListDao.save("tld", USD, ImmutableList.of("rofl,USD 90", "paper, 80"));
assertThat(PremiumListDao.getPremiumPrice("tld", "rofl").get()).isEqualTo(Money.of(USD, 90));
assertThat(PremiumListDao.getPremiumPrice("tld", "paper").get()).isEqualTo(Money.of(USD, 80));
}
@Test @Test
void testValidation_labelMustBeLowercase() { void testValidation_labelMustBeLowercase() {
Exception e = Exception e =
assertThrows( assertThrows(
IllegalArgumentException.class, IllegalArgumentException.class,
() -> () ->
new PremiumListEntry.Builder() new PremiumEntry.Builder()
.setPrice(Money.parse("USD 399")) .setPrice(BigDecimal.valueOf(399))
.setLabel("UPPER.tld") .setLabel("UPPER.tld")
.build()); .build());
assertThat(e).hasMessageThat().contains("must be in puny-coded, lower-case form"); assertThat(e).hasMessageThat().contains("must be in puny-coded, lower-case form");
@ -121,8 +124,8 @@ public class PremiumListTest {
assertThrows( assertThrows(
IllegalArgumentException.class, IllegalArgumentException.class,
() -> () ->
new PremiumListEntry.Builder() new PremiumEntry.Builder()
.setPrice(Money.parse("USD 399")) .setPrice(BigDecimal.valueOf(399))
.setLabel("lower.みんな") .setLabel("lower.みんな")
.build()); .build());
assertThat(e).hasMessageThat().contains("must be in puny-coded, lower-case form"); assertThat(e).hasMessageThat().contains("must be in puny-coded, lower-case form");

View file

@ -57,7 +57,7 @@ public class ReservedListDaoTest {
testReservedList = testReservedList =
new ReservedList.Builder() new ReservedList.Builder()
.setName("testlist") .setName("testlist")
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.setShouldPublish(false) .setShouldPublish(false)
.setReservedListMap(testReservations) .setReservedListMap(testReservations)
.build(); .build();
@ -76,7 +76,7 @@ public class ReservedListDaoTest {
.getSingleResult(); .getSingleResult();
assertThat(persistedList.getReservedListEntries()) assertThat(persistedList.getReservedListEntries())
.containsExactlyEntriesIn(testReservations); .containsExactlyEntriesIn(testReservations);
assertThat(persistedList.getLastUpdateTime()).isEqualTo(fakeClock.nowUtc()); assertThat(persistedList.getCreationTimestamp()).isEqualTo(fakeClock.nowUtc());
}); });
} }
@ -108,7 +108,7 @@ public class ReservedListDaoTest {
ReservedListDao.save(testReservedList); ReservedListDao.save(testReservedList);
ReservedList persistedList = ReservedListDao.getLatestRevision("testlist").get(); ReservedList persistedList = ReservedListDao.getLatestRevision("testlist").get();
assertThat(persistedList.getRevisionId()).isNotNull(); assertThat(persistedList.getRevisionId()).isNotNull();
assertThat(persistedList.getLastUpdateTime()).isEqualTo(fakeClock.nowUtc()); assertThat(persistedList.getCreationTimestamp()).isEqualTo(fakeClock.nowUtc());
assertThat(persistedList.getName()).isEqualTo("testlist"); assertThat(persistedList.getName()).isEqualTo("testlist");
assertThat(persistedList.getShouldPublish()).isFalse(); assertThat(persistedList.getShouldPublish()).isFalse();
assertThat(persistedList.getReservedListEntries()).containsExactlyEntriesIn(testReservations); assertThat(persistedList.getReservedListEntries()).containsExactlyEntriesIn(testReservations);
@ -119,7 +119,7 @@ public class ReservedListDaoTest {
ReservedListDao.save( ReservedListDao.save(
new ReservedList.Builder() new ReservedList.Builder()
.setName("testlist") .setName("testlist")
.setLastUpdateTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.setShouldPublish(false) .setShouldPublish(false)
.setReservedListMap( .setReservedListMap(
ImmutableMap.of( ImmutableMap.of(
@ -130,7 +130,7 @@ public class ReservedListDaoTest {
ReservedListDao.save(testReservedList); ReservedListDao.save(testReservedList);
ReservedList persistedList = ReservedListDao.getLatestRevision("testlist").get(); ReservedList persistedList = ReservedListDao.getLatestRevision("testlist").get();
assertThat(persistedList.getRevisionId()).isNotNull(); assertThat(persistedList.getRevisionId()).isNotNull();
assertThat(persistedList.getLastUpdateTime()).isEqualTo(fakeClock.nowUtc()); assertThat(persistedList.getCreationTimestamp()).isEqualTo(fakeClock.nowUtc());
assertThat(persistedList.getName()).isEqualTo("testlist"); assertThat(persistedList.getName()).isEqualTo("testlist");
assertThat(persistedList.getShouldPublish()).isFalse(); assertThat(persistedList.getShouldPublish()).isFalse();
assertThat(persistedList.getReservedListEntries()).containsExactlyEntriesIn(testReservations); assertThat(persistedList.getReservedListEntries()).containsExactlyEntriesIn(testReservations);

View file

@ -35,7 +35,7 @@ import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet; import com.google.common.collect.ImmutableSet;
import google.registry.model.ofy.Ofy; import google.registry.model.ofy.Ofy;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.schema.tld.PremiumEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.testing.AppEngineExtension; import google.registry.testing.AppEngineExtension;
import google.registry.testing.FakeClock; import google.registry.testing.FakeClock;
import google.registry.testing.InjectExtension; import google.registry.testing.InjectExtension;
@ -255,17 +255,17 @@ class ReservedListTest {
assertThat(reservedList.getReservedListEntries()).hasSize(3); assertThat(reservedList.getReservedListEntries()).hasSize(3);
ReservedListEntry trombone = reservedList.getReservedListEntries().get("trombone"); ReservedListEntry trombone = reservedList.getReservedListEntries().get("trombone");
assertThat(trombone.label).isEqualTo("trombone"); assertThat(trombone.domainLabel).isEqualTo("trombone");
assertThat(trombone.reservationType).isEqualTo(FULLY_BLOCKED); assertThat(trombone.reservationType).isEqualTo(FULLY_BLOCKED);
assertThat(trombone.comment).isEqualTo("yup"); assertThat(trombone.comment).isEqualTo("yup");
ReservedListEntry oysters = reservedList.getReservedListEntries().get("oysters"); ReservedListEntry oysters = reservedList.getReservedListEntries().get("oysters");
assertThat(oysters.label).isEqualTo("oysters"); assertThat(oysters.domainLabel).isEqualTo("oysters");
assertThat(oysters.reservationType).isEqualTo(FULLY_BLOCKED); assertThat(oysters.reservationType).isEqualTo(FULLY_BLOCKED);
assertThat(oysters.comment).isEqualTo("this is a loooong comment"); assertThat(oysters.comment).isEqualTo("this is a loooong comment");
ReservedListEntry nullComment = reservedList.getReservedListEntries().get("nullcomment"); ReservedListEntry nullComment = reservedList.getReservedListEntries().get("nullcomment");
assertThat(nullComment.label).isEqualTo("nullcomment"); assertThat(nullComment.domainLabel).isEqualTo("nullcomment");
assertThat(nullComment.reservationType).isEqualTo(ALLOWED_IN_SUNRISE); assertThat(nullComment.reservationType).isEqualTo(ALLOWED_IN_SUNRISE);
assertThat(nullComment.comment).isEmpty(); assertThat(nullComment.comment).isEmpty();
} }
@ -299,8 +299,7 @@ class ReservedListTest {
ReservedList original = persistReservedList("tld-reserved-cloning", "trombone,FULLY_BLOCKED"); ReservedList original = persistReservedList("tld-reserved-cloning", "trombone,FULLY_BLOCKED");
ReservedList clone = original.asBuilder().build(); ReservedList clone = original.asBuilder().build();
assertThat(clone.getName()).isEqualTo("tld-reserved-cloning"); assertThat(clone.getName()).isEqualTo("tld-reserved-cloning");
assertThat(clone.creationTime).isEqualTo(original.creationTime); assertThat(clone.creationTimestamp).isEqualTo(original.creationTimestamp);
assertThat(clone.lastUpdateTime).isEqualTo(original.lastUpdateTime);
assertThat(clone.parent).isEqualTo(original.parent); assertThat(clone.parent).isEqualTo(original.parent);
assertThat(original.getReservedListEntries()).isEqualTo(clone.getReservedListEntries()); assertThat(original.getReservedListEntries()).isEqualTo(clone.getReservedListEntries());
} }

View file

@ -1,90 +0,0 @@
// Copyright 2020 The Nomulus Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package google.registry.persistence.converter;
import static com.google.common.truth.Truth.assertThat;
import static google.registry.model.common.EntityGroupRoot.getCrossTldKey;
import static google.registry.persistence.transaction.TransactionManagerFactory.jpaTm;
import com.googlecode.objectify.Key;
import google.registry.model.ImmutableObject;
import google.registry.model.registry.label.PremiumList;
import google.registry.testing.AppEngineExtension;
import javax.persistence.Entity;
import javax.persistence.Id;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
/** Unit tests for {@link PremiumListKeyConverter}. */
class PremiumListKeyConverterTest {
@RegisterExtension
final AppEngineExtension appEngine =
AppEngineExtension.builder()
.withDatastoreAndCloudSql()
.withJpaUnitTestEntities(PremiumListEntity.class)
.build();
private final PremiumListKeyConverter converter = new PremiumListKeyConverter();
@Test
void convertToDatabaseColumn_returnsNullIfInputIsNull() {
assertThat(converter.convertToDatabaseColumn(null)).isNull();
}
@Test
void convertToDatabaseColumn_convertsCorrectly() {
assertThat(
converter.convertToDatabaseColumn(
Key.create(getCrossTldKey(), PremiumList.class, "testList")))
.isEqualTo("testList");
}
@Test
void convertToEntityAttribute_returnsNullIfInputIsNull() {
assertThat(converter.convertToEntityAttribute(null)).isNull();
}
@Test
void convertToEntityAttribute_convertsCorrectly() {
assertThat(converter.convertToEntityAttribute("testList"))
.isEqualTo(Key.create(getCrossTldKey(), PremiumList.class, "testList"));
}
@Test
void testRoundTrip() {
Key<PremiumList> key = Key.create(getCrossTldKey(), PremiumList.class, "test");
PremiumListEntity testEntity = new PremiumListEntity(key);
jpaTm().transact(() -> jpaTm().insert(testEntity));
PremiumListEntity persisted =
jpaTm().transact(() -> jpaTm().getEntityManager().find(PremiumListEntity.class, "test"));
assertThat(persisted.premiumList).isEqualTo(key);
}
@Entity(name = "PremiumListEntity")
private static class PremiumListEntity extends ImmutableObject {
@Id String name;
Key<PremiumList> premiumList;
public PremiumListEntity() {}
PremiumListEntity(Key<PremiumList> premiumList) {
this.name = premiumList.getName();
this.premiumList = premiumList;
}
}
}

View file

@ -49,7 +49,8 @@ public class PricingEngineProxyTest {
@BeforeEach @BeforeEach
void beforeEach() { void beforeEach() {
PremiumList premiumList = PremiumList premiumList =
persistPremiumList("rich,USD 100", "richer,USD 999", "fraction,USD 20.50"); persistPremiumList(
"premium list", USD, "rich,USD 100", "richer,USD 999", "fraction,USD 20.50");
createTld("moka"); createTld("moka");
persistResource(Registry.get("moka").asBuilder().setPremiumList(premiumList).build()); persistResource(Registry.get("moka").asBuilder().setPremiumList(premiumList).build());
clock = new FakeClock(DateTime.parse("2016-03-17T12:01:00Z")); clock = new FakeClock(DateTime.parse("2016-03-17T12:01:00Z"));

View file

@ -76,7 +76,7 @@ public class PremiumListDaoTest {
.setName("testname") .setName("testname")
.setCurrency(USD) .setCurrency(USD)
.setLabelsToPrices(TEST_PRICES) .setLabelsToPrices(TEST_PRICES)
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build(); .build();
} }
@ -90,7 +90,7 @@ public class PremiumListDaoTest {
assertThat(persistedListOpt).isPresent(); assertThat(persistedListOpt).isPresent();
PremiumList persistedList = persistedListOpt.get(); PremiumList persistedList = persistedListOpt.get();
assertThat(persistedList.getLabelsToPrices()).containsExactlyEntriesIn(TEST_PRICES); assertThat(persistedList.getLabelsToPrices()).containsExactlyEntriesIn(TEST_PRICES);
assertThat(persistedList.getCreationTime()).isEqualTo(fakeClock.nowUtc()); assertThat(persistedList.getCreationTimestamp()).isEqualTo(fakeClock.nowUtc());
}); });
} }
@ -112,7 +112,7 @@ public class PremiumListDaoTest {
BigDecimal.valueOf(0.01), BigDecimal.valueOf(0.01),
"silver", "silver",
BigDecimal.valueOf(30.03))) BigDecimal.valueOf(30.03)))
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
jpaTm() jpaTm()
.transact( .transact(
@ -129,9 +129,8 @@ public class PremiumListDaoTest {
BigDecimal.valueOf(0.01), BigDecimal.valueOf(0.01),
"silver", "silver",
BigDecimal.valueOf(30.03))); BigDecimal.valueOf(30.03)));
assertThat(savedList.getCreationTime()).isEqualTo(fakeClock.nowUtc()); assertThat(savedList.getCreationTimestamp()).isEqualTo(fakeClock.nowUtc());
assertThat(savedList.getRevisionId()).isGreaterThan(firstRevisionId); assertThat(savedList.getRevisionId()).isGreaterThan(firstRevisionId);
assertThat(savedList.getCreationTime()).isEqualTo(fakeClock.nowUtc());
}); });
} }
@ -154,14 +153,14 @@ public class PremiumListDaoTest {
.setName("list1") .setName("list1")
.setCurrency(USD) .setCurrency(USD)
.setLabelsToPrices(ImmutableMap.of("wrong", BigDecimal.valueOf(1000.50))) .setLabelsToPrices(ImmutableMap.of("wrong", BigDecimal.valueOf(1000.50)))
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
PremiumListDao.save( PremiumListDao.save(
new PremiumList.Builder() new PremiumList.Builder()
.setName("list1") .setName("list1")
.setCurrency(USD) .setCurrency(USD)
.setLabelsToPrices(TEST_PRICES) .setLabelsToPrices(TEST_PRICES)
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
jpaTm() jpaTm()
.transact( .transact(
@ -182,7 +181,7 @@ public class PremiumListDaoTest {
.setName("list1") .setName("list1")
.setCurrency(JPY) .setCurrency(JPY)
.setLabelsToPrices(TEST_PRICES) .setLabelsToPrices(TEST_PRICES)
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
jpaTm() jpaTm()
.transact( .transact(
@ -207,7 +206,7 @@ public class PremiumListDaoTest {
.setName("premlist") .setName("premlist")
.setCurrency(USD) .setCurrency(USD)
.setLabelsToPrices(TEST_PRICES) .setLabelsToPrices(TEST_PRICES)
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
persistResource( persistResource(
newRegistry("foobar", "FOOBAR").asBuilder().setPremiumList(premiumList).build()); newRegistry("foobar", "FOOBAR").asBuilder().setPremiumList(premiumList).build());
@ -231,7 +230,7 @@ public class PremiumListDaoTest {
BigDecimal.valueOf(1000.0), BigDecimal.valueOf(1000.0),
"palladium", "palladium",
BigDecimal.valueOf(15000))) BigDecimal.valueOf(15000)))
.setCreationTime(fakeClock.nowUtc()) .setCreationTimestamp(fakeClock.nowUtc())
.build()); .build());
persistResource( persistResource(
newRegistry("foobar", "FOOBAR").asBuilder().setPremiumList(premiumList).build()); newRegistry("foobar", "FOOBAR").asBuilder().setPremiumList(premiumList).build());
@ -247,7 +246,7 @@ public class PremiumListDaoTest {
PremiumListDao.save(testList); PremiumListDao.save(testList);
PremiumList pl = PremiumListDao.getLatestRevision("testname").get(); PremiumList pl = PremiumListDao.getLatestRevision("testname").get();
assertThat(PremiumListDao.premiumListCache.getIfPresent("testname").get()).isEqualTo(pl); assertThat(PremiumListDao.premiumListCache.getIfPresent("testname").get()).isEqualTo(pl);
transactIfJpaTm(() -> PremiumListDao.save("testname", ImmutableList.of("test,USD 1"))); transactIfJpaTm(() -> PremiumListDao.save("testname", USD, ImmutableList.of("test,USD 1")));
assertThat(PremiumListDao.premiumListCache.getIfPresent("testname")).isNull(); assertThat(PremiumListDao.premiumListCache.getIfPresent("testname")).isNull();
} }

View file

@ -16,6 +16,7 @@ package google.registry.schema.tld;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static google.registry.schema.tld.PremiumListUtils.parseToPremiumList; import static google.registry.schema.tld.PremiumListUtils.parseToPremiumList;
import static org.joda.money.CurrencyUnit.USD;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableList;
@ -36,7 +37,7 @@ class PremiumListUtilsTest {
void parseInputToPremiumList_works() { void parseInputToPremiumList_works() {
PremiumList premiumList = PremiumList premiumList =
parseToPremiumList( parseToPremiumList(
"testlist", ImmutableList.of("foo,USD 99.50", "bar,USD 30", "baz,USD 10")); "testlist", USD, ImmutableList.of("foo,USD 99.50", "bar,USD 30", "baz,USD 10"));
assertThat(premiumList.getName()).isEqualTo("testlist"); assertThat(premiumList.getName()).isEqualTo("testlist");
assertThat(premiumList.getLabelsToPrices()) assertThat(premiumList.getLabelsToPrices())
.containsExactly("foo", twoDigits(99.50), "bar", twoDigits(30), "baz", twoDigits(10)); .containsExactly("foo", twoDigits(99.50), "bar", twoDigits(30), "baz", twoDigits(10));
@ -49,10 +50,10 @@ class PremiumListUtilsTest {
IllegalArgumentException.class, IllegalArgumentException.class,
() -> () ->
parseToPremiumList( parseToPremiumList(
"testlist", ImmutableList.of("foo,USD 99.50", "bar,USD 30", "baz,JPY 990"))); "testlist",
assertThat(thrown) USD,
.hasMessageThat() ImmutableList.of("foo,USD 99.50", "bar,USD 30", "baz,JPY 990")));
.isEqualTo("The Cloud SQL schema requires exactly one currency, but got: [JPY, USD]"); assertThat(thrown).hasMessageThat().isEqualTo("The currency unit must be USD");
} }
private static BigDecimal twoDigits(double num) { private static BigDecimal twoDigits(double num) {

View file

@ -24,6 +24,7 @@ import static google.registry.testing.DatabaseHelper.newDomainBase;
import static google.registry.testing.DatabaseHelper.persistActiveHost; import static google.registry.testing.DatabaseHelper.persistActiveHost;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static org.joda.money.CurrencyUnit.USD;
import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet; import com.google.common.collect.ImmutableSet;
@ -61,7 +62,7 @@ public enum Fixture {
createTlds("xn--q9jyb4c", "example"); createTlds("xn--q9jyb4c", "example");
// Used for OT&E TLDs // Used for OT&E TLDs
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
try { try {
OteStatsTestHelper.setupCompleteOte("otefinished"); OteStatsTestHelper.setupCompleteOte("otefinished");

View file

@ -101,7 +101,7 @@ import google.registry.model.registry.Registry;
import google.registry.model.registry.Registry.TldState; import google.registry.model.registry.Registry.TldState;
import google.registry.model.registry.Registry.TldType; import google.registry.model.registry.Registry.TldType;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.model.registry.label.PremiumList.PremiumListEntry; import google.registry.model.registry.label.PremiumList.PremiumEntry;
import google.registry.model.registry.label.ReservedList; import google.registry.model.registry.label.ReservedList;
import google.registry.model.registry.label.ReservedListDao; import google.registry.model.registry.label.ReservedListDao;
import google.registry.model.reporting.HistoryEntry; import google.registry.model.reporting.HistoryEntry;
@ -264,7 +264,7 @@ public class DatabaseHelper {
.setRestoreBillingCost(Money.of(USD, 17)) .setRestoreBillingCost(Money.of(USD, 17))
.setServerStatusChangeBillingCost(Money.of(USD, 19)) .setServerStatusChangeBillingCost(Money.of(USD, 19))
// Always set a default premium list. Tests that don't want it can delete it. // Always set a default premium list. Tests that don't want it can delete it.
.setPremiumList(persistPremiumList(tld, DEFAULT_PREMIUM_LIST_CONTENTS.get())) .setPremiumList(persistPremiumList(tld, USD, DEFAULT_PREMIUM_LIST_CONTENTS.get()))
.setPremiumPricingEngine(StaticPremiumListPricingEngine.NAME) .setPremiumPricingEngine(StaticPremiumListPricingEngine.NAME)
.setDnsWriters(ImmutableSet.of(VoidDnsWriter.NAME)) .setDnsWriters(ImmutableSet.of(VoidDnsWriter.NAME))
.build(); .build();
@ -361,7 +361,7 @@ public class DatabaseHelper {
.setName(listName) .setName(listName)
.setReservedListMapFromLines(ImmutableList.copyOf(lines)) .setReservedListMapFromLines(ImmutableList.copyOf(lines))
.setShouldPublish(shouldPublish) .setShouldPublish(shouldPublish)
.setLastUpdateTime(DateTime.now(DateTimeZone.UTC)) .setCreationTimestamp(DateTime.now(DateTimeZone.UTC))
.build(); .build();
return persistReservedList(reservedList); return persistReservedList(reservedList);
} }
@ -374,29 +374,27 @@ public class DatabaseHelper {
* incrementing FakeClock for all tests in order to persist the commit logs properly because of * incrementing FakeClock for all tests in order to persist the commit logs properly because of
* the requirement to have monotonically increasing timestamps. * the requirement to have monotonically increasing timestamps.
*/ */
public static PremiumList persistPremiumList(String listName, String... lines) { public static PremiumList persistPremiumList(
String listName, CurrencyUnit currencyUnit, String... lines) {
checkState(lines.length != 0, "Must provide at least one premium entry"); checkState(lines.length != 0, "Must provide at least one premium entry");
PremiumList partialPremiumList = new PremiumList.Builder().setName(listName).build(); PremiumList partialPremiumList = new PremiumList.Builder().setName(listName).build();
ImmutableMap<String, PremiumListEntry> entries = partialPremiumList.parse(asList(lines)); ImmutableMap<String, PremiumEntry> entries = partialPremiumList.parse(asList(lines));
CurrencyUnit currencyUnit =
entries.entrySet().iterator().next().getValue().getValue().getCurrencyUnit();
PremiumList premiumList = PremiumList premiumList =
partialPremiumList partialPremiumList
.asBuilder() .asBuilder()
.setCreationTime(DateTime.now(DateTimeZone.UTC)) .setCreationTimestamp(DateTime.now(DateTimeZone.UTC))
.setCurrency(currencyUnit) .setCurrency(currencyUnit)
.setLabelsToPrices( .setLabelsToPrices(
entries.entrySet().stream() entries.entrySet().stream()
.collect( .collect(
toImmutableMap( toImmutableMap(Map.Entry::getKey, entry -> entry.getValue().getValue())))
Map.Entry::getKey, entry -> entry.getValue().getValue().getAmount())))
.build(); .build();
// Since we used to persist a PremiumList to Datastore here, it is necessary to allocate an ID // Since we used to persist a PremiumList to Datastore here, it is necessary to allocate an ID
// here to prevent breaking some of the hard-coded flow tests. IDs in tests are allocated in a // here to prevent breaking some of the hard-coded flow tests. IDs in tests are allocated in a
// strictly increasing sequence, if we don't pad out the ID here, we would have to renumber // strictly increasing sequence, if we don't pad out the ID here, we would have to renumber
// hundreds of unit tests. // hundreds of unit tests.
allocateId(); allocateId();
jpaTm().transact(() -> jpaTm().insert(premiumList)); PremiumListDao.save(premiumList);
maybeAdvanceClock(); maybeAdvanceClock();
return premiumList; return premiumList;
} }
@ -1245,11 +1243,10 @@ public class DatabaseHelper {
return result; return result;
} }
/** Returns the entire map of {@link PremiumListEntry}s for the given {@link PremiumList}. */ /** Returns the entire map of {@link PremiumEntry}s for the given {@link PremiumList}. */
public static ImmutableMap<String, PremiumListEntry> loadPremiumListEntries( public static ImmutableMap<String, PremiumEntry> loadPremiumEntries(PremiumList premiumList) {
PremiumList premiumList) { return Streams.stream(PremiumListDao.loadAllPremiumEntries(premiumList.getName()))
return Streams.stream(PremiumListDao.loadAllPremiumListEntries(premiumList.getName())) .collect(toImmutableMap(PremiumEntry::getDomainLabel, Function.identity()));
.collect(toImmutableMap(PremiumListEntry::getLabel, Function.identity()));
} }
/** Loads and returns the registrar with the given client ID, or throws IAE if not present. */ /** Loads and returns the registrar with the given client ID, or throws IAE if not present. */

View file

@ -17,6 +17,7 @@ package google.registry.tools;
import static google.registry.testing.DatabaseHelper.createTld; import static google.registry.testing.DatabaseHelper.createTld;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static org.joda.money.CurrencyUnit.JPY;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import com.beust.jcommander.ParameterException; import com.beust.jcommander.ParameterException;
@ -76,7 +77,7 @@ class CreateAnchorTenantCommandTest extends EppToolCommandTestCase<CreateAnchorT
persistResource( persistResource(
Registry.get("tld") Registry.get("tld")
.asBuilder() .asBuilder()
.setPremiumList(persistPremiumList("tld", "premium,JPY 20000")) .setPremiumList(persistPremiumList("tld", JPY, "premium,JPY 20000"))
.build()); .build());
runCommandForced("--client=NewRegistrar", "--superuser", "--fee", runCommandForced("--client=NewRegistrar", "--superuser", "--fee",
"--reason=anchor-tenant-test", "--contact=jd1234", "--domain_name=premium.tld"); "--reason=anchor-tenant-test", "--contact=jd1234", "--domain_name=premium.tld");

View file

@ -19,6 +19,7 @@ import static google.registry.persistence.transaction.TransactionManagerFactory.
import static google.registry.testing.DatabaseHelper.createTld; import static google.registry.testing.DatabaseHelper.createTld;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static org.joda.money.CurrencyUnit.JPY;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import com.beust.jcommander.ParameterException; import com.beust.jcommander.ParameterException;
@ -121,7 +122,7 @@ class CreateDomainCommandTest extends EppToolCommandTestCase<CreateDomainCommand
@Test @Test
void testSuccess_premiumJpyDomain() throws Exception { void testSuccess_premiumJpyDomain() throws Exception {
createTld("baar"); createTld("baar");
persistPremiumList("baar", "parajiumu,JPY 96083"); persistPremiumList("baar", JPY, "parajiumu,JPY 96083");
persistResource( persistResource(
Registry.get("baar") Registry.get("baar")
.asBuilder() .asBuilder()

View file

@ -32,6 +32,7 @@ import google.registry.model.registry.Registry;
import google.registry.model.registry.Registry.TldType; import google.registry.model.registry.Registry.TldType;
import java.io.File; import java.io.File;
import java.io.IOException; import java.io.IOException;
import org.joda.money.CurrencyUnit;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.BeforeEach;
/** Base class for common testing setup for create and update commands for Premium Lists. */ /** Base class for common testing setup for create and update commands for Premium Lists. */
@ -51,7 +52,7 @@ abstract class CreateOrUpdatePremiumListCommandTestCase<T extends CreateOrUpdate
premiumTermsPath = premiumTermsFile.getPath(); premiumTermsPath = premiumTermsFile.getPath();
} }
Registry createRegistry(String tldStr, String premiumListInput) { Registry createRegistry(String tldStr, CurrencyUnit currency, String premiumListInput) {
Registry registry; Registry registry;
if (premiumListInput != null) { if (premiumListInput != null) {
registry = registry =
@ -60,7 +61,7 @@ abstract class CreateOrUpdatePremiumListCommandTestCase<T extends CreateOrUpdate
Ascii.toUpperCase(tldStr), Ascii.toUpperCase(tldStr),
ImmutableSortedMap.of(START_OF_TIME, GENERAL_AVAILABILITY), ImmutableSortedMap.of(START_OF_TIME, GENERAL_AVAILABILITY),
TldType.TEST); TldType.TEST);
persistPremiumList(tldStr, premiumListInput); persistPremiumList(tldStr, currency, premiumListInput);
persistResource(registry); persistResource(registry);
} else { } else {
registry = registry =

View file

@ -107,7 +107,7 @@ abstract class CreateOrUpdateReservedListCommandTestCase<
ImmutableMap<String, ReservedListEntry> labelsToEntries) { ImmutableMap<String, ReservedListEntry> labelsToEntries) {
return new ReservedList.Builder() return new ReservedList.Builder()
.setName(name) .setName(name)
.setLastUpdateTime(creationTime) .setCreationTimestamp(creationTime)
.setShouldPublish(shouldPublish) .setShouldPublish(shouldPublish)
.setReservedListMap(labelsToEntries) .setReservedListMap(labelsToEntries)
.build(); .build();

View file

@ -33,7 +33,7 @@ class CreatePremiumListCommandTest<C extends CreatePremiumListCommand>
@BeforeEach @BeforeEach
void beforeEach() { void beforeEach() {
registry = createRegistry(TLD_TEST, null); registry = createRegistry(TLD_TEST, null, null);
} }
@Test @Test
@ -45,7 +45,7 @@ class CreatePremiumListCommandTest<C extends CreatePremiumListCommand>
@Test @Test
void commandRun_successCreateList() throws Exception { void commandRun_successCreateList() throws Exception {
runCommandForced("--name=" + TLD_TEST, "--input=" + premiumTermsPath); runCommandForced("--name=" + TLD_TEST, "--input=" + premiumTermsPath, "--currency=USD");
assertThat(registry.getTld().toString()).isEqualTo(TLD_TEST); assertThat(registry.getTld().toString()).isEqualTo(TLD_TEST);
assertThat(PremiumListDao.getLatestRevision(TLD_TEST).isPresent()).isTrue(); assertThat(PremiumListDao.getLatestRevision(TLD_TEST).isPresent()).isTrue();
} }
@ -53,50 +53,52 @@ class CreatePremiumListCommandTest<C extends CreatePremiumListCommand>
@Test @Test
// since the old entity is always null and file cannot be empty, the prompt will NOT be "No entity // since the old entity is always null and file cannot be empty, the prompt will NOT be "No entity
// changes to apply." // changes to apply."
void commandInit_successStageNewEntity() throws Exception { void commandPrompt_successStageNewEntity() throws Exception {
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
command.inputFile = Paths.get(premiumTermsPath); command.inputFile = Paths.get(premiumTermsPath);
command.init(); command.currencyUnit = "USD";
assertThat(command.prompt()).contains("Create PremiumList@"); command.prompt();
assertThat(command.prompt()).contains(String.format("name=%s", TLD_TEST)); assertThat(command.prompt()).isEqualTo("Create new premium list for prime?");
} }
@Test @Test
void commandInit_successStageNewEntityWithOverride() throws Exception { void commandPrompt_successStageNewEntityWithOverride() throws Exception {
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
String alterTld = "override"; String alterTld = "override";
command.inputFile = Paths.get(premiumTermsPath); command.inputFile = Paths.get(premiumTermsPath);
command.override = true; command.override = true;
command.name = alterTld; command.name = alterTld;
command.init(); command.currencyUnit = "USD";
assertThat(command.prompt()).contains("Create PremiumList@"); command.prompt();
assertThat(command.prompt()).contains(String.format("name=%s", alterTld)); assertThat(command.prompt()).isEqualTo("Create new premium list for override?");
} }
@Test @Test
void commandInit_failureNoInputFile() { void commandPrompt_failureNoInputFile() {
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
assertThrows(NullPointerException.class, command::init); assertThrows(NullPointerException.class, command::prompt);
} }
@Test @Test
void commandInit_failurePremiumListAlreadyExists() { void commandPrompt_failurePremiumListAlreadyExists() {
String randomStr = "random"; String randomStr = "random";
createTld(randomStr); createTld(randomStr);
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
command.name = randomStr; command.name = randomStr;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); command.currencyUnit = "USD";
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown).hasMessageThat().isEqualTo("A premium list already exists by this name"); assertThat(thrown).hasMessageThat().isEqualTo("A premium list already exists by this name");
} }
@Test @Test
void commandInit_failureMismatchedTldFileName_noOverride() throws Exception { void commandPrompt_failureMismatchedTldFileName_noOverride() throws Exception {
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
String fileName = "random"; String fileName = "random";
Path tmpPath = tmpDir.resolve(String.format("%s.txt", fileName)); Path tmpPath = tmpDir.resolve(String.format("%s.txt", fileName));
Files.write(new byte[0], tmpPath.toFile()); Files.write(new byte[0], tmpPath.toFile());
command.inputFile = tmpPath; command.inputFile = tmpPath;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); command.currencyUnit = "USD";
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown)
.hasMessageThat() .hasMessageThat()
.contains( .contains(
@ -108,11 +110,12 @@ class CreatePremiumListCommandTest<C extends CreatePremiumListCommand>
} }
@Test @Test
void commandInit_failureMismatchedTldName_noOverride() { void commandPrompt_failureMismatchedTldName_noOverride() {
CreatePremiumListCommand command = new CreatePremiumListCommand(); CreatePremiumListCommand command = new CreatePremiumListCommand();
String fileName = "random"; String fileName = "random";
command.name = fileName; command.name = fileName;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); command.currencyUnit = "USD";
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown)
.hasMessageThat() .hasMessageThat()
.contains( .contains(
@ -122,18 +125,4 @@ class CreatePremiumListCommandTest<C extends CreatePremiumListCommand>
+ "yet TLD %s does not exist", + "yet TLD %s does not exist",
fileName)); fileName));
} }
@Test
void commandInit_failureUseEmptyFile() throws Exception {
CreatePremiumListCommand command = new CreatePremiumListCommand();
String fileName = "empty";
Path tmpPath = tmpDir.resolve(String.format("%s.txt", fileName));
Files.write(new byte[0], tmpPath.toFile());
command.inputFile = tmpPath;
command.name = TLD_TEST;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init);
assertThat(thrown)
.hasMessageThat()
.contains("The Cloud SQL schema requires exactly one currency");
}
} }

View file

@ -49,7 +49,7 @@ class CreateTldCommandTest extends CommandTestCase<CreateTldCommand> {
persistReservedList("xn--q9jyb4c_abuse", "lamb,FULLY_BLOCKED"); persistReservedList("xn--q9jyb4c_abuse", "lamb,FULLY_BLOCKED");
persistReservedList("tld_banned", "kilo,FULLY_BLOCKED", "lima,FULLY_BLOCKED"); persistReservedList("tld_banned", "kilo,FULLY_BLOCKED", "lima,FULLY_BLOCKED");
persistReservedList("soy_expurgated", "fireflies,FULLY_BLOCKED"); persistReservedList("soy_expurgated", "fireflies,FULLY_BLOCKED");
persistPremiumList("xn--q9jyb4c", "minecraft,USD 1000"); persistPremiumList("xn--q9jyb4c", USD, "minecraft,USD 1000");
command.validDnsWriterNames = ImmutableSet.of("VoidDnsWriter", "FooDnsWriter"); command.validDnsWriterNames = ImmutableSet.of("VoidDnsWriter", "FooDnsWriter");
} }

View file

@ -17,9 +17,10 @@ package google.registry.tools;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth8.assertThat; import static com.google.common.truth.Truth8.assertThat;
import static google.registry.testing.DatabaseHelper.createTld; import static google.registry.testing.DatabaseHelper.createTld;
import static google.registry.testing.DatabaseHelper.loadPremiumListEntries; import static google.registry.testing.DatabaseHelper.loadPremiumEntries;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static org.joda.money.CurrencyUnit.USD;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
@ -32,8 +33,8 @@ class DeletePremiumListCommandTest extends CommandTestCase<DeletePremiumListComm
@Test @Test
void testSuccess() throws Exception { void testSuccess() throws Exception {
PremiumList premiumList = persistPremiumList("xn--q9jyb4c", "blah,USD 100"); PremiumList premiumList = persistPremiumList("xn--q9jyb4c", USD, "blah,USD 100");
assertThat(loadPremiumListEntries(premiumList)).hasSize(1); assertThat(loadPremiumEntries(premiumList)).hasSize(1);
runCommand("--force", "--name=xn--q9jyb4c"); runCommand("--force", "--name=xn--q9jyb4c");
assertThat(PremiumListDao.getLatestRevision("xn--q9jyb4c")).isEmpty(); assertThat(PremiumListDao.getLatestRevision("xn--q9jyb4c")).isEmpty();
} }
@ -49,7 +50,7 @@ class DeletePremiumListCommandTest extends CommandTestCase<DeletePremiumListComm
@Test @Test
void testFailure_whenPremiumListIsInUse() { void testFailure_whenPremiumListIsInUse() {
PremiumList premiumList = persistPremiumList("xn--q9jyb4c", "blah,USD 100"); PremiumList premiumList = persistPremiumList("xn--q9jyb4c", USD, "blah,USD 100");
createTld("xn--q9jyb4c"); createTld("xn--q9jyb4c");
persistResource(Registry.get("xn--q9jyb4c").asBuilder().setPremiumList(premiumList).build()); persistResource(Registry.get("xn--q9jyb4c").asBuilder().setPremiumList(premiumList).build());
IllegalArgumentException thrown = IllegalArgumentException thrown =

View file

@ -24,6 +24,7 @@ import static google.registry.testing.DatabaseHelper.createTld;
import static google.registry.testing.DatabaseHelper.loadRegistrar; import static google.registry.testing.DatabaseHelper.loadRegistrar;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static google.registry.testing.DatabaseHelper.persistResource; import static google.registry.testing.DatabaseHelper.persistResource;
import static org.joda.money.CurrencyUnit.USD;
import static org.joda.time.DateTimeZone.UTC; import static org.joda.time.DateTimeZone.UTC;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
@ -39,7 +40,6 @@ import google.registry.testing.DeterministicStringGenerator;
import google.registry.testing.FakeClock; import google.registry.testing.FakeClock;
import google.registry.util.CidrAddressBlock; import google.registry.util.CidrAddressBlock;
import java.security.cert.CertificateParsingException; import java.security.cert.CertificateParsingException;
import org.joda.money.CurrencyUnit;
import org.joda.money.Money; import org.joda.money.Money;
import org.joda.time.DateTime; import org.joda.time.DateTime;
import org.joda.time.Duration; import org.joda.time.Duration;
@ -58,7 +58,7 @@ class SetupOteCommandTest extends CommandTestCase<SetupOteCommand> {
void beforeEach() { void beforeEach() {
command.passwordGenerator = passwordGenerator; command.passwordGenerator = passwordGenerator;
command.clock = new FakeClock(DateTime.parse("2018-07-07TZ")); command.clock = new FakeClock(DateTime.parse("2018-07-07TZ"));
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
} }
/** Verify TLD creation. */ /** Verify TLD creation. */
@ -79,17 +79,17 @@ class SetupOteCommandTest extends CommandTestCase<SetupOteCommand> {
ImmutableSortedMap<DateTime, Money> eapFeeSchedule = registry.getEapFeeScheduleAsMap(); ImmutableSortedMap<DateTime, Money> eapFeeSchedule = registry.getEapFeeScheduleAsMap();
if (!isEarlyAccess) { if (!isEarlyAccess) {
assertThat(eapFeeSchedule) assertThat(eapFeeSchedule)
.isEqualTo(ImmutableSortedMap.of(new DateTime(0), Money.of(CurrencyUnit.USD, 0))); .isEqualTo(ImmutableSortedMap.of(new DateTime(0), Money.of(USD, 0)));
} else { } else {
assertThat(eapFeeSchedule) assertThat(eapFeeSchedule)
.isEqualTo( .isEqualTo(
ImmutableSortedMap.of( ImmutableSortedMap.of(
new DateTime(0), new DateTime(0),
Money.of(CurrencyUnit.USD, 0), Money.of(USD, 0),
DateTime.parse("2018-03-01T00:00:00Z"), DateTime.parse("2018-03-01T00:00:00Z"),
Money.of(CurrencyUnit.USD, 100), Money.of(USD, 100),
DateTime.parse("2030-03-01T00:00:00Z"), DateTime.parse("2030-03-01T00:00:00Z"),
Money.of(CurrencyUnit.USD, 0))); Money.of(USD, 0)));
} }
} }

View file

@ -16,15 +16,18 @@ package google.registry.tools;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static java.nio.charset.StandardCharsets.UTF_8; import static java.nio.charset.StandardCharsets.UTF_8;
import static org.joda.money.CurrencyUnit.USD;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import com.google.common.collect.ImmutableSet; import com.google.common.collect.ImmutableSet;
import com.google.common.io.Files; import com.google.common.io.Files;
import google.registry.model.registry.Registry; import google.registry.model.registry.Registry;
import google.registry.model.registry.label.PremiumList;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
import java.io.File; import java.io.File;
import java.nio.file.Path; import java.nio.file.Path;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.util.Optional;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
@ -35,40 +38,32 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
@BeforeEach @BeforeEach
void beforeEach() { void beforeEach() {
registry = createRegistry(TLD_TEST, initialPremiumListData); registry = createRegistry(TLD_TEST, USD, initialPremiumListData);
} }
@Test @Test
void verify_registryIsSetUpCorrectly() { void verify_registryIsSetUpCorrectly() {
Optional<PremiumList> list = PremiumListDao.getLatestRevision(TLD_TEST);
// ensure that no premium list is created before running the command // ensure that no premium list is created before running the command
assertThat(PremiumListDao.getLatestRevision(TLD_TEST).isPresent()).isTrue(); assertThat(list.isPresent()).isTrue();
// ensure that there's value in existing premium list; // ensure that there's value in existing premium list;
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
ImmutableSet<String> entries = command.getExistingPremiumListEntry(TLD_TEST); ImmutableSet<String> entries = command.getExistingPremiumEntry(list.get());
assertThat(entries.size()).isEqualTo(1); assertThat(entries.size()).isEqualTo(1);
// data from @beforeEach of CreateOrUpdatePremiumListCommandTestCase.java // data from @beforeEach of CreateOrUpdatePremiumListCommandTestCase.java
assertThat(entries.contains("doge,USD 9090.00")).isTrue(); assertThat(entries.contains("doge,USD 9090.00")).isTrue();
} }
@Test @Test
void commandInit_successStageNoEntityChange() throws Exception { void commandPrompt_successStageEntityChange() throws Exception {
UpdatePremiumListCommand command = new UpdatePremiumListCommand();
command.inputFile = Paths.get(premiumTermsPath);
command.name = TLD_TEST;
command.init();
assertThat(command.prompt()).contains("No entity changes to apply.");
}
@Test
void commandInit_successStageEntityChange() throws Exception {
File tmpFile = tmpDir.resolve(String.format("%s.txt", TLD_TEST)).toFile(); File tmpFile = tmpDir.resolve(String.format("%s.txt", TLD_TEST)).toFile();
String newPremiumListData = "omg,JPY 1234"; String newPremiumListData = "omg,USD 1234";
Files.asCharSink(tmpFile, UTF_8).write(newPremiumListData); Files.asCharSink(tmpFile, UTF_8).write(newPremiumListData);
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
command.inputFile = Paths.get(tmpFile.getPath()); command.inputFile = Paths.get(tmpFile.getPath());
command.name = TLD_TEST; command.name = TLD_TEST;
command.init(); command.prompt();
assertThat(command.prompt()).contains("Update PremiumList@"); assertThat(command.prompt()).contains("Update premium list for prime?");
} }
@Test @Test
@ -82,7 +77,8 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
command.inputFile = Paths.get(tmpFile.getPath()); command.inputFile = Paths.get(tmpFile.getPath());
runCommandForced("--name=" + TLD_TEST, "--input=" + command.inputFile); runCommandForced("--name=" + TLD_TEST, "--input=" + command.inputFile);
ImmutableSet<String> entries = command.getExistingPremiumListEntry(TLD_TEST); ImmutableSet<String> entries =
command.getExistingPremiumEntry(PremiumListDao.getLatestRevision(TLD_TEST).get());
assertThat(entries.size()).isEqualTo(1); assertThat(entries.size()).isEqualTo(1);
// verify that list is updated; cannot use only string since price is formatted; // verify that list is updated; cannot use only string since price is formatted;
assertThat(entries.contains("eth,USD 9999.00")).isTrue(); assertThat(entries.contains("eth,USD 9999.00")).isTrue();
@ -99,7 +95,8 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
runCommandForced("--name=" + TLD_TEST, "--input=" + command.inputFile); runCommandForced("--name=" + TLD_TEST, "--input=" + command.inputFile);
// assert all three lines from premiumTerms are added // assert all three lines from premiumTerms are added
ImmutableSet<String> entries = command.getExistingPremiumListEntry(TLD_TEST); ImmutableSet<String> entries =
command.getExistingPremiumEntry(PremiumListDao.getLatestRevision(TLD_TEST).get());
assertThat(entries.size()).isEqualTo(3); assertThat(entries.size()).isEqualTo(3);
assertThat(entries.contains("foo,USD 9000.00")).isTrue(); assertThat(entries.contains("foo,USD 9000.00")).isTrue();
assertThat(entries.contains("doge,USD 100.00")).isTrue(); assertThat(entries.contains("doge,USD 100.00")).isTrue();
@ -107,26 +104,24 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
} }
@Test @Test
void commandInit_failureUpdateEmptyList() throws Exception { void commandPrompt_failureUpdateEmptyList() throws Exception {
Path tmpPath = tmpDir.resolve(String.format("%s.txt", TLD_TEST)); Path tmpPath = tmpDir.resolve(String.format("%s.txt", TLD_TEST));
Files.write(new byte[0], tmpPath.toFile()); Files.write(new byte[0], tmpPath.toFile());
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
command.inputFile = tmpPath; command.inputFile = tmpPath;
command.name = TLD_TEST; command.name = TLD_TEST;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown).hasMessageThat().contains("Input cannot be empty");
.hasMessageThat()
.contains("The Cloud SQL schema requires exactly one currency");
} }
@Test @Test
void commandInit_failureNoPreviousVersion() { void commandPrompt_failureNoPreviousVersion() {
String fileName = "random"; String fileName = "random";
registry = createRegistry(fileName, null); registry = createRegistry(fileName, null, null);
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
command.name = fileName; command.name = fileName;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown)
.hasMessageThat() .hasMessageThat()
.isEqualTo( .isEqualTo(
@ -134,17 +129,17 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
} }
@Test @Test
void commandInit_failureNoInputFile() { void commandPrompt_failureNoInputFile() {
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
assertThrows(NullPointerException.class, command::init); assertThrows(NullPointerException.class, command::prompt);
} }
@Test @Test
void commandInit_failureTldFromNameDoesNotExist() { void commandPrompt_failureTldFromNameDoesNotExist() {
String fileName = "random"; String fileName = "random";
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
command.name = fileName; command.name = fileName;
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown)
.hasMessageThat() .hasMessageThat()
.isEqualTo( .isEqualTo(
@ -152,13 +147,13 @@ class UpdatePremiumListCommandTest<C extends UpdatePremiumListCommand>
} }
@Test @Test
void commandInit_failureTldFromInputFileDoesNotExist() { void commandPrompt_failureTldFromInputFileDoesNotExist() {
String fileName = "random"; String fileName = "random";
UpdatePremiumListCommand command = new UpdatePremiumListCommand(); UpdatePremiumListCommand command = new UpdatePremiumListCommand();
// using tld extracted from file name but this tld is not part of the registry // using tld extracted from file name but this tld is not part of the registry
command.inputFile = command.inputFile =
Paths.get(tmpDir.resolve(String.format("%s.txt", fileName)).toFile().getPath()); Paths.get(tmpDir.resolve(String.format("%s.txt", fileName)).toFile().getPath());
IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::init); IllegalArgumentException thrown = assertThrows(IllegalArgumentException.class, command::prompt);
assertThat(thrown) assertThat(thrown)
.hasMessageThat() .hasMessageThat()
.isEqualTo( .isEqualTo(

View file

@ -48,8 +48,7 @@ class UpdateReservedListCommandTest
new ReservedList.Builder() new ReservedList.Builder()
.setName("xn--q9jyb4c_common-reserved") .setName("xn--q9jyb4c_common-reserved")
.setReservedListMapFromLines(ImmutableList.of("helicopter,FULLY_BLOCKED")) .setReservedListMapFromLines(ImmutableList.of("helicopter,FULLY_BLOCKED"))
.setCreationTime(START_OF_TIME) .setCreationTimestamp(START_OF_TIME)
.setLastUpdateTime(START_OF_TIME)
.setShouldPublish(shouldPublish) .setShouldPublish(shouldPublish)
.build()); .build());
} }

View file

@ -54,7 +54,7 @@ class UpdateTldCommandTest extends CommandTestCase<UpdateTldCommand> {
persistReservedList("xn--q9jyb4c_abuse", "lamb,FULLY_BLOCKED"); persistReservedList("xn--q9jyb4c_abuse", "lamb,FULLY_BLOCKED");
persistReservedList("tld_banned", "kilo,FULLY_BLOCKED", "lima,FULLY_BLOCKED"); persistReservedList("tld_banned", "kilo,FULLY_BLOCKED", "lima,FULLY_BLOCKED");
persistReservedList("soy_expurgated", "fireflies,FULLY_BLOCKED"); persistReservedList("soy_expurgated", "fireflies,FULLY_BLOCKED");
persistPremiumList("xn--q9jyb4c", "minecraft,USD 1000"); persistPremiumList("xn--q9jyb4c", USD, "minecraft,USD 1000");
persistReservedList("xn--q9jyb4c_r1", "foo,FULLY_BLOCKED"); persistReservedList("xn--q9jyb4c_r1", "foo,FULLY_BLOCKED");
persistReservedList("xn--q9jyb4c_r2", "moop,FULLY_BLOCKED"); persistReservedList("xn--q9jyb4c_r2", "moop,FULLY_BLOCKED");
createTld("xn--q9jyb4c"); createTld("xn--q9jyb4c");

View file

@ -17,8 +17,9 @@ package google.registry.tools.server;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth8.assertThat; import static com.google.common.truth.Truth8.assertThat;
import static google.registry.testing.DatabaseHelper.createTlds; import static google.registry.testing.DatabaseHelper.createTlds;
import static google.registry.testing.DatabaseHelper.loadPremiumListEntries; import static google.registry.testing.DatabaseHelper.loadPremiumEntries;
import static javax.servlet.http.HttpServletResponse.SC_OK; import static javax.servlet.http.HttpServletResponse.SC_OK;
import static org.joda.money.CurrencyUnit.USD;
import google.registry.model.registry.label.PremiumList; import google.registry.model.registry.label.PremiumList;
import google.registry.schema.tld.PremiumListDao; import google.registry.schema.tld.PremiumListDao;
@ -86,20 +87,21 @@ public class CreatePremiumListActionTest {
action.name = "zanzibar"; action.name = "zanzibar";
action.inputData = "zanzibar,USD 100"; action.inputData = "zanzibar,USD 100";
action.override = true; action.override = true;
action.currency = USD;
action.run(); action.run();
assertThat(response.getStatus()).isEqualTo(SC_OK); assertThat(response.getStatus()).isEqualTo(SC_OK);
assertThat(loadPremiumListEntries(PremiumListDao.getLatestRevision("zanzibar").get())) assertThat(loadPremiumEntries(PremiumListDao.getLatestRevision("zanzibar").get())).hasSize(1);
.hasSize(1);
} }
@Test @Test
void test_success() { void test_success() {
action.name = "foo"; action.name = "foo";
action.inputData = "rich,USD 25\nricher,USD 1000\n"; action.inputData = "rich,USD 25\nricher,USD 1000\n";
action.currency = USD;
action.run(); action.run();
assertThat(response.getStatus()).isEqualTo(SC_OK); assertThat(response.getStatus()).isEqualTo(SC_OK);
PremiumList premiumList = PremiumListDao.getLatestRevision("foo").get(); PremiumList premiumList = PremiumListDao.getLatestRevision("foo").get();
assertThat(loadPremiumListEntries(premiumList)).hasSize(2); assertThat(loadPremiumEntries(premiumList)).hasSize(2);
assertThat(PremiumListDao.getPremiumPrice(premiumList.getName(), "rich")) assertThat(PremiumListDao.getPremiumPrice(premiumList.getName(), "rich"))
.hasValue(Money.parse("USD 25")); .hasValue(Money.parse("USD 25"));
assertThat(PremiumListDao.getPremiumPrice(premiumList.getName(), "diamond")).isEmpty(); assertThat(PremiumListDao.getPremiumPrice(premiumList.getName(), "diamond")).isEmpty();

View file

@ -15,6 +15,8 @@
package google.registry.tools.server; package google.registry.tools.server;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static org.joda.money.CurrencyUnit.JPY;
import static org.joda.money.CurrencyUnit.USD;
import java.util.Optional; import java.util.Optional;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.BeforeEach;
@ -27,8 +29,8 @@ class ListPremiumListsActionTest extends ListActionTestCase {
@BeforeEach @BeforeEach
void beforeEach() { void beforeEach() {
persistPremiumList("xn--q9jyb4c", "rich,USD 100"); persistPremiumList("xn--q9jyb4c", USD, "rich,USD 100");
persistPremiumList("how", "richer,JPY 5000"); persistPremiumList("how", JPY, "richer,JPY 5000");
action = new ListPremiumListsAction(); action = new ListPremiumListsAction();
} }

View file

@ -17,9 +17,10 @@ package google.registry.tools.server;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static google.registry.persistence.transaction.TransactionManagerFactory.jpaTm; import static google.registry.persistence.transaction.TransactionManagerFactory.jpaTm;
import static google.registry.testing.DatabaseHelper.createTlds; import static google.registry.testing.DatabaseHelper.createTlds;
import static google.registry.testing.DatabaseHelper.loadPremiumListEntries; import static google.registry.testing.DatabaseHelper.loadPremiumEntries;
import static google.registry.util.ResourceUtils.readResourceUtf8; import static google.registry.util.ResourceUtils.readResourceUtf8;
import static javax.servlet.http.HttpServletResponse.SC_OK; import static javax.servlet.http.HttpServletResponse.SC_OK;
import static org.joda.money.CurrencyUnit.USD;
import com.google.common.base.Splitter; import com.google.common.base.Splitter;
import com.google.common.truth.Truth8; import com.google.common.truth.Truth8;
@ -79,12 +80,12 @@ class UpdatePremiumListActionTest {
.omitEmptyStrings() .omitEmptyStrings()
.splitToList( .splitToList(
readResourceUtf8(DatabaseHelper.class, "default_premium_list_testdata.csv")); readResourceUtf8(DatabaseHelper.class, "default_premium_list_testdata.csv"));
PremiumListDao.save("foo", inputLines); PremiumListDao.save("foo", USD, inputLines);
action.name = "foo"; action.name = "foo";
action.inputData = "rich,USD 75\nricher,USD 5000\npoor, USD 0.99"; action.inputData = "rich,USD 75\nricher,USD 5000\npoor, USD 0.99";
action.run(); action.run();
assertThat(response.getStatus()).isEqualTo(SC_OK); assertThat(response.getStatus()).isEqualTo(SC_OK);
assertThat(loadPremiumListEntries(PremiumListDao.getLatestRevision("foo").get())).hasSize(3); assertThat(loadPremiumEntries(PremiumListDao.getLatestRevision("foo").get())).hasSize(3);
Truth8.assertThat(PremiumListDao.getPremiumPrice("foo", "rich")) Truth8.assertThat(PremiumListDao.getPremiumPrice("foo", "rich"))
.hasValue(Money.parse("USD 75")); .hasValue(Money.parse("USD 75"));
Truth8.assertThat(PremiumListDao.getPremiumPrice("foo", "richer")) Truth8.assertThat(PremiumListDao.getPremiumPrice("foo", "richer"))

View file

@ -20,6 +20,7 @@ import static com.google.common.truth.Truth8.assertThat;
import static google.registry.model.registrar.Registrar.loadByClientId; import static google.registry.model.registrar.Registrar.loadByClientId;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static javax.servlet.http.HttpServletResponse.SC_MOVED_TEMPORARILY; import static javax.servlet.http.HttpServletResponse.SC_MOVED_TEMPORARILY;
import static org.joda.money.CurrencyUnit.USD;
import static org.junit.jupiter.api.Assertions.assertThrows; import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@ -78,7 +79,7 @@ public final class ConsoleOteSetupActionTest {
@BeforeEach @BeforeEach
void beforeEach() throws Exception { void beforeEach() throws Exception {
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
action.req = request; action.req = request;
action.method = Method.GET; action.method = Method.GET;

View file

@ -20,6 +20,7 @@ import static google.registry.model.common.GaeUserIdConverter.convertEmailAddres
import static google.registry.model.registrar.Registrar.loadByClientId; import static google.registry.model.registrar.Registrar.loadByClientId;
import static google.registry.testing.DatabaseHelper.persistPremiumList; import static google.registry.testing.DatabaseHelper.persistPremiumList;
import static javax.servlet.http.HttpServletResponse.SC_MOVED_TEMPORARILY; import static javax.servlet.http.HttpServletResponse.SC_MOVED_TEMPORARILY;
import static org.joda.money.CurrencyUnit.USD;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@ -81,7 +82,7 @@ final class ConsoleRegistrarCreatorActionTest {
@BeforeEach @BeforeEach
void beforeEach() throws Exception { void beforeEach() throws Exception {
persistPremiumList("default_sandbox_list", "sandbox,USD 1000"); persistPremiumList("default_sandbox_list", USD, "sandbox,USD 1000");
action.req = request; action.req = request;
action.method = Method.GET; action.method = Method.GET;
@ -198,8 +199,7 @@ final class ConsoleRegistrarCreatorActionTest {
Registrar registrar = loadByClientId("myclientid").orElse(null); Registrar registrar = loadByClientId("myclientid").orElse(null);
assertThat(registrar).isNotNull(); assertThat(registrar).isNotNull();
assertThat(registrar.getClientId()).isEqualTo("myclientid"); assertThat(registrar.getClientId()).isEqualTo("myclientid");
assertThat(registrar.getBillingAccountMap()) assertThat(registrar.getBillingAccountMap()).containsExactly(USD, "billing-account");
.containsExactly(CurrencyUnit.USD, "billing-account");
assertThat(registrar.getDriveFolderId()).isEqualTo("drive-id"); assertThat(registrar.getDriveFolderId()).isEqualTo("drive-id");
assertThat(registrar.getIanaIdentifier()).isEqualTo(12321L); assertThat(registrar.getIanaIdentifier()).isEqualTo(12321L);
@ -296,9 +296,12 @@ final class ConsoleRegistrarCreatorActionTest {
assertThat(registrar).isNotNull(); assertThat(registrar).isNotNull();
assertThat(registrar.getBillingAccountMap()) assertThat(registrar.getBillingAccountMap())
.containsExactly( .containsExactly(
CurrencyUnit.JPY, "billing-account-yen", CurrencyUnit.JPY,
CurrencyUnit.USD, "billing-account-usd", "billing-account-yen",
CurrencyUnit.EUR, "billing-account-eur"); USD,
"billing-account-usd",
CurrencyUnit.EUR,
"billing-account-eur");
} }
@TestOfyAndSql @TestOfyAndSql

View file

@ -15,9 +15,6 @@ LastSqlTransaction
Modification Modification
OneTime OneTime
PollMessage PollMessage
PremiumList
PremiumListEntry
PremiumListRevision
RdeRevision RdeRevision
Recurring Recurring
Registrar Registrar

View file

@ -2,9 +2,6 @@ ClaimsList
ClaimsListSingleton ClaimsListSingleton
Cursor Cursor
DatabaseMigrationStateSchedule DatabaseMigrationStateSchedule
PremiumList
PremiumListEntry
PremiumListRevision
Registrar Registrar
RegistrarContact RegistrarContact
Registry Registry

View file

@ -11,9 +11,6 @@ HostResource
Modification Modification
OneTime OneTime
PollMessage PollMessage
PremiumList
PremiumListEntry
PremiumListRevision
Recurring Recurring
Registrar Registrar
RegistrarContact RegistrarContact

View file

@ -711,24 +711,6 @@ enum google.registry.model.registry.Registry$TldType {
REAL; REAL;
TEST; TEST;
} }
class google.registry.model.registry.label.PremiumList {
@Id java.lang.String name;
@Parent com.googlecode.objectify.Key<google.registry.model.common.EntityGroupRoot> parent;
com.googlecode.objectify.Key<google.registry.model.registry.label.PremiumList$PremiumListRevision> revisionKey;
org.joda.time.DateTime creationTime;
org.joda.time.DateTime lastUpdateTime;
}
class google.registry.model.registry.label.PremiumList$PremiumListEntry {
@Id java.lang.String label;
@Parent com.googlecode.objectify.Key<google.registry.model.registry.label.PremiumList$PremiumListRevision> parent;
java.lang.String comment;
org.joda.money.Money price;
}
class google.registry.model.registry.label.PremiumList$PremiumListRevision {
@Id long revisionId;
@Parent com.googlecode.objectify.Key<google.registry.model.registry.label.PremiumList> parent;
com.google.common.hash.BloomFilter<java.lang.String> probablePremiumLabels;
}
enum google.registry.model.registry.label.ReservationType { enum google.registry.model.registry.label.ReservationType {
ALLOWED_IN_SUNRISE; ALLOWED_IN_SUNRISE;
FULLY_BLOCKED; FULLY_BLOCKED;
@ -741,11 +723,10 @@ class google.registry.model.registry.label.ReservedList {
@Parent com.googlecode.objectify.Key<google.registry.model.common.EntityGroupRoot> parent; @Parent com.googlecode.objectify.Key<google.registry.model.common.EntityGroupRoot> parent;
boolean shouldPublish; boolean shouldPublish;
java.util.Map<java.lang.String, google.registry.model.registry.label.ReservedList$ReservedListEntry> reservedListMap; java.util.Map<java.lang.String, google.registry.model.registry.label.ReservedList$ReservedListEntry> reservedListMap;
org.joda.time.DateTime creationTime; org.joda.time.DateTime creationTimestamp;
org.joda.time.DateTime lastUpdateTime;
} }
class google.registry.model.registry.label.ReservedList$ReservedListEntry { class google.registry.model.registry.label.ReservedList$ReservedListEntry {
@Id java.lang.String label; @Id java.lang.String domainLabel;
google.registry.model.registry.label.ReservationType reservationType; google.registry.model.registry.label.ReservationType reservationType;
java.lang.Long revisionId; java.lang.Long revisionId;
java.lang.String comment; java.lang.String comment;