From a606ffc4c7a61ae981450b9dcc5605f78bba7e07 Mon Sep 17 00:00:00 2001
From: Pablo Mazzini <pmazzini@gmail.com>
Date: Wed, 1 Aug 2018 19:12:37 +0200
Subject: rename dnscompress to rfc1035label

---
 rfc1035label/label_test.go | 66 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 rfc1035label/label_test.go

(limited to 'rfc1035label/label_test.go')

diff --git a/rfc1035label/label_test.go b/rfc1035label/label_test.go
new file mode 100644
index 0000000..f91110b
--- /dev/null
+++ b/rfc1035label/label_test.go
@@ -0,0 +1,66 @@
+package rfc1035label
+
+import (
+	"bytes"
+	"testing"
+)
+
+func TestLabelsFromBytes(t *testing.T) {
+	labels, err := LabelsFromBytes([]byte{
+		0x9, 's', 'l', 'a', 'c', 'k', 'w', 'a', 'r', 'e',
+		0x2, 'i', 't',
+		0x0,
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(labels) != 1 {
+		t.Fatalf("Invalid labels length. Expected: 1, got: %v", len(labels))
+	}
+	if labels[0] != "slackware.it" {
+		t.Fatalf("Invalid label. Expected: %v, got: %v'", "slackware.it", labels[0])
+	}
+}
+
+func TestLabelsFromBytesZeroLength(t *testing.T) {
+	labels, err := LabelsFromBytes([]byte{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(labels) != 0 {
+		t.Fatalf("Invalid labels length. Expected: 0, got: %v", len(labels))
+	}
+}
+
+func TestLabelsFromBytesInvalidLength(t *testing.T) {
+	labels, err := LabelsFromBytes([]byte{0x3, 0xaa, 0xbb}) // short length
+	if err == nil {
+		t.Fatal("Expected error, got nil")
+	}
+	if len(labels) != 0 {
+		t.Fatalf("Invalid labels length. Expected: 0, got: %v", len(labels))
+	}
+	if labels != nil {
+		t.Fatalf("Invalid label. Expected nil, got %v", labels)
+	}
+}
+
+func TestLabelToBytes(t *testing.T) {
+	encodedLabel := LabelToBytes("slackware.it")
+	expected := []byte{
+		0x9, 's', 'l', 'a', 'c', 'k', 'w', 'a', 'r', 'e',
+		0x2, 'i', 't',
+		0x0,
+	}
+	if !bytes.Equal(encodedLabel, expected) {
+		t.Fatalf("Invalid label. Expected: %v, got: %v", expected, encodedLabel)
+	}
+}
+
+func TestLabelToBytesZeroLength(t *testing.T) {
+	encodedLabel := LabelToBytes("")
+	expected := []byte{0}
+	if !bytes.Equal(encodedLabel, expected) {
+		t.Fatalf("Invalid label. Expected: %v, got: %v", expected, encodedLabel)
+	}
+}
-- 
cgit v1.2.3