golang-lib/web/webtest/main.go

package webtest

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/http/httptest"
	"strings"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"

	"dev.sum7.eu/genofire/golang-lib/database"
	"dev.sum7.eu/genofire/golang-lib/web"
)

var (
	DBConnection = "user=root password=root dbname=defaultdb host=localhost port=26257 sslmode=disable"
)

type testServer struct {
	db          *database.Database
	gin         *gin.Engine
	ws          *web.Service
	assert      *assert.Assertions
	lastCookies []*http.Cookie
}

type Login struct {
	Username string `json:"username"`
	Password string `json:"password"`
}

func New(assert *assert.Assertions) *testServer {
	// db setup
	dbConfig := database.Database{
		Connection: DBConnection,
		Testdata:   true,
		Debug:      false,
		LogLevel:   0,
	}
	err := dbConfig.Run()
	if err != nil && err != database.ErrNothingToMigrate {
		fmt.Println(err.Error())
		assert.Nil(err)
	}
	assert.NotNil(dbConfig.DB)

	// api setup
	gin.EnableJsonDecoderDisallowUnknownFields()
	gin.SetMode(gin.TestMode)

	ws := &web.Service{
		DB: dbConfig.DB,
	}
	ws.Session.Name = "mysession"
	ws.Session.Secret = "hidden"

	r := gin.Default()
	ws.LoadSession(r)
	ws.Bind(r)
	return &testServer{
		db:     &dbConfig,
		gin:    r,
		ws:     ws,
		assert: assert,
	}
}
func (this *testServer) DatabaseMigration(f func(db *database.Database)) {
	f(this.db)
	this.db.MigrateTestdata()
}
func (this *testServer) Request(method, url string, body interface{}, expectCode int, jsonObj interface{}) {
	var jsonBody io.Reader
	if body != nil {
		if strBody, ok := body.(string); ok {
			jsonBody = strings.NewReader(strBody)
		} else {
			jsonBodyArray, err := json.Marshal(body)
			this.assert.Nil(err, "no request created")
			jsonBody = bytes.NewBuffer(jsonBodyArray)
		}
	}
	req, err := http.NewRequest(method, url, jsonBody)
	this.assert.Nil(err, "no request created")
	if len(this.lastCookies) > 0 {
		for _, c := range this.lastCookies {
			req.AddCookie(c)
		}
	}
	w := httptest.NewRecorder()
	this.gin.ServeHTTP(w, req)

	// valid statusCode
	this.assert.Equal(expectCode, w.Code, "expected http status code")
	if expectCode != w.Code {
		fmt.Printf("wrong status code, body:%v\n", w.Body)
		return
	}

	if jsonObj != nil {
		// fetch JSON
		err = json.NewDecoder(w.Body).Decode(jsonObj)
		this.assert.Nil(err, "decode json")
	}

	result := w.Result()
	if result != nil {
		cookies := result.Cookies()
		if len(cookies) > 0 {
			this.lastCookies = cookies
		}
	}
}

func (this *testServer) Login(login Login) {
	// POST: correct login
	this.Request(http.MethodPost, "/api/v1/auth/login", &login, http.StatusOK, nil)
}

func (this *testServer) TestLogin() {
	this.Login(Login{
		Username: "admin",
		Password: "CHANGEME",
	})
}