zrok/controller/store/frontend_test.go

package store

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestPublicFrontend(t *testing.T) {
	str, err := Open(&Config{Path: ":memory:", Type: "sqlite3"})
	assert.Nil(t, err)
	assert.NotNil(t, str)

	tx, err := str.Begin()
	assert.Nil(t, err)
	assert.NotNil(t, tx)

	acctId, err := str.CreateAccount(&Account{
		Email:    "test@test.com",
		Password: "password",
		Token:    "token",
	}, tx)
	assert.Nil(t, err)

	envId, err := str.CreateEnvironment(acctId, &Environment{
		Description: "description",
		Host:        "host",
		Address:     "address",
		ZId:         "zId0",
	}, tx)
	assert.Nil(t, err)

	feName := "public"
	feId, err := str.CreateFrontend(envId, &Frontend{
		Token:      "token",
		ZId:        "zId0",
		PublicName: &feName,
	}, tx)
	assert.Nil(t, err)

	fe, err := str.GetFrontend(feId, tx)
	assert.Nil(t, err)
	assert.NotNil(t, fe)
	assert.Equal(t, envId, fe.EnvironmentId)
	assert.Equal(t, feName, *fe.PublicName)

	fe0, err := str.FindFrontendPubliclyNamed(feName, tx)
	assert.Nil(t, err)
	assert.NotNil(t, fe0)
	assert.EqualValues(t, fe, fe0)

	err = str.DeleteFrontend(fe.Id, tx)
	assert.Nil(t, err)

	fe0, err = str.FindFrontendNamed(feName, tx)
	assert.NotNil(t, err)
	assert.Nil(t, fe0)

	fe0, err = str.GetFrontend(fe.Id, tx)
	assert.NotNil(t, err)
	assert.Nil(t, fe0)
}