Merge pull request #9 from Hidayathamir/feature/source-to-store-code

feat: source to store code

helper.go

@@ -275,3 +275,10 @@ func IsValidSource(source string) bool {
 
 	return false
 }
+
+func SourceToStoreCodeCCM(source string) (storeCode string) {
+	if source == constants.XSource_Enterprise {
+		return "en"
+	}
+	return source
+}

helper_test.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"reflect"
 	"testing"
+
+	"github.com/erajayatech/go-helper/constants"
 )
 
 func TestValidateStruct(t *testing.T) {
@@ -637,3 +639,36 @@ func TestIsSourcePaymentLink(t *testing.T) {
 		})
 	}
 }
+
+func TestSourceToStoreCodeCCM(t *testing.T) {
+	type args struct {
+		source string
+	}
+	tests := []struct {
+		name          string
+		args          args
+		wantStoreCode string
+	}{
+		{
+			name: "source enterprise should return en",
+			args: args{
+				source: constants.XSource_Enterprise,
+			},
+			wantStoreCode: "en",
+		},
+		{
+			name: "source except enterprise should return itself",
+			args: args{
+				source: constants.XSource_Eraspace,
+			},
+			wantStoreCode: constants.XSource_Eraspace,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if gotStoreCode := SourceToStoreCodeCCM(tt.args.source); gotStoreCode != tt.wantStoreCode {
+				t.Errorf("SourceToStoreCodeCCM() = %v, want %v", gotStoreCode, tt.wantStoreCode)
+			}
+		})
+	}
+}