postgres_test.go

package postgres

import (
	"database/sql"
	"fmt"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
)

type PostgresSetupTestSuite struct {
	suite.Suite
	client      *sql.DB
	databaseUrl string
}

// Create Table for running test
func (suite *PostgresSetupTestSuite) BeforeTest(suiteName, testName string) {
	var err error

	suite.databaseUrl = fmt.Sprintf(
		"postgresql://%s:%s@%s:%s/%s?sslmode=disable",
		os.Getenv("POSTGRES_USER"),
		os.Getenv("POSTGRES_PASSWORD"),
		os.Getenv("POSTGRES_HOST"),
		os.Getenv("POSTGRES_PORT"),
		os.Getenv("POSTGRES_DB"),
	)

	if suite.client, err = sql.Open("postgres", suite.databaseUrl); err != nil {
		suite.T().Error(err)
	}
	if _, err := suite.client.Query("CREATE TABLE test (test_field TEXT)"); err != nil {
		suite.T().Error(err)
	}
}

// Delete Table after test
func (suite *PostgresSetupTestSuite) AfterTest(suiteName, testName string) {
	if _, err := suite.client.Query("DROP TABLE test"); err != nil {
		suite.T().Error(err)
	}
}

func (suite *PostgresSetupTestSuite) TestPostgresName() {
	newPostgres := storage{}
	assert.Equal(suite.T(), "postgres", newPostgres.Name())
}

func (suite *PostgresSetupTestSuite) TestPostgresNewStorage() {
	_, err := NewStorage(map[string]interface{}{
		"databaseUrl": []int{1},
	})
	assert.Error(suite.T(), err)

	_, err = NewStorage(map[string]interface{}{
		"databaseUrl": suite.databaseUrl,
		"tableName":   "test",
		"dataField":   "test_field",
	})
	assert.NoError(suite.T(), err)
}

func (suite *PostgresSetupTestSuite) TestPostgresPush() {
	newClient, _ := NewStorage(map[string]interface{}{
		"databaseUrl": suite.databaseUrl,
		"tableName":   "Not Exist",
		"dataField":   "Not exist",
	})
	err := newClient.Push("Hello")
	assert.Error(suite.T(), err)

	newClient, err = NewStorage(map[string]interface{}{
		"databaseUrl": suite.databaseUrl,
		"tableName":   "test",
		"dataField":   "test_field",
	})
	assert.NoError(suite.T(), err)

	err = newClient.Push("Hello")
	assert.NoError(suite.T(), err)
}

func TestRunPostgresPush(t *testing.T) {
	if testing.Short() {
		t.Skip("postgresql testing is skiped in short version of test")
		return
	}

	suite.Run(t, new(PostgresSetupTestSuite))
}