From 22c66af9141bd8e631fda7a1a92b2e18d59e30c2 Mon Sep 17 00:00:00 2001
From: Erik Ekman <yarrick@kryo.se>
Date: Sun, 28 Jan 2007 03:28:02 +0000
Subject: [PATCH] More dns tests

---
 tests/dns.c | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/tests/dns.c b/tests/dns.c
index 89bacb5..1a0cc57 100644
--- a/tests/dns.c
+++ b/tests/dns.c
@@ -30,19 +30,44 @@
 
 START_TEST(test_encode_hostname)
 {
+	char out[] = "\x06" "BADGER\x06" "BADGER\x04" "KRYO\x02" "SE\x00";
 	char buf[256];
 	int len;
 	int ret;
 
 	len = 256;
+
+	memset(buf, 0, 256);
+	ret = dns_encode_hostname("BADGER.BADGER.KRYO.SE", buf, len);
+	fail_unless(strncmp(buf, out, ret) == 0, "Happy flow failed");
+}
+END_TEST
 	
+START_TEST(test_encode_hostname_nodot)
+{
+	char buf[256];
+	int len;
+	int ret;
+
+	len = 256;
+
 	memset(buf, 0, 256);
 	ret = dns_encode_hostname(	// More than 63 chars between dots
 		"ABCDEFGHIJKLMNOPQRSTUVWXYZABCDEFGHIJKLMNOPQRSTUVWXYZ"
 		"ABCDEFGHIJKLMNOPQRSTUVWXYZABCDEFGHIJKLMNOPQRSTUVWXYZ"
 		, buf, len);
 	fail_unless(ret == -1, NULL);
+}
+END_TEST
 	
+START_TEST(test_encode_hostname_toolong)
+{
+	char buf[256];
+	int len;
+	int ret;
+
+	len = 256;
+
 	memset(buf, 0, 256);
 	ret = dns_encode_hostname(	// More chars than fits into array
 		"ABCDEFGHIJKLMNOPQRSTUVWXYZ.ABCDEFGHIJKLMNOPQRSTUVWXYZ."
@@ -65,6 +90,8 @@ test_dns_create_tests()
 
 	tc = tcase_create("Dns");
 	tcase_add_test(tc, test_encode_hostname);
+	tcase_add_test(tc, test_encode_hostname_nodot);
+	tcase_add_test(tc, test_encode_hostname_toolong);
 
 	return tc;
 }