/*
 *    Copyright (c) 2021, The OpenThread Authors.
 *    All rights reserved.
 *
 *    Redistribution and use in source and binary forms, with or without
 *    modification, are permitted provided that the following conditions are met:
 *    1. Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *    2. Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *    3. Neither the name of the copyright holder nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 *    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *    AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *    IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *    ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 *    LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *    CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *    SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 *    INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 *    CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 *    ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *    POSSIBILITY OF SUCH DAMAGE.
 */

#include "common/dns_utils.hpp"

#include <assert.h>
#include <gtest/gtest.h>

static void CheckSplitFullDnsName(const std::string &aFullName,
                                  bool               aIsServiceInstance,
                                  bool               aIsService,
                                  bool               aIsHost,
                                  const std::string &aInstanceName,
                                  const std::string &aServiceName,
                                  const std::string &aHostName,
                                  const std::string &aDomain)
{
    DnsNameInfo info;

    assert(aFullName.empty() || aFullName.back() != '.');

    info = SplitFullDnsName(aFullName);

    EXPECT_EQ(aIsServiceInstance, info.IsServiceInstance());
    EXPECT_EQ(aIsService, info.IsService());
    EXPECT_EQ(aIsHost, info.IsHost());
    EXPECT_EQ(aInstanceName, info.mInstanceName);
    EXPECT_EQ(aServiceName, info.mServiceName);
    EXPECT_EQ(aHostName, info.mHostName);
    EXPECT_EQ(aDomain, info.mDomain);

    info = SplitFullDnsName(aFullName + ".");

    EXPECT_EQ(aIsServiceInstance, info.IsServiceInstance());
    EXPECT_EQ(aIsService, info.IsService());
    EXPECT_EQ(aIsHost, info.IsHost());
    EXPECT_EQ(aInstanceName, info.mInstanceName);
    EXPECT_EQ(aServiceName, info.mServiceName);
    EXPECT_EQ(aHostName, info.mHostName);
    EXPECT_EQ(aDomain, info.mDomain);
}

TEST(DnsUtils, TestSplitFullDnsName)
{
    // Check service instance names
    CheckSplitFullDnsName("ins1._ipps._tcp.default.service.arpa", true, false, false, "ins1", "_ipps._tcp", "",
                          "default.service.arpa.");
    CheckSplitFullDnsName("Instance Name._ipps._tcp.default.service.arpa", true, false, false, "Instance Name",
                          "_ipps._tcp", "", "default.service.arpa.");
    CheckSplitFullDnsName("Instance.Name.With.Dots._ipps._tcp.default.service.arpa", true, false, false,
                          "Instance.Name.With.Dots", "_ipps._tcp", "", "default.service.arpa.");

    // Check service names
    CheckSplitFullDnsName("_ipps._tcp.default.service.arpa", false, true, false, "", "_ipps._tcp", "",
                          "default.service.arpa.");
    CheckSplitFullDnsName("_meshcop._udp.default.service.arpa", false, true, false, "", "_meshcop._udp", "",
                          "default.service.arpa.");

    // Check invalid service names
    CheckSplitFullDnsName("_meshcop._abc.default.service.arpa", false, false, true, "", "", "_meshcop",
                          "_abc.default.service.arpa.");
    CheckSplitFullDnsName("_tcp.default.service.arpa", false, false, true, "", "", "_tcp", "default.service.arpa.");

    // Check host names
    CheckSplitFullDnsName("abc.example.com", false, false, true, "", "", "abc", "example.com.");
    CheckSplitFullDnsName("example.com", false, false, true, "", "", "example", "com.");
    CheckSplitFullDnsName("com", false, false, true, "", "", "com", ".");
    CheckSplitFullDnsName("", false, false, true, "", "", "", ".");
}