haiku.git

/*
 * Copyright 2011, Axel Dörfler, axeld@pinc-software.de.
 * Distributed under the terms of the MIT License.
 */


#include "NetworkInterfaceTest.h"

#include <NetworkAddress.h>
#include <NetworkInterface.h>
#include <NetworkRoster.h>

#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>


NetworkInterfaceTest::NetworkInterfaceTest()
{
}


NetworkInterfaceTest::~NetworkInterfaceTest()
{
}


void
NetworkInterfaceTest::setUp()
{
	fInterface.SetTo("loopTest");
	BNetworkRoster::Default().RemoveInterface(fInterface);
		// just in case

	CPPUNIT_ASSERT(BNetworkRoster::Default().AddInterface(fInterface) == B_OK);
	CPPUNIT_ASSERT(fInterface.CountAddresses() == 1);
		// every interface has one unspec address
}


void
NetworkInterfaceTest::tearDown()
{
	CPPUNIT_ASSERT(BNetworkRoster::Default().RemoveInterface(fInterface)
		== B_OK);
}


void
NetworkInterfaceTest::TestUnset()
{
	BNetworkInterface unset("test");
	unset.Unset();
	CPPUNIT_ASSERT(unset.Name() == NULL || unset.Name()[0] == '\0');
}


void
NetworkInterfaceTest::TestFindAddress()
{
	fInterface.RemoveAddressAt(0);
		// Remove empty address

	// Add first

	BNetworkInterfaceAddress first;
	first.SetAddress(BNetworkAddress(AF_INET, "8.8.8.8"));

	CPPUNIT_ASSERT(fInterface.FindAddress(first.Address()) < 0);
	CPPUNIT_ASSERT(fInterface.AddAddress(first) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindAddress(first.Address()) == 0);

	// Add second

	BNetworkInterfaceAddress second;
	second.SetAddress(BNetworkAddress(AF_INET6, "::1"));

	CPPUNIT_ASSERT(fInterface.FindAddress(second.Address()) < 0);
	CPPUNIT_ASSERT(fInterface.AddAddress(second) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindAddress(second.Address()) >= 0);

	// Remove them again

	CPPUNIT_ASSERT(fInterface.RemoveAddress(first) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindAddress(first.Address()) < 0);
	CPPUNIT_ASSERT(fInterface.FindAddress(second.Address()) >= 0);

	CPPUNIT_ASSERT(fInterface.RemoveAddress(second.Address()) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindAddress(first.Address()) < 0);
	CPPUNIT_ASSERT(fInterface.FindAddress(second.Address()) < 0);
}


void
NetworkInterfaceTest::TestFindFirstAddress()
{
	fInterface.RemoveAddressAt(0);
		// Remove empty address

	// Add first

	BNetworkInterfaceAddress first;
	first.SetAddress(BNetworkAddress(AF_INET, "8.8.8.8"));

	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET) < 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET6) < 0);
	CPPUNIT_ASSERT(fInterface.AddAddress(first) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET) == 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET6) < 0);

	// Add second

	BNetworkInterfaceAddress second;
	second.SetAddress(BNetworkAddress(AF_INET6, "::1"));

	CPPUNIT_ASSERT(fInterface.AddAddress(second) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET) >= 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET6) >= 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET)
		!= fInterface.FindFirstAddress(AF_INET6));

	// Remove them again

	CPPUNIT_ASSERT(fInterface.RemoveAddress(first) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET) < 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET6) >= 0);

	CPPUNIT_ASSERT(fInterface.RemoveAddress(second.Address()) == B_OK);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET) < 0);
	CPPUNIT_ASSERT(fInterface.FindFirstAddress(AF_INET6) < 0);
}


/*static*/ void
NetworkInterfaceTest::AddTests(BTestSuite& parent)
{
	CppUnit::TestSuite& suite = *new CppUnit::TestSuite("NetworkInterfaceTest");

	suite.addTest(new CppUnit::TestCaller<NetworkInterfaceTest>(
		"NetworkInterfaceTest::TestUnset", &NetworkInterfaceTest::TestUnset));
	suite.addTest(new CppUnit::TestCaller<NetworkInterfaceTest>(
		"NetworkInterfaceTest::TestFindAddress",
		&NetworkInterfaceTest::TestFindAddress));
	suite.addTest(new CppUnit::TestCaller<NetworkInterfaceTest>(
		"NetworkInterfaceTest::TestFindFirstAddress",
		&NetworkInterfaceTest::TestFindFirstAddress));

	parent.addTest("NetworkInterfaceTest", &suite);
}