kea/src/lib/datasrc/tests/zonetable_unittest.cc

// Copyright (C) 2010  Internet Systems Consortium, Inc. ("ISC")
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
// OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

#include <exceptions/exceptions.h>

#include <dns/name.h>
#include <dns/rrclass.h>

#include <datasrc/zonetable.h>

#include <gtest/gtest.h>

using namespace isc::dns;
using namespace isc::datasrc;

namespace {
TEST(ZoneTest, init) {
    Zone zone(RRClass::IN(), Name("example.com"));
    EXPECT_EQ(Name("example.com"), zone.getOrigin());
    EXPECT_EQ(RRClass::IN(), zone.getClass());

    Zone ch_zone(RRClass::CH(), Name("example"));
    EXPECT_EQ(Name("example"), ch_zone.getOrigin());
    EXPECT_EQ(RRClass::CH(), ch_zone.getClass());
}

class ZoneTableTest : public ::testing::Test {
protected:
    ZoneTableTest() : zone1(new Zone(RRClass::IN(), Name("example.com"))),
                      zone2(new Zone(RRClass::IN(), Name("example.net"))),
                      zone3(new Zone(RRClass::IN(), Name("example")))
    {}
    ZoneTable zone_table;
    ZonePtr zone1, zone2, zone3;
};

TEST_F(ZoneTableTest, add) {
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone1));
    EXPECT_EQ(ZoneTable::EXIST, zone_table.add(zone1));
    // names are compared in a case insensitive manner.
    EXPECT_EQ(ZoneTable::EXIST, zone_table.add(
                  ZonePtr(new Zone(RRClass::IN(), Name("EXAMPLE.COM")))));

    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone2));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone3));

    // Zone table is indexed only by name.  Duplicate origin name with
    // different zone class isn't allowed.
    EXPECT_EQ(ZoneTable::EXIST, zone_table.add(
                  ZonePtr(new Zone(RRClass::CH(), Name("example.com")))));

    /// Bogus zone (NULL)
    EXPECT_THROW(zone_table.add(ZonePtr()), isc::InvalidParameter);
}

TEST_F(ZoneTableTest, remove) {
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone1));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone2));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone3));

    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.remove(Name("example.net")));
    EXPECT_EQ(ZoneTable::NOTFOUND, zone_table.remove(Name("example.net")));
}

TEST_F(ZoneTableTest, find) {
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone1));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone2));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone3));

    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.find(Name("example.com")).code);
    EXPECT_EQ(Name("example.com"),
              zone_table.find(Name("example.com")).zone->getOrigin());

    EXPECT_EQ(ZoneTable::NOTFOUND,
              zone_table.find(Name("example.org")).code);
    EXPECT_EQ(static_cast<const Zone*>(NULL),
              zone_table.find(Name("example.org")).zone);

    // there's no exact match.  the result should be the longest match,
    // and the code should be PARTIALMATCH.
    EXPECT_EQ(ZoneTable::PARTIALMATCH,
              zone_table.find(Name("www.example.com")).code);
    EXPECT_EQ(Name("example.com"),
              zone_table.find(Name("www.example.com")).zone->getOrigin());

    // make sure the partial match is indeed the longest match by adding
    // a zone with a shorter origin and query again.
    ZonePtr zone_com(new Zone(RRClass::IN(), Name("com")));
    EXPECT_EQ(ZoneTable::SUCCESS, zone_table.add(zone_com));
    EXPECT_EQ(Name("example.com"),
              zone_table.find(Name("www.example.com")).zone->getOrigin());
}
}