dati

commit a58d3b11313a15d3782bc3fba8bd06d695700c20
parent 4f88966e2722f480f01139fb723674b1dff58eab
Author: gearsix <gearsix@tuta.io>
Date:   Fri,  3 Feb 2023 16:19:26 +0000

updated data_test.go

Diffstat:
M
data_test.go
 | 
33
+++++++++++++++++++++++++++++++++

1 file changed, 33 insertions(+), 0 deletions(-)
diff --git a/data_test.go b/data_test.go
@@ -56,6 +56,35 @@ func TestIsSupportedDataLang(t *testing.T) {
 	}
 }
 
+func TestReadDataFormat(t *testing.T) {
+	exts := []string{
+		".json", "json", "JSON", ".JSON",
+		".yaml", "yaml", "YAML", ".YAML",
+		".toml", "toml", "TOML", ".TOML",
+		".misc", "-", ".", "",
+	}
+
+	for i, ext := range exts {
+		var target SupportedDataFormat
+		
+		if i < 4 {
+			target = JSON
+		} else if i < 8 {
+			target = YAML
+		} else if i < 12 {
+			target = TOML
+		}
+
+		if ReadDataFormat(ext) != target {
+			if target == "" {
+				t.Fatalf("%s is not a supported data language", ext)
+			} else {
+				t.Fatalf("%s did not return %s", ext, target)
+			}
+		}
+	}
+}
+
 var good = map[string]string{
 	"json": `{"eg":0}`,
 	"yaml": `eg: 0
@@ -127,6 +156,10 @@ func TestLoadData(t *testing.T) {
 }
 
 func TestLoadDataFilepath(t *testing.T) {
+	TestLoadDataFile(t)
+}
+
+func TestLoadDataFile(t *testing.T) {
 	var d interface{}
 	var e error
 	var p string