Import code from internal repository to git

javatests/com/google/domain/registry/dns/PublishDnsUpdatesActionTest.java

@@ -0,0 +1,134 @@
+// Copyright 2016 Google Inc. 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 com.google.domain.registry.dns;
+
+import static com.google.domain.registry.testing.DatastoreHelper.createTld;
+import static com.google.domain.registry.testing.DatastoreHelper.persistActiveDomain;
+import static com.google.domain.registry.testing.DatastoreHelper.persistActiveSubordinateHost;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.verifyNoMoreInteractions;
+
+import com.google.common.collect.ImmutableSet;
+import com.google.domain.registry.dns.writer.api.DnsWriter;
+import com.google.domain.registry.model.domain.DomainResource;
+import com.google.domain.registry.model.ofy.Ofy;
+import com.google.domain.registry.testing.AppEngineRule;
+import com.google.domain.registry.testing.FakeClock;
+import com.google.domain.registry.testing.InjectRule;
+
+import org.joda.time.DateTime;
+import org.joda.time.Duration;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.junit.runner.RunWith;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import javax.inject.Provider;
+
+/** Unit tests for {@link WriteDnsTask}. */
+@RunWith(MockitoJUnitRunner.class)
+public class PublishDnsUpdatesActionTest {
+
+  @Rule
+  public final AppEngineRule appEngine = AppEngineRule.builder()
+      .withDatastore()
+      .withTaskQueue()
+      .build();
+
+  @Rule
+  public final InjectRule inject = new InjectRule();
+
+  @Rule
+  public final ExpectedException thrown = ExpectedException.none();
+
+  private final FakeClock clock = new FakeClock(DateTime.parse("1971-01-01TZ"));
+  private final DnsWriter dnsWriter = mock(DnsWriter.class);
+  private PublishDnsUpdatesAction action;
+
+  @Before
+  public void setUp() throws Exception {
+    inject.setStaticField(Ofy.class, "clock", clock);
+    createTld("xn--q9jyb4c");
+    DomainResource domain1 = persistActiveDomain("example.xn--q9jyb4c");
+    persistActiveSubordinateHost("ns1.example.xn--q9jyb4c", domain1);
+    persistActiveSubordinateHost("ns2.example.xn--q9jyb4c", domain1);
+    DomainResource domain2 = persistActiveDomain("example2.xn--q9jyb4c");
+    persistActiveSubordinateHost("ns1.example.xn--q9jyb4c", domain2);
+    clock.advanceOneMilli();
+  }
+
+  private PublishDnsUpdatesAction createAction(String tld) {
+    PublishDnsUpdatesAction action = new PublishDnsUpdatesAction();
+    action.timeout = Duration.standardSeconds(10);
+    action.tld = tld;
+    action.hosts = ImmutableSet.<String>of();
+    action.domains = ImmutableSet.<String>of();
+    action.writerProvider = new Provider<DnsWriter>() {
+      @Override
+      public DnsWriter get() {
+        return dnsWriter;
+      }};
+    return action;
+  }
+
+  @Test
+  public void testHost_published() throws Exception {
+    action = createAction("xn--q9jyb4c");
+    action.hosts = ImmutableSet.of("ns1.example.xn--q9jyb4c");
+    action.run();
+    verify(dnsWriter).publishHost("ns1.example.xn--q9jyb4c");
+    verify(dnsWriter).close();
+    verifyNoMoreInteractions(dnsWriter);
+  }
+
+  @Test
+  public void testDomain_published() throws Exception {
+    action = createAction("xn--q9jyb4c");
+    action.domains = ImmutableSet.of("example.xn--q9jyb4c");
+    action.run();
+    verify(dnsWriter).publishDomain("example.xn--q9jyb4c");
+    verify(dnsWriter).close();
+    verifyNoMoreInteractions(dnsWriter);
+  }
+
+  @Test
+  public void testHostAndDomain_published() throws Exception {
+    action = createAction("xn--q9jyb4c");
+    action.domains = ImmutableSet.of("example.xn--q9jyb4c", "example2.xn--q9jyb4c");
+    action.hosts = ImmutableSet.of(
+        "ns1.example.xn--q9jyb4c", "ns2.example.xn--q9jyb4c", "ns1.example2.xn--q9jyb4c");
+    action.run();
+    verify(dnsWriter).publishDomain("example.xn--q9jyb4c");
+    verify(dnsWriter).publishDomain("example2.xn--q9jyb4c");
+    verify(dnsWriter).publishHost("ns1.example.xn--q9jyb4c");
+    verify(dnsWriter).publishHost("ns2.example.xn--q9jyb4c");
+    verify(dnsWriter).publishHost("ns1.example2.xn--q9jyb4c");
+    verify(dnsWriter).close();
+    verifyNoMoreInteractions(dnsWriter);
+  }
+
+  @Test
+  public void testWrongTld_notPublished() throws Exception {
+    action = createAction("xn--q9jyb4c");
+    action.domains = ImmutableSet.of("example.com", "example2.com");
+    action.hosts = ImmutableSet.of("ns1.example.com", "ns2.example.com", "ns1.example2.com");
+    action.run();
+    verify(dnsWriter).close();
+    verifyNoMoreInteractions(dnsWriter);
+  }
+}