add: added testcases for GetFileExtension utils

base/component_test.go

@@ -1,8 +1,13 @@
 package base
 
 import (
+	"bufio"
 	_ "embed"
+	"encoding/base64"
 	"encoding/json"
+	"fmt"
+	"io"
+	"os"
 	"testing"
 
 	qt "github.com/frankban/quicktest"
@@ -48,3 +53,24 @@ func TestComponent_ListConnectorDefinitions(t *testing.T) {
 	c.Assert(err, qt.IsNil)
 	c.Check(gotJSON, qt.JSONEquals, wantConnectorDefinitionStruct)
 }
+
+func TestUtil_GetFileExtension(t *testing.T) {
+	c := qt.New(t)
+
+	file, err := os.Open("./testdata/test_image.png")
+	c.Assert(err, qt.IsNil)
+	defer file.Close()
+	wantFileExtension := "png"
+
+	reader := bufio.NewReader(file)
+	content, err := io.ReadAll(reader)
+	c.Assert(err, qt.IsNil)
+
+	fileBase64 := base64.StdEncoding.EncodeToString(content)
+	fileBase64 = "data:image/png;base64," + fileBase64
+	fmt.Println(fileBase64)
+	gotFileExtension := GetBase64FileExtensionSlow(fileBase64)
+	c.Check(gotFileExtension, qt.Equals, wantFileExtension)
+	gotFileExtension = GetBase64FileExtensionFast(fileBase64)
+	c.Check(gotFileExtension, qt.Equals, wantFileExtension)
+}

base/formats.go

@@ -2,7 +2,6 @@ package base
 
 import (
 	"encoding/base64"
-	"fmt"
 	"mime"
 	"strings"
 
@@ -218,7 +217,6 @@ func GetBase64FileExtensionFast(b64 string) string {
 	header := splitB64[0]
 	header = strings.TrimPrefix(header, "data:")
 	header = strings.TrimSuffix(header, ";base64")
-	fmt.Println("### header:", header)
 	mtype, _, err := mime.ParseMediaType(header)
 	if err != nil {
 		return err.Error()