/*
 * Copyright 2014, Haiku, inc.
 * Distributed under the terms of the MIT licence
 */


#include "GetAddrInfo.h"

#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>

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


GetAddrInfoTest::GetAddrInfoTest()
{
}


GetAddrInfoTest::~GetAddrInfoTest()
{
}


void
GetAddrInfoTest::EmptyTest()
{
	struct addrinfo* info;
	addrinfo hint = {0};
	hint.ai_family = AF_INET;
	hint.ai_flags = AI_PASSIVE;

	// Trying to resolve an empty host...
	int result = getaddrinfo("", NULL, &hint, &info);

	if (info)
		freeaddrinfo(info);

	// getaddrinfo shouldn't suggest that this may work better by trying again.
	CPPUNIT_ASSERT(result != EAI_AGAIN);
}


void
GetAddrInfoTest::AddrConfigTest()
{
	struct addrinfo *info = NULL, hints = {0};
	hints.ai_flags = AI_ADDRCONFIG;

	// localhost is guaranteed to have an IPv6 address.
	int result = getaddrinfo("localhost", NULL, &hints, &info);
	CPPUNIT_ASSERT(result == 0);
	CPPUNIT_ASSERT(info != NULL);

	bool check;
	for (struct addrinfo* iter = info; iter != NULL; iter = iter->ai_next) {
		// only IPv4 addresses should be returned as we don't have IPv6 routing.
		check = iter->ai_family == AF_INET;
		if (!check) break;
	}

	freeaddrinfo(info);
	CPPUNIT_ASSERT(check);
}


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

	suite.addTest(new CppUnit::TestCaller<GetAddrInfoTest>(
		"GetAddrInfoTest::EmptyTest", &GetAddrInfoTest::EmptyTest));
	suite.addTest(new CppUnit::TestCaller<GetAddrInfoTest>(
		"GetAddrInfoTest::AddrConfigTest", &GetAddrInfoTest::AddrConfigTest));

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