golang-api-skeleton/lib/common/common_test.go

package common

import (
	"encoding/json"
	"testing"

	"git.devices.local/mawas/golang-api-skeleton/lib/cache"
	"github.com/oklog/ulid/v2"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

const (
	username = "unittest"
	userID   = "01F5Z7CFER2D9QH2VY2V7PVQHF"
)

func mock(t *testing.T) (*gorm.DB, cache.Cache) {
	db, err := gorm.Open(
		sqlite.Open(":memory:"),
		&gorm.Config{DisableForeignKeyConstraintWhenMigrating: true},
	)
	if err != nil {
		t.Error(err)
	}
	appCache, err := cache.Bootstrap()
	if err != nil {
		t.Error(err)
	}
	if err := appCache.Set("user:"+userID, username); err != nil {
		t.Error(err)
	}
	if err := appCache.Set("user:"+username, userID); err != nil {
		t.Error(err)
	}
	return db, appCache
}

func basicfieldsMock(t *testing.T) (*gorm.DB, cache.Cache, BasicFields) {
	db, appCache := mock(t)
	return db, appCache, BasicFields{}
}

func modelGUIDPKMock(t *testing.T) (*gorm.DB, cache.Cache, ModelGUIDPK) {
	db, appCache := mock(t)
	return db, appCache, ModelGUIDPK{}
}

func modelHiddenGUIDPKMock(t *testing.T) (*gorm.DB, cache.Cache, ModelHiddenGUIDPK) {
	db, appCache := mock(t)
	return db, appCache, ModelHiddenGUIDPK{}
}

func TestBasicFieldsBeforeCreate(t *testing.T) {
	db, _, basicFields := basicfieldsMock(t)
	err := basicFields.BeforeCreate(db.Set("userID", userID))
	if err != nil {
		t.Error(err)
	}
	if basicFields.CreatedByDB.String() != userID {
		t.Error("BeforeCreate doesn't set CreatedByDB")
	}
	if basicFields.UpdatedByDB.String() != userID {
		t.Error("BeforeCreate doesn't set UpdatedByDB")
	}
}

func TestBasicFieldsAfterCreate(t *testing.T) {
	db, _, basicFields := basicfieldsMock(t)
	err := basicFields.AfterCreate(db.Set("username", username))
	if err != nil {
		t.Error(err)
	}
	if basicFields.CreatedByJSON != username {
		t.Error("BasicFields AfterCreate doesn't set CreatedByJSON")
	}
	if basicFields.UpdatedByJSON != username {
		t.Error("BasicFields AfterCreate doesn't set UpdatedByJSON")
	}
}

func TestModelGUIDPKBeforeCreate(t *testing.T) {
	db, _, modelGUIDPK := modelGUIDPKMock(t)
	err := modelGUIDPK.BeforeCreate(db.Set("userID", userID))
	if err != nil {
		t.Error(err)
	}
	if modelGUIDPK.CreatedByDB.String() != userID {
		t.Error("ModelGUIDPK BeforeCreate doesn't set CreatedByDB")
	}
	if modelGUIDPK.UpdatedByDB.String() != userID {
		t.Error("ModelGUIDPK BeforeCreate doesn't set UpdatedByDB")
	}
	if _, err := ulid.Parse(modelGUIDPK.ID.String()); err != nil {
		t.Error("ModelGUIDPK BeforeCreate doesn't set proper ID ->", err.Error())
	}
	jsonString, err := json.Marshal(modelGUIDPK)
	if err != nil {
		t.Error(err)
	}
	var j map[string]interface{}
	if err := json.Unmarshal([]byte(jsonString), &j); err != nil {
		t.Error(err)
	}
	if _, exists := j["id"]; !exists {
		t.Error("ModelGUIDPK ID must be visible in json")
	}
}

func TestModelHiddenGUIDPKBeforeCreate(t *testing.T) {
	db, _, modelHiddenGUIDPK := modelHiddenGUIDPKMock(t)
	err := modelHiddenGUIDPK.BeforeCreate(db.Set("userID", userID))
	if err != nil {
		t.Error(err)
	}
	if modelHiddenGUIDPK.CreatedByDB.String() != userID {
		t.Error("ModelHiddenGUIDPK BeforeCreate doesn't set CreatedByDB")
	}
	if modelHiddenGUIDPK.UpdatedByDB.String() != userID {
		t.Error("ModelHiddenGUIDPK BeforeCreate doesn't set UpdatedByDB")
	}
	if _, err := ulid.Parse(modelHiddenGUIDPK.ID.String()); err != nil {
		t.Error("ModelHiddenGUIDPK BeforeCreate doesn't set proper ID ->", err.Error())
	}
	jsonString, err := json.Marshal(modelHiddenGUIDPK)
	if err != nil {
		t.Error(err)
	}
	var j map[string]interface{}
	if err := json.Unmarshal([]byte(jsonString), &j); err != nil {
		t.Error(err)
	}
	if _, exists := j["id"]; exists {
		t.Error("ModelGUIDPK ID must be hidden in json")
	}
}