From 1c71487e5dd840bdcd30d6d0f7d9807648625292 Mon Sep 17 00:00:00 2001
From: larryruili <larryruili@google.com>
Date: Thu, 7 Jun 2018 13:19:48 -0700
Subject: [PATCH] Set EAP billing event to one-year period

This is consistent with how we treat RESTORE billing events as well- in
general, fees are considered to be amortized over the course of a year (by the
invoicing team).

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=199684843
---
 java/google/registry/flows/domain/DomainCreateFlow.java        | 1 +
 java/google/registry/model/billing/BillingEvent.java           | 3 ++-
 javatests/google/registry/flows/EppLifecycleDomainTest.java    | 1 +
 .../google/registry/flows/domain/DomainCreateFlowTest.java     | 1 +
 4 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/java/google/registry/flows/domain/DomainCreateFlow.java b/java/google/registry/flows/domain/DomainCreateFlow.java
index dab9a7a03..77d069478 100644
--- a/java/google/registry/flows/domain/DomainCreateFlow.java
+++ b/java/google/registry/flows/domain/DomainCreateFlow.java
@@ -525,6 +525,7 @@ public class DomainCreateFlow implements TransactionalFlow {
         .setReason(Reason.FEE_EARLY_ACCESS)
         .setTargetId(createBillingEvent.getTargetId())
         .setClientId(createBillingEvent.getClientId())
+        .setPeriodYears(1)
         .setCost(feesAndCredits.getEapCost())
         .setEventTime(createBillingEvent.getEventTime())
         .setBillingTime(createBillingEvent.getBillingTime())
diff --git a/java/google/registry/model/billing/BillingEvent.java b/java/google/registry/model/billing/BillingEvent.java
index f64deef29..4d3c364dc 100644
--- a/java/google/registry/model/billing/BillingEvent.java
+++ b/java/google/registry/model/billing/BillingEvent.java
@@ -304,13 +304,14 @@ public abstract class BillingEvent extends ImmutableObject
         ImmutableSet<Reason> reasonsWithPeriods =
             Sets.immutableEnumSet(
                 Reason.CREATE,
+                Reason.FEE_EARLY_ACCESS,
                 Reason.RENEW,
                 Reason.RESTORE,
                 Reason.TRANSFER);
         checkState(
             reasonsWithPeriods.contains(instance.reason) == (instance.periodYears != null),
             "Period years must be set if and only if reason is "
-                + "CREATE, RENEW, RESTORE or TRANSFER.");
+                + "CREATE, FEE_EARLY_ACCESS, RENEW, RESTORE or TRANSFER.");
         checkState(
             instance.getFlags().contains(Flag.SYNTHETIC)
                 == (instance.syntheticCreationTime != null),
diff --git a/javatests/google/registry/flows/EppLifecycleDomainTest.java b/javatests/google/registry/flows/EppLifecycleDomainTest.java
index 2da45646e..dbeb4c40f 100644
--- a/javatests/google/registry/flows/EppLifecycleDomainTest.java
+++ b/javatests/google/registry/flows/EppLifecycleDomainTest.java
@@ -305,6 +305,7 @@ public class EppLifecycleDomainTest extends EppTestCase {
             .setReason(Reason.FEE_EARLY_ACCESS)
             .setTargetId("example.tld")
             .setClientId("NewRegistrar")
+            .setPeriodYears(1)
             .setCost(Money.parse("USD 100.00"))
             .setEventTime(createTime)
             .setBillingTime(createTime.plus(Registry.get("tld").getRenewGracePeriodLength()))
diff --git a/javatests/google/registry/flows/domain/DomainCreateFlowTest.java b/javatests/google/registry/flows/domain/DomainCreateFlowTest.java
index cf564ce83..7cb97bd91 100644
--- a/javatests/google/registry/flows/domain/DomainCreateFlowTest.java
+++ b/javatests/google/registry/flows/domain/DomainCreateFlowTest.java
@@ -267,6 +267,7 @@ public class DomainCreateFlowTest extends ResourceFlowTestCase<DomainCreateFlow,
               .setReason(Reason.FEE_EARLY_ACCESS)
               .setTargetId(getUniqueIdFromCommand())
               .setClientId("TheRegistrar")
+              .setPeriodYears(1)
               .setCost(eapFee)
               .setEventTime(clock.nowUtc())
               .setBillingTime(billingTime)