initial v1.0-beta commit

fsd/util_test.go

@@ -0,0 +1,54 @@
+package fsd
+
+import (
+	"testing"
+)
+
+// TestCountFields verifies the countFields function.
+func TestCountFields(t *testing.T) {
+	tests := []struct {
+		packet []byte
+		want   int
+	}{
+		{[]byte(""), 1},
+		{[]byte("abc"), 1},
+		{[]byte("a:b"), 2},
+		{[]byte("a:b:c"), 3},
+		{[]byte("a:b:"), 3},
+		{[]byte(":a:b"), 3},
+		{[]byte(":"), 2},
+	}
+	for _, tt := range tests {
+		got := countFields(tt.packet)
+		if got != tt.want {
+			t.Errorf("countFields(%q) = %d, want %d", tt.packet, got, tt.want)
+		}
+	}
+}
+
+// TestGetField verifies the getField function.
+func TestGetField(t *testing.T) {
+	tests := []struct {
+		packet []byte
+		index  int
+		want   string
+	}{
+		{[]byte("a:b:c"), 0, "a"},
+		{[]byte("a:b:c"), 1, "b"},
+		{[]byte("a:b:c"), 2, "c"},
+		{[]byte("a:"), 0, "a"},
+		{[]byte("a:"), 1, ""},
+		{[]byte("a:"), 2, ""},
+		{[]byte(":a"), 0, ""},
+		{[]byte(":a"), 1, "a"},
+		{[]byte(""), 0, ""},
+		{[]byte(""), 1, ""},
+		{[]byte("a"), 0, "a"},
+	}
+	for _, tt := range tests {
+		got := string(getField(tt.packet, tt.index))
+		if got != tt.want {
+			t.Errorf("getField(%q, %d) = %q, want %q", tt.packet, tt.index, got, tt.want)
+		}
+	}
+}