From 3f3c426240f47b0c505f19cd6111050a836ec18f Mon Sep 17 00:00:00 2001
From: Sarah Botwinick <sarahbot@google.com>
Date: Mon, 24 Apr 2023 15:23:19 -0400
Subject: [PATCH] Add premium name check for default token

---
 .../java/google/registry/flows/domain/DomainFlowUtils.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/core/src/main/java/google/registry/flows/domain/DomainFlowUtils.java b/core/src/main/java/google/registry/flows/domain/DomainFlowUtils.java
index 21ccf96c3..2f41c54ba 100644
--- a/core/src/main/java/google/registry/flows/domain/DomainFlowUtils.java
+++ b/core/src/main/java/google/registry/flows/domain/DomainFlowUtils.java
@@ -131,6 +131,7 @@ import google.registry.model.tld.label.ReservationType;
 import google.registry.model.tld.label.ReservedList;
 import google.registry.model.tmch.ClaimsList;
 import google.registry.persistence.VKey;
+import google.registry.pricing.PricingEngineProxy;
 import google.registry.tldconfig.idn.IdnLabelValidator;
 import google.registry.tools.DigestType;
 import google.registry.util.Idn;
@@ -1231,6 +1232,11 @@ public class DomainFlowUtils {
         // the list
         continue;
       }
+      // Don't apply token to premium names if token not valid for premiums
+      if (PricingEngineProxy.isDomainPremium(domainName, now)
+          && !token.get().shouldDiscountPremiums()) {
+        continue;
+      }
       // Only use the first valid token in the list
       return token;
     }