feat: add test cases

libdns · Nov 29, 2024 · b29d8eb · b29d8eb
1 parent 5a1e831
commit b29d8eb
Showing 1 changed file with 45 additions and 0 deletions.
diff --git a/utils_test.go b/utils_test.go
@@ -0,0 +1,45 @@
+package huaweicloud
+
+import (
+	"testing"
+)
+
+func TestPrepareRecordValueTXTWithQuotes(t *testing.T) {
+	result := prepareRecordValue("TXT", `"example"`)
+	expected := []string{`"example"`}
+	if len(result) != len(expected) || result[0] != expected[0] {
+		t.Errorf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestPrepareRecordValueTXTWithoutStartingQuote(t *testing.T) {
+	result := prepareRecordValue("TXT", `example"`)
+	expected := []string{`"example"`}
+	if len(result) != len(expected) || result[0] != expected[0] {
+		t.Errorf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestPrepareRecordValueTXTWithoutEndingQuote(t *testing.T) {
+	result := prepareRecordValue("TXT", `"example`)
+	expected := []string{`"example"`}
+	if len(result) != len(expected) || result[0] != expected[0] {
+		t.Errorf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestPrepareRecordValueTXTWithoutQuotes(t *testing.T) {
+	result := prepareRecordValue("TXT", `example`)
+	expected := []string{`"example"`}
+	if len(result) != len(expected) || result[0] != expected[0] {
+		t.Errorf("expected %v, got %v", expected, result)
+	}
+}
+
+func TestPrepareRecordValueNonTXT(t *testing.T) {
+	result := prepareRecordValue("A", `example`)
+	expected := []string{`example`}
+	if len(result) != len(expected) || result[0] != expected[0] {
+		t.Errorf("expected %v, got %v", expected, result)
+	}
+}