feat: implementing unit test for pg db client

go.mod

@@ -5,4 +5,5 @@ go 1.16
 require (
 	github.com/go-pg/pg/v10 v10.9.0
 	github.com/go-redis/redis/v8 v8.8.0
+	github.com/stretchr/testify v1.7.0
 )

pgdb/pg_db_test.go

@@ -0,0 +1,49 @@
+package pgdb
+
+import (
+	"fmt"
+	"github.com/go-pg/pg/v10"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+var pgOptions = &pg.Options{
+	Addr:     "localhost:5432",
+	User:     "postgres",
+	Password: "postgres",
+	Database: "pg-db-go",
+}
+
+func TestNewPgDB_WhenOptionsOk_ThenCreateClient(t *testing.T) {
+	got := NewPgDB(pgOptions)
+	got.Close()
+
+	assert.NotNil(t, got)
+	assert.Equal(t, pgOptions.Addr, got.Options().Addr)
+	assert.Equal(t, pgOptions.User, got.Options().User)
+	assert.Equal(t, pgOptions.Password, got.Options().Password)
+	assert.Equal(t, pgOptions.Database, got.Options().Database)
+}
+
+func TestNewPgDB_WhenOptionsNil_ThenPanic(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			assert.Equal(t, fmt.Sprint("runtime error: invalid memory address or nil pointer dereference"), fmt.Sprint(r))
+		}
+	}()
+	NewPgDB(nil)
+}
+
+func TestNewPgDB_WhenOptionsContainBadAddr_ThenPanic(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			assert.Equal(t, fmt.Sprintln("Postgres connection error dial tcp: lookup bad: no such host"), fmt.Sprint(r))
+		}
+	}()
+	NewPgDB(&pg.Options{
+		Addr:     "bad:6379",
+		User:     "postgres",
+		Password: "postgres",
+		Database: "pg-db-go",
+	})
+}

redisdb/redis_db_test.go

@@ -35,7 +35,7 @@ func TestNewRedisDB_WhenOptionsNil_ThenPanic(t *testing.T) {
 func TestNewRedisDB_WhenOptionsContainBadAddr_ThenPanic(t *testing.T) {
 	defer func() {
 		if r := recover(); r != nil {
-			assert.Equal(t, fmt.Sprintln("Redis connection error dial tcp: lookup awdawdaw: no such host"), fmt.Sprint(r))
+			assert.Equal(t, fmt.Sprintln("Redis connection error dial tcp: lookup bad: no such host"), fmt.Sprint(r))
 		}
 	}()
 	NewRedisDB(&redis.Options{