zone_test.rb

require "minitest/autorun"
require "test_helper"

describe Zone do
  subject do
    Zone.new("example.com", nil)
  end

  describe "soa" do
    it "should create host" do
      subject.soa ttl: "42m"
      _(subject.zonefile.output).must_include %w[42m IN SOA].join("\t")
    end
  end

  describe "a record" do
    it "should create host" do
      subject.a "127.0.0.1"
      _(subject.zonefile.a).must_equal [{class: "IN", name: "@", ttl: nil, host: "127.0.0.1"}]
    end

    it "should create host, ttl" do
      subject.a "127.0.0.1", 600
      _(subject.zonefile.a).must_equal [{class: "IN", name: "@", ttl: 600, host: "127.0.0.1"}]
    end

    it "should create name, host, ttl" do
      subject.a "www", "127.0.0.1", 600
      _(subject.zonefile.a).must_equal [{class: "IN", name: "www", ttl: 600, host: "127.0.0.1"}]
    end

    it "should create name with ipv4 and ipv6" do
      subject.a "www", "127.0.0.1", "::ffff:7f00:1"
      _(subject.zonefile.a).must_equal  [{class: "IN", name: "www", ttl: nil, host: "127.0.0.1"}]
      _(subject.zonefile.a4).must_equal [{class: "IN", name: "www", ttl: nil, host: "::ffff:7f00:1"}]
    end

    it "should create name with ipv4, ipv6 and TTL" do
      subject.a "www", "127.0.0.1", "::ffff:7f00:1", 600
      _(subject.zonefile.a).must_equal  [{class: "IN", name: "www", ttl: 600, host: "127.0.0.1"}]
      _(subject.zonefile.a4).must_equal [{class: "IN", name: "www", ttl: 600, host: "::ffff:7f00:1"}]
    end
  end

  describe "cname record" do
    it "without args" do
      assert_raises ArgumentError do
        subject.cname
      end
    end

    it "with name" do
      subject.cname "www"
      _(subject.zonefile.cname).must_equal [{class: "IN", name: "www", ttl: nil, host: "@"}]
    end

    it "with name, host" do
      subject.cname "www", "other-server."
      _(subject.zonefile.cname).must_equal [{class: "IN", name: "www", ttl: nil, host: "other-server."}]
    end

    it "with name, ttl" do
      subject.cname "www", 600
      _(subject.zonefile.cname).must_equal [{class: "IN", name: "www", ttl: 600, host: "@"}]
    end

    it "with name, ttl, host" do
      subject.cname "www", "other-server.", 600
      _(subject.zonefile.cname).must_equal [{class: "IN", name: "www", ttl: 600, host: "other-server."}]
    end
  end

  describe "nesting records" do
    it "under a" do
      subject.a "test", "127.0.0.1" do
        cname "www"
        cname "www2", 600
        txt "yada"
        txt "yada-yada", 60
      end

      _(subject.zonefile.a).must_equal     [{class: "IN", name: "test", ttl: nil, host: "127.0.0.1"}]
      _(subject.zonefile.cname).must_equal [{class: "IN", name: "www",  ttl: nil, host: "test"},
                                         {class: "IN", name: "www2", ttl: 600, host: "test"}]
      _(subject.zonefile.txt).must_equal   [{class: "IN", name: "test", ttl: nil, text: "yada"},
                                         {class: "IN", name: "test", ttl: 60,  text: "yada-yada"}]
    end
  end

  describe "mx record" do
    it "should create without args" do
      subject.mx
      _(subject.zonefile.mx).must_equal [{class: "IN", name: "@", ttl: nil, host: "@", pri: 10}]
    end

    it "should create with host" do
      subject.mx "mail"
      _(subject.zonefile.mx).must_equal [{class: "IN", name: "@", ttl: nil, host: "mail", pri: 10}]
    end

    it "should create with host, priority" do
      subject.mx "mail", 20
      _(subject.zonefile.mx).must_equal [{class: "IN", name: "@", ttl: nil, host: "mail", pri: 20}]
    end

    it "should create with host, priority, ttl " do
      subject.mx "mail", 20, 600
      _(subject.zonefile.mx).must_equal [{class: "IN", name: "@", ttl: 600, host: "mail", pri: 20}]
    end
  end

  describe "srv record" do
    it "without port" do
      assert_raises ArgumentError do
        subject.srv :ldap, :tcp, "ldap01"
      end
    end

    it "should create srv record" do
      subject.srv :ldap, :tcp, "ldap01", 389
      _(subject.zonefile.srv).must_equal [{class: "IN", name: "_ldap._tcp", ttl: nil, pri: 10, weight: 0, host: "ldap01", port: 389}]
    end

    it "should create srv record with ttl" do
      subject.srv :ldap, :tcp, "ldap01", 389, 600
      _(subject.zonefile.srv).must_equal [{class: "IN", name: "_ldap._tcp", ttl: 600, pri: 10, weight: 0, host: "ldap01", port: 389}]
    end

    it "should create srv record with pri and weight" do
      subject.srv :ldap, :tcp, "ldap01", 389, pri: 15, weight: 3
      _(subject.zonefile.srv).must_equal [{class: "IN", name: "_ldap._tcp", ttl: nil, pri: 15, weight: 3, host: "ldap01", port: 389}]
    end

    it "should create srv record with name" do
      subject.srv "foo", :ldap, :tcp, "ldap01", 389
      _(subject.zonefile.srv).must_equal [{class: "IN", name: "_ldap._tcp.foo", ttl: nil, pri: 10, weight: 0, host: "ldap01", port: 389}]
    end
  end

  describe "ptr record" do
    it "should create ptr record" do
      subject.ptr 127, "foobar.org"
      _(subject.zonefile.ptr).must_equal [{class: "IN", name: "127", ttl: nil, host: "foobar.org."}]
    end
  end

  describe "txt record" do
    it "should create txt record" do
      subject.txt "site-verification-token"
      _(subject.zonefile.txt).must_equal [{class: "IN", name: "@", ttl: nil, text: "site-verification-token"}]
    end

    it "should create txt record, with ttl" do
      subject.txt "site-verification-token", 60
      _(subject.zonefile.txt).must_equal [{class: "IN", name: "@", ttl: 60, text: "site-verification-token"}]
    end

    it "should create txt record, with host" do
      subject.txt "example.com.", "site-verification-token"
      _(subject.zonefile.txt).must_equal [{class: "IN", name: "example.com.", ttl: nil, text: "site-verification-token"}]
    end

    it "should create txt record, with ttl, host" do
      subject.txt "example.com.", "site-verification-token", 120
      _(subject.zonefile.txt).must_equal [{class: "IN", name: "example.com.", ttl: 120, text: "site-verification-token"}]
    end
  end

  describe "ptr6 record" do
    it "with a double colon" do
      assert_raises ArgumentError do
        subject.ptr6 "1319::1", "example.com"
      end
    end

    it "should create ptr record" do
      subject.ptr6 "1319:8a2e:0370:7344", "example.com"
      _(subject.zonefile.ptr).must_equal [{class: "IN", name: "4.4.3.7.0.7.3.0.e.2.a.8.9.1.3.1", ttl: nil, host: "example.com."}]
    end

    it "should create ptr record with filled zeros" do
      subject.ptr6 "1319:8a2e:70:1", "example.com"
      _(subject.zonefile.ptr).must_equal [{class: "IN", name: "1.0.0.0.0.7.0.0.e.2.a.8.9.1.3.1", ttl: nil, host: "example.com."}]
    end
  end

  describe "tlsa record" do
    it "should create tlsa record" do
      subject.tlsa "www", 443, :tcp, 3, 0, 1, "e31d9e402c6308273375b68297f7af207521238f0cd812622672f0f2ce67eb1c"
      _(subject.zonefile.tlsa).must_equal [{
        class:              "IN",
        name:               "_443._tcp.www",
        ttl:                nil,
        certificate_usage:  3,
        selector:           0,
        matching_type:      1,
        data:               "e31d9e402c6308273375b68297f7af207521238f0cd812622672f0f2ce67eb1c"
      }]
    end

    it "should create tlsa record without subdomain" do
      subject.tlsa 443, :tcp, 3, 0, 1, "e31d9e402c6308273375b68297f7af207521238f0cd812622672f0f2ce67eb1c", 3600
      _(subject.zonefile.tlsa).must_equal [{
        class:              "IN",
        name:               "_443._tcp",
        ttl:                3600,
        certificate_usage:  3,
        selector:           0,
        matching_type:      1,
        data:               "e31d9e402c6308273375b68297f7af207521238f0cd812622672f0f2ce67eb1c"
      }]
    end

    it "with invalid port" do
      assert_raises ArgumentError do
        subject.tlsa "www", 123456, :tcp, 3, 0, 1, "e31d9e402c6308273375b68297f7af207521238f0cd812622672f0f2ce67eb1c"
      end
    end
  end
end