added unit test

Signed-off-by: Shashank <[email protected]>

Author: moki1202

pkg/configs/db/db.go

@@ -59,6 +59,21 @@ type DB interface {
 	Close() error
 }
 
+func SetUserPassword(u *url.URL, passwordFilePath string) (*url.URL, error) {
+	if u.User == nil {
+		return nil, fmt.Errorf("--database.password-file requires username in --database.uri")
+	}
+
+	passwordBytes, err := os.ReadFile(passwordFilePath)
+	if err != nil {
+		return nil, fmt.Errorf("Could not read database password file: %v", err)
+	}
+
+	password := strings.TrimSpace(string(passwordBytes))
+	u.User = url.UserPassword(u.User.Username(), password)
+	return u, nil
+}
+
 // New creates a new database.
 func New(cfg Config) (DB, error) {
 	if cfg.Mock != nil {
@@ -71,15 +86,11 @@ func New(cfg Config) (DB, error) {
 	}
 
 	if len(cfg.PasswordFile) != 0 {
-		if u.User == nil {
-			return nil, fmt.Errorf("--database.password-file requires username in --database.uri")
-		}
-		passwordBytes, err := os.ReadFile(cfg.PasswordFile)
+		updatedUrl, err := SetUserPassword(u, cfg.PasswordFile)
 		if err != nil {
-			return nil, fmt.Errorf("Could not read database password file: %v", err)
+			return nil, err
 		}
-		password := strings.TrimSpace(string(passwordBytes))
-		u.User = url.UserPassword(u.User.Username(), password)
+		u = updatedUrl
 	}
 
 	var d DB

pkg/configs/db/dbtest/unit.go

@@ -4,6 +4,9 @@
 package dbtest
 
 import (
+	"io/ioutil"
+	"net/url"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -26,3 +29,48 @@ func Setup(t *testing.T) db.DB {
 func Cleanup(t *testing.T, database db.DB) {
 	require.NoError(t, database.Close())
 }
+
+func TestUserPasswordFromPasswordFile(t *testing.T) {
+
+	tempFile, _ := ioutil.TempFile("", "testpassword")
+	defer tempFile.Close()
+	defer os.Remove(tempFile.Name())
+
+	_, writeErr := tempFile.WriteString("   testpassword   ")
+	if writeErr != nil {
+		t.Fatalf("Error writing to temporary file: %v", writeErr)
+	}
+
+	testCases := []struct {
+		testName         string
+		url              string
+		expectedUsername string
+		expectedPassword string
+	}{
+		{
+			testName:         "set username and password correctly from url and password file",
+			url:              "testscheme://[email protected]",
+			expectedUsername: "testuser",
+			expectedPassword: "testpassword",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.testName, func(t *testing.T) {
+			u, _ := url.Parse(tc.url)
+
+			updatedUrl, err := db.SetUserPassword(u, tempFile.Name())
+			if err != nil {
+				t.Errorf("Unexpected error: %v", err)
+			}
+
+			if u.User == nil || updatedUrl.User.Username() != tc.expectedUsername {
+				t.Errorf("Username not set as expected. Got: %v, Expected: %v", u.User, tc.expectedUsername)
+			}
+
+			if trimmedPassword, _ := updatedUrl.User.Password(); trimmedPassword != tc.expectedPassword {
+				t.Errorf("Password not set as expected. Got: %v, Expected: %v", trimmedPassword, tc.expectedPassword)
+			}
+		})
+	}
+}