2fas-server/e2e-tests/icons/icons_test.go

package tests

import (
	"encoding/base64"
	"io/ioutil"
	"testing"

	"github.com/jaswdr/faker"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/suite"
	"github.com/twofas/2fas-server/e2e-tests"
	query "github.com/twofas/2fas-server/internal/api/icons/app/queries"
)

func createIcon(t *testing.T) *query.IconPresenter {
	img := faker.New().Image().Image(120, 120)

	pngImg, err := ioutil.ReadFile(img.Name())

	if err != nil {
		t.Error(err)
	}

	iconBase64Encoded := base64.StdEncoding.EncodeToString(pngImg)

	payload := []byte(`
		{
			"name":"facebook",
			"description":"desc",
			"type":"light",
			"icon":"` + iconBase64Encoded + `"
		}
	`)

	var icon *query.IconPresenter

	e2e_tests.DoAdminAPISuccessPost(t, "mobile/icons", payload, &icon)

	return icon
}

func TestIconsTestSuite(t *testing.T) {
	suite.Run(t, new(IconsTestSuite))
}

type IconsTestSuite struct {
	suite.Suite
}

func (s *IconsTestSuite) SetupTest() {
	e2e_tests.DoAdminSuccessDelete(s.T(), "mobile/icons")
}

func (s *IconsTestSuite) TestCreateIcon() {
	icon := createIcon(s.T())

	assert.Equal(s.T(), "facebook", icon.Name)
}

func (s *IconsTestSuite) TestUpdateIcon() {
	icon := createIcon(s.T())

	updatePayload := []byte(`
		{
			"name":"meta"
		}
	`)

	var updatedIcon *query.IconPresenter
	e2e_tests.DoAdminSuccessPut(s.T(), "mobile/icons/"+icon.Id, updatePayload, &updatedIcon)

	assert.Equal(s.T(), "meta", updatedIcon.Name)
}

func (s *IconsTestSuite) TestDeleteIcon() {
	icon := createIcon(s.T())

	e2e_tests.DoAdminSuccessDelete(s.T(), "mobile/icons/"+icon.Id)

	response := e2e_tests.DoAPIGet(s.T(), "mobile/icons/"+icon.Id, nil)
	assert.Equal(s.T(), 404, response.StatusCode)
}

func (s *IconsTestSuite) TestFindAllIcons() {
	createIcon(s.T())
	createIcon(s.T())

	var Icons []*query.IconPresenter
	e2e_tests.DoAPISuccessGet(s.T(), "mobile/icons", &Icons)

	assert.Len(s.T(), Icons, 2)
}

func (s *IconsTestSuite) TestFindIcon() {
	icon := createIcon(s.T())

	var searchResult *query.IconPresenter
	e2e_tests.DoAPISuccessGet(s.T(), "mobile/icons/"+icon.Id, &searchResult)

	assert.Equal(s.T(), "facebook", searchResult.Name)
}