test ping

Author: thomaspoignant

dao/pgimpl/postgres_impl_priv_test.go

@@ -0,0 +1,14 @@
+package pgimpl
+
+import (
+	"testing"
+
+	_ "github.com/golang-migrate/migrate/v4/source/file"
+	_ "github.com/lib/pq" // we import the driver used by sqlx
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPingNoDb(t *testing.T) {
+	impl := pgFlagImpl{}
+	assert.Error(t, impl.Ping())
+}

dao/pgimpl/postgres_impl_test.go

@@ -943,5 +943,11 @@ func TestUpdateFlag(t *testing.T) {
 			}
 		})
 	}
+}
 
+func TestPingSuccess(t *testing.T) {
+	pgContainer, conn := setupTest(t, []string{})
+	defer tearDownTest(t, pgContainer, conn)
+	pgDao := getPostgresDao(t, pgContainer)
+	assert.NoError(t, pgDao.Ping())
 }