-- Prosody IM

-- Copyright (C) 2011-2013 Florian Zeitz

--

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

function source(source)

	local new_ip = require"util.ip".new_ip;

	assert_equal(source(new_ip("2001:db8:1::1", "IPv6"),

			{new_ip("2001:db8:3::1", "IPv6"), new_ip("fe80::1", "IPv6")}).addr,

		"2001:db8:3::1",

		"prefer appropriate scope");

	assert_equal(source(new_ip("ff05::1", "IPv6"),

			{new_ip("2001:db8:3::1", "IPv6"), new_ip("fe80::1", "IPv6")}).addr,

		"2001:db8:3::1",

		"prefer appropriate scope");

	assert_equal(source(new_ip("2001:db8:1::1", "IPv6"),

			{new_ip("2001:db8:1::1", "IPv6"), new_ip("2001:db8:2::1", "IPv6")}).addr,

		"2001:db8:1::1",

		"prefer same address"); -- "2001:db8:1::1" should be marked "deprecated" here, we don't handle that right now

	assert_equal(source(new_ip("fe80::1", "IPv6"),

			{new_ip("fe80::2", "IPv6"), new_ip("2001:db8:1::1", "IPv6")}).addr,

		"fe80::2",

		"prefer appropriate scope"); -- "fe80::2" should be marked "deprecated" here, we don't handle that right now

	assert_equal(source(new_ip("2001:db8:1::1", "IPv6"),

			{new_ip("2001:db8:1::2", "IPv6"), new_ip("2001:db8:3::2", "IPv6")}).addr,

		"2001:db8:1::2",

		"longest matching prefix");

--[[ "2001:db8:1::2" should be a care-of address and "2001:db8:3::2" a home address, we can't handle this and would fail

	assert_equal(source(new_ip("2001:db8:1::1", "IPv6"),

			{new_ip("2001:db8:1::2", "IPv6"), new_ip("2001:db8:3::2", "IPv6")}).addr,

		"2001:db8:3::2",

		"prefer home address");

]]

	assert_equal(source(new_ip("2002:c633:6401::1", "IPv6"),

			{new_ip("2002:c633:6401::d5e3:7953:13eb:22e8", "IPv6"), new_ip("2001:db8:1::2", "IPv6")}).addr,

		"2002:c633:6401::d5e3:7953:13eb:22e8",

		"prefer matching label"); -- "2002:c633:6401::d5e3:7953:13eb:22e8" should be marked "temporary" here, we don't handle that right now

	assert_equal(source(new_ip("2001:db8:1::d5e3:0:0:1", "IPv6"),

			{new_ip("2001:db8:1::2", "IPv6"), new_ip("2001:db8:1::d5e3:7953:13eb:22e8", "IPv6")}).addr,

		"2001:db8:1::d5e3:7953:13eb:22e8",

		"prefer temporary address") -- "2001:db8:1::2" should be marked "public" and "2001:db8:1::d5e3:7953:13eb:22e8" should be marked "temporary" here, we don't handle that right now

end

function destination(dest)

	local order;

	local new_ip = require"util.ip".new_ip;

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("198.51.100.121", "IPv4")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("fe80::1", "IPv6"), new_ip("169.254.13.78", "IPv4")})

	assert_equal(order[1].addr, "2001:db8:1::1", "prefer matching scope");

	assert_equal(order[2].addr, "198.51.100.121", "prefer matching scope");

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("198.51.100.121", "IPv4")},

		{new_ip("fe80::1", "IPv6"), new_ip("198.51.100.117", "IPv4")})

	assert_equal(order[1].addr, "198.51.100.121", "prefer matching scope");

	assert_equal(order[2].addr, "2001:db8:1::1", "prefer matching scope");

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("10.1.2.3", "IPv4")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("fe80::1", "IPv6"), new_ip("10.1.2.4", "IPv4")})

	assert_equal(order[1].addr, "2001:db8:1::1", "prefer higher precedence");

	assert_equal(order[2].addr, "10.1.2.3", "prefer higher precedence");

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("fe80::1", "IPv6")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "fe80::1", "prefer smaller scope");

	assert_equal(order[2].addr, "2001:db8:1::1", "prefer smaller scope");

--[[ "2001:db8:1::2" and "fe80::2" should be marked "care-of address", while "2001:db8:3::1" should be marked "home address", we can't currently handle this and would fail the test

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("fe80::1", "IPv6")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("2001:db8:3::1", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "2001:db8:1::1", "prefer home address");

	assert_equal(order[2].addr, "fe80::1", "prefer home address");

]]

--[[ "fe80::2" should be marked "deprecated", we can't currently handle this and would fail the test

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("fe80::1", "IPv6")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "2001:db8:1::1", "avoid deprecated addresses");

	assert_equal(order[2].addr, "fe80::1", "avoid deprecated addresses");

]]

	order = dest({new_ip("2001:db8:1::1", "IPv6"), new_ip("2001:db8:3ffe::1", "IPv6")},

		{new_ip("2001:db8:1::2", "IPv6"), new_ip("2001:db8:3f44::2", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "2001:db8:1::1", "longest matching prefix");

	assert_equal(order[2].addr, "2001:db8:3ffe::1", "longest matching prefix");

	order = dest({new_ip("2002:c633:6401::1", "IPv6"), new_ip("2001:db8:1::1", "IPv6")},

		{new_ip("2002:c633:6401::2", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "2002:c633:6401::1", "prefer matching label");

	assert_equal(order[2].addr, "2001:db8:1::1", "prefer matching label");

	order = dest({new_ip("2002:c633:6401::1", "IPv6"), new_ip("2001:db8:1::1", "IPv6")},

		{new_ip("2002:c633:6401::2", "IPv6"), new_ip("2001:db8:1::2", "IPv6"), new_ip("fe80::2", "IPv6")})

	assert_equal(order[1].addr, "2001:db8:1::1", "prefer higher precedence");

	assert_equal(order[2].addr, "2002:c633:6401::1", "prefer higher precedence");

end