aboutsummaryrefslogtreecommitdiffstats
path: root/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/persistence/NameServiceQueueSerializerTest.java
blob: ad9ebd2a96897e9ef90b5e594237512019d420c3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.persistence;

import com.yahoo.config.provision.HostName;
import com.yahoo.config.provision.zone.ZoneId;
import com.yahoo.vespa.hosted.controller.api.integration.dns.LatencyAliasTarget;
import com.yahoo.vespa.hosted.controller.api.integration.dns.Record;
import com.yahoo.vespa.hosted.controller.api.integration.dns.RecordData;
import com.yahoo.vespa.hosted.controller.api.integration.dns.RecordName;
import com.yahoo.vespa.hosted.controller.api.integration.dns.WeightedDirectTarget;
import com.yahoo.vespa.hosted.controller.application.TenantAndApplicationId;
import com.yahoo.vespa.hosted.controller.dns.CreateRecord;
import com.yahoo.vespa.hosted.controller.dns.CreateRecords;
import com.yahoo.vespa.hosted.controller.dns.NameServiceQueue;
import com.yahoo.vespa.hosted.controller.dns.NameServiceRequest;
import com.yahoo.vespa.hosted.controller.dns.RemoveRecords;
import org.junit.jupiter.api.Test;

import java.util.List;
import java.util.Optional;

import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * @author mpolden
 */
public class NameServiceQueueSerializerTest {

    private final NameServiceQueueSerializer serializer = new NameServiceQueueSerializer();

    @Test
    void test_serialization() {
        Optional<TenantAndApplicationId> owner = Optional.of(TenantAndApplicationId.from("t", "a"));
        var record1 = new Record(Record.Type.CNAME, RecordName.from("cname.example.com"), RecordData.from("example.com"));
        var record2 = new Record(Record.Type.TXT, RecordName.from("txt.example.com"), RecordData.from("text"));
        var requests = List.<NameServiceRequest>of(
                new CreateRecord(owner, record1),
                new CreateRecords(owner, List.of(record2)),
                new CreateRecords(owner, List.of(new Record(Record.Type.ALIAS, RecordName.from("alias.example.com"),
                                                            new LatencyAliasTarget(HostName.of("alias1"),
                                                                                   "dns-zone-01",
                                                                                   ZoneId.from("prod", "us-north-1")).pack()),
                                                 new Record(Record.Type.ALIAS, RecordName.from("alias.example.com"),
                                                            new LatencyAliasTarget(HostName.of("alias2"),
                                                                                   "dns-zone-02",
                                                                                   ZoneId.from("prod", "us-north-2")).pack()),
                                                 new Record(Record.Type.ALIAS, RecordName.from("alias.example.com"),
                                                            new LatencyAliasTarget(HostName.of("alias2"),
                                                                                   "ignored",
                                                                                   ZoneId.from("prod", "us-south-1")).pack()))
                ),
                new CreateRecords(Optional.empty(), List.of(new Record(Record.Type.DIRECT, RecordName.from("direct.example.com"),
                                                                       new WeightedDirectTarget(RecordData.from("10.1.2.3"),
                                                                                                ZoneId.from("prod", "us-north-1"),
                                                                                                100).pack()))),
                new RemoveRecords(Optional.empty(), record1.type(), record1.name()),
                new RemoveRecords(owner, record2.type(), record2.name(), Optional.of(record2.data()))
        );

        var queue = new NameServiceQueue(requests);
        var serialized = serializer.fromSlime(serializer.toSlime(queue));
        assertEquals(List.copyOf(queue.requests()), List.copyOf(serialized.requests()));
    }

}