You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

111 lines
3.4 KiB

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "discovery/dnssd/impl/instance_key.h"
#include <unordered_map>
#include "absl/hash/hash.h"
#include "discovery/dnssd/impl/service_key.h"
#include "discovery/dnssd/testing/fake_dns_record_factory.h"
#include "gtest/gtest.h"
namespace openscreen {
namespace discovery {
TEST(DnsSdInstanceKeyTest, TestInstanceKeyEquals) {
InstanceKey key1("instance", "_service._udp", "domain");
InstanceKey key2("instance", "_service._udp", "domain");
EXPECT_TRUE(key1 == key2);
EXPECT_FALSE(key1 != key2);
key1 = InstanceKey("instance", "_service2._udp", "domain");
EXPECT_FALSE(key1 == key2);
EXPECT_TRUE(key1 != key2);
key2 = InstanceKey("instance", "_service2._udp", "domain");
EXPECT_TRUE(key1 == key2);
EXPECT_FALSE(key1 != key2);
key1 = InstanceKey("instance", "_service2._udp", "domain2");
EXPECT_FALSE(key1 == key2);
EXPECT_TRUE(key1 != key2);
key2 = InstanceKey("instance", "_service2._udp", "domain2");
EXPECT_TRUE(key1 == key2);
EXPECT_FALSE(key1 != key2);
key1 = InstanceKey("instance2", "_service2._udp", "domain2");
EXPECT_FALSE(key1 == key2);
EXPECT_TRUE(key1 != key2);
key2 = InstanceKey("instance2", "_service2._udp", "domain2");
EXPECT_TRUE(key1 == key2);
EXPECT_FALSE(key1 != key2);
}
TEST(DnsSdInstanceKeyTest, TestIsInstanceOf) {
ServiceKey ptr("_service._udp", "domain");
InstanceKey svc("instance", "_service._udp", "domain");
EXPECT_EQ(svc, ptr);
svc = InstanceKey("other id", "_service._udp", "domain");
EXPECT_EQ(svc, ptr);
svc = InstanceKey("instance", "_service._udp", "domain2");
EXPECT_FALSE(svc == ptr);
ptr = ServiceKey("_service._udp", "domain2");
EXPECT_EQ(svc, ptr);
svc = InstanceKey("instance", "_service2._udp", "domain");
EXPECT_NE(svc, ptr);
ptr = ServiceKey("_service2._udp", "domain");
EXPECT_EQ(svc, ptr);
}
TEST(DnsSdInstanceKeyTest, InstanceKeyInMap) {
InstanceKey key{"instance", "_service._udp", "domain"};
InstanceKey key2{"instance", "_service._udp", "domain"};
InstanceKey key3{"instance2", "_service._udp", "domain"};
InstanceKey key4{"instance", "_service2._udp", "domain"};
InstanceKey key5{"instance", "_service2._udp", "domain2"};
std::unordered_map<InstanceKey, int32_t, absl::Hash<InstanceKey>> map;
map[key] = 1;
map[key2] = 2;
EXPECT_EQ(map[key], 2);
map[key3] = 3;
EXPECT_EQ(map[key], 2);
EXPECT_EQ(map[key3], 3);
map[key4] = 4;
EXPECT_EQ(map[key], 2);
EXPECT_EQ(map[key3], 3);
EXPECT_EQ(map[key4], 4);
map[key5] = 5;
EXPECT_EQ(map[key], 2);
EXPECT_EQ(map[key3], 3);
EXPECT_EQ(map[key4], 4);
EXPECT_EQ(map[key5], 5);
}
TEST(DnsSdInstanceKeyTest, CreateFromRecordTest) {
MdnsRecord record = FakeDnsRecordFactory::CreateFullyPopulatedSrvRecord();
InstanceKey key(record);
EXPECT_EQ(key.instance_id(), FakeDnsRecordFactory::kInstanceName);
EXPECT_EQ(key.service_id(), FakeDnsRecordFactory::kServiceName);
EXPECT_EQ(key.domain_id(), FakeDnsRecordFactory::kDomainName);
}
TEST(DnsSdInstanceKeyTest, GetNameTest) {
InstanceKey key("instance", "_service._udp", "domain");
DomainName expected{"instance", "_service", "_udp", "domain"};
EXPECT_EQ(expected, key.GetName());
key = InstanceKey("foo", "_bar._tcp", "local");
expected = DomainName{"foo", "_bar", "_tcp", "local"};
EXPECT_EQ(expected, key.GetName());
}
} // namespace discovery
} // namespace openscreen