154 lines
5.2 KiB
C++
154 lines
5.2 KiB
C++
//
|
|
// Copyright (C) 2013 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
#include "shill/socket_info.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace shill {
|
|
|
|
namespace {
|
|
|
|
const unsigned char kIPAddress1[] = { 192, 168, 1, 1 };
|
|
const unsigned char kIPAddress2[] = { 192, 168, 1, 2 };
|
|
const unsigned char kIPAddress3[] = { 192, 168, 1, 3 };
|
|
const uint16_t kPort1 = 1000;
|
|
const uint16_t kPort2 = 2000;
|
|
const uint16_t kPort3 = 3000;
|
|
|
|
} // namespace
|
|
|
|
class SocketInfoTest : public testing::Test {
|
|
protected:
|
|
void ExpectSocketInfoEqual(const SocketInfo& info1, const SocketInfo& info2) {
|
|
EXPECT_EQ(info1.connection_state(), info2.connection_state());
|
|
EXPECT_TRUE(info1.local_ip_address().Equals(info2.local_ip_address()));
|
|
EXPECT_EQ(info1.local_port(), info2.local_port());
|
|
EXPECT_TRUE(info1.remote_ip_address().Equals(info2.remote_ip_address()));
|
|
EXPECT_EQ(info1.remote_port(), info2.remote_port());
|
|
EXPECT_EQ(info1.transmit_queue_value(), info2.transmit_queue_value());
|
|
EXPECT_EQ(info1.receive_queue_value(), info2.receive_queue_value());
|
|
EXPECT_EQ(info1.timer_state(), info2.timer_state());
|
|
}
|
|
};
|
|
|
|
TEST_F(SocketInfoTest, CopyConstructor) {
|
|
SocketInfo info(SocketInfo::kConnectionStateEstablished,
|
|
IPAddress(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress1, sizeof(kIPAddress1))),
|
|
kPort1,
|
|
IPAddress(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress2, sizeof(kIPAddress2))),
|
|
kPort2,
|
|
10,
|
|
20,
|
|
SocketInfo::kTimerStateRetransmitTimerPending);
|
|
|
|
SocketInfo info_copy(info);
|
|
ExpectSocketInfoEqual(info, info_copy);
|
|
}
|
|
|
|
TEST_F(SocketInfoTest, AssignmentOperator) {
|
|
SocketInfo info(SocketInfo::kConnectionStateEstablished,
|
|
IPAddress(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress1, sizeof(kIPAddress1))),
|
|
kPort1,
|
|
IPAddress(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress2, sizeof(kIPAddress2))),
|
|
kPort2,
|
|
10,
|
|
20,
|
|
SocketInfo::kTimerStateRetransmitTimerPending);
|
|
|
|
SocketInfo info_copy = info;
|
|
ExpectSocketInfoEqual(info, info_copy);
|
|
}
|
|
|
|
TEST_F(SocketInfoTest, IsSameSocketAs) {
|
|
IPAddress ip_address1(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress1, sizeof(kIPAddress1)));
|
|
IPAddress ip_address2(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress2, sizeof(kIPAddress2)));
|
|
IPAddress ip_address3(IPAddress::kFamilyIPv4,
|
|
ByteString(kIPAddress3, sizeof(kIPAddress3)));
|
|
|
|
SocketInfo info(SocketInfo::kConnectionStateEstablished,
|
|
ip_address1,
|
|
kPort1,
|
|
ip_address2,
|
|
kPort2,
|
|
0,
|
|
0,
|
|
SocketInfo::kTimerStateNoTimerPending);
|
|
|
|
// Differs only by local address.
|
|
EXPECT_FALSE(info.IsSameSocketAs(
|
|
SocketInfo(SocketInfo::kConnectionStateEstablished,
|
|
ip_address3,
|
|
kPort1,
|
|
ip_address2,
|
|
kPort2,
|
|
0,
|
|
0,
|
|
SocketInfo::kTimerStateNoTimerPending)));
|
|
|
|
// Differs only by local port.
|
|
EXPECT_FALSE(info.IsSameSocketAs(
|
|
SocketInfo(SocketInfo::kConnectionStateEstablished,
|
|
ip_address1,
|
|
kPort3,
|
|
ip_address2,
|
|
kPort2,
|
|
0,
|
|
0,
|
|
SocketInfo::kTimerStateNoTimerPending)));
|
|
|
|
// Differs only by remote address.
|
|
EXPECT_FALSE(info.IsSameSocketAs(
|
|
SocketInfo(SocketInfo::kConnectionStateEstablished,
|
|
ip_address1,
|
|
kPort1,
|
|
ip_address3,
|
|
kPort2,
|
|
0,
|
|
0,
|
|
SocketInfo::kTimerStateNoTimerPending)));
|
|
|
|
// Differs only by remote port.
|
|
EXPECT_FALSE(info.IsSameSocketAs(
|
|
SocketInfo(SocketInfo::kConnectionStateEstablished,
|
|
ip_address1,
|
|
kPort1,
|
|
ip_address2,
|
|
kPort3,
|
|
0,
|
|
0,
|
|
SocketInfo::kTimerStateNoTimerPending)));
|
|
|
|
// Only local address, local port, remote address, and remote port are
|
|
// identical.
|
|
EXPECT_TRUE(info.IsSameSocketAs(
|
|
SocketInfo(SocketInfo::kConnectionStateClosing,
|
|
ip_address1,
|
|
kPort1,
|
|
ip_address2,
|
|
kPort2,
|
|
10,
|
|
20,
|
|
SocketInfo::kTimerStateRetransmitTimerPending)));
|
|
}
|
|
|
|
} // namespace shill
|