# HG changeset patch
# User Matthew Wild <mwild1@gmail.com>
# Date 1368895441 -3600
# Node ID 3dc141acf381e65e2780e329d2a4a141ca65ff6b
# Parent  468d7a2f85bac9eaf4a767a39585075e093ef59f
test_util_ip: Add tests for IP matching

diff -r 468d7a2f85ba -r 3dc141acf381 tests/test_util_ip.lua
--- a/tests/test_util_ip.lua	Sat May 18 17:17:56 2013 +0100
+++ b/tests/test_util_ip.lua	Sat May 18 17:44:01 2013 +0100
@@ -1,7 +1,30 @@
 
-function test_match(match_ip) 
-	assert(match_ip("10.20.30.40", "10.0.0.0/8"));
-	assert(match_ip("80.244.94.84", "80.244.94.84"));
-	assert(match_ip("8.8.8.8", "8.8.0.0/16"));
-	assert(match_ip("8.8.4.4", "8.8.0.0/16"));
+function match(match)
+	local _ = require "util.ip".new_ip;
+	local ip = _"10.20.30.40";
+	assert_equal(match(ip, _"10.0.0.0", 8), true);
+	assert_equal(match(ip, _"10.0.0.0", 16), false);
+	assert_equal(match(ip, _"10.0.0.0", 24), false);
+	assert_equal(match(ip, _"10.0.0.0", 32), false);
+
+	assert_equal(match(ip, _"10.20.0.0", 8), true);
+	assert_equal(match(ip, _"10.20.0.0", 16), true);
+	assert_equal(match(ip, _"10.20.0.0", 24), false);
+	assert_equal(match(ip, _"10.20.0.0", 32), false);
+
+	assert_equal(match(ip, _"0.0.0.0", 32), false);
+	assert_equal(match(ip, _"0.0.0.0", 0), true);
+	assert_equal(match(ip, _"0.0.0.0"), false);
+
+	assert_equal(match(ip, _"10.0.0.0", 255), false, "excessive number of bits");
+	assert_equal(match(ip, _"10.0.0.0", -8), true, "negative number of bits");
+	assert_equal(match(ip, _"10.0.0.0", -32), true, "negative number of bits");
+	assert_equal(match(ip, _"10.0.0.0", 0), true, "zero bits");
+	assert_equal(match(ip, _"10.0.0.0"), false, "no specified number of bits (differing ip)");
+	assert_equal(match(ip, _"10.20.30.40"), true, "no specified number of bits (same ip)");
+
+	assert_equal(match(_"80.244.94.84", _"80.244.94.84"), true, "simple ip");
+
+	assert_equal(match(_"8.8.8.8", _"8.8.0.0", 16), true);
+	assert_equal(match(_"8.8.4.4", _"8.8.0.0", 16), true);
 end