Create password_validation_postgres_test.go

test/password_validation_postgres_test.go

@@ -0,0 +1,80 @@
+package main
+
+import (
+    "context"
+    "database/sql"
+    "fmt"
+    "log"
+    "testing"
+    _ "github.com/lib/pq" // PostgreSQL driver
+)
+
+const (
+    connStr = "user=youruser dbname=yourdb sslmode=disable" // replace with your actual connection string
+)
+
+func TestPasswordValidationWithPostgres(t *testing.T) {
+    // Connect to the database
+    db, err := sql.Open("postgres", connStr)
+    if err != nil {
+        t.Fatalf("Failed to connect to database: %v", err)
+    }
+    defer db.Close()
+
+    ctx := context.Background()
+
+    // Create a table for testing
+    _, err = db.ExecContext(ctx, `
+        CREATE TEMPORARY TABLE users (
+            id SERIAL PRIMARY KEY,
+            username VARCHAR(255),
+            password VARCHAR(255)
+        )
+    `)
+    if err != nil {
+        t.Fatalf("Failed to create temporary table: %v", err)
+    }
+
+    // Define test cases
+    tests := []struct {
+        password   string
+        expectErr  bool
+    }{
+        {"short", true},                // Password too short
+        {"ValidP@ssw0rd", false},       // Valid password
+        {"WithoutUpperCase1!", true},   // Missing uppercase letter
+        {"withoutlowercase1!", true},   // Missing lowercase letter
+        {"WithoutDigit!", true},        // Missing digit
+        {"WithoutSpecial1", true},      // Missing special character
+        {string(make([]byte, maxPasswordLength+1)), true}, // Password too long
+    }
+
+    for _, tt := range tests {
+        t.Run(fmt.Sprintf("password: %s", tt.password), func(t *testing.T) {
+            err := ValidatePassword(tt.password)
+            if (err != nil) != tt.expectErr {
+                t.Errorf("ValidatePassword(%s) = %v, expected error = %v", tt.password, err, tt.expectErr)
+            }
+
+            if !tt.expectErr {
+                // Insert the valid password into the database
+                _, err = db.ExecContext(ctx, "INSERT INTO users (username, password) VALUES ($1, $2)", "testuser", tt.password)
+                if err != nil {
+                    t.Fatalf("Failed to insert password into database: %v", err)
+                }
+
+                // Retrieve the password back from the database
+                var storedPassword string
+                err = db.QueryRowContext(ctx, "SELECT password FROM users WHERE username=$1", "testuser").Scan(&storedPassword)
+                if err != nil {
+                    t.Fatalf("Failed to retrieve password from database: %v", err)
+                }
+
+                // Check if the stored password matches the original
+                if storedPassword != tt.password {
+                    t.Errorf("Stored password does not match original. Got %s, want %s", storedPassword, tt.password)
+                }
+            }
+        })
+    }
+}