onasty @ 09a22a834f95a9c5a2c710c084cdb039c29283fb

package e2e_test

import (

	"net/http"

	"net/http/httptest"

	"time"

	"github.com/gofrs/uuid/v5"

)

type (

	apiv1NoteCreateRequest struct {

		Content              string    `json:"content"`

		Slug                 string    `json:"slug"`

		Password             string    `json:"password"`

		BurnBeforeExpiration bool      `json:"burn_before_expiration"`

		ExpiresAt            time.Time `json:"expires_at"`

	}

	apiv1NoteCreateResponse struct {

		Slug string `json:"slug"`

	}

)

func (e *AppTestSuite) TestNoteV1_Create() {

	tests := []struct {

		name   string

		inp    apiv1NoteCreateRequest

		assert func(*httptest.ResponseRecorder, apiv1NoteCreateRequest)

	}{

		{

			name: "empty request",

			inp:  apiv1NoteCreateRequest{}, //nolint:exhaustruct

			assert: func(r *httptest.ResponseRecorder, _ apiv1NoteCreateRequest) {

				e.Equal(r.Code, http.StatusBadRequest)

			},

		},

		{

			name: "content only",

			inp:  apiv1NoteCreateRequest{Content: e.uuid()}, //nolint:exhaustruct

			assert: func(r *httptest.ResponseRecorder, _ apiv1NoteCreateRequest) {

				e.Equal(r.Code, http.StatusCreated)

				var body apiv1NoteCreateResponse

				e.readBodyAndUnjsonify(r.Body, &body)

				_, err := uuid.FromString(body.Slug)

				e.require.NoError(err)

				dbNote := e.getNoteFromDBbySlug(body.Slug)

				e.NotEmpty(dbNote)

			},

		},

		{

			name: "set slug",

			inp: apiv1NoteCreateRequest{ //nolint:exhaustruct

				Slug:    e.uuid() + "fuker",

				Content: e.uuid(),

			},

			assert: func(r *httptest.ResponseRecorder, inp apiv1NoteCreateRequest) {

				e.Equal(r.Code, http.StatusCreated)

				var body apiv1NoteCreateResponse

				e.readBodyAndUnjsonify(r.Body, &body)

				dbNote := e.getNoteFromDBbySlug(inp.Slug)

				e.NotEmpty(dbNote)

			},

		},

		{

			name: "set password",

			inp: apiv1NoteCreateRequest{ //nolint:exhaustruct

				Content:  e.uuid(),

				Password: e.uuid(),

			},

			assert: func(r *httptest.ResponseRecorder, _ apiv1NoteCreateRequest) {

				e.Equal(r.Code, http.StatusCreated)

			},

		},

		{

			name: "all possible fields",

			inp: apiv1NoteCreateRequest{ //nolint:exhaustruct

				Content:              e.uuid(),

				BurnBeforeExpiration: true,

				ExpiresAt:            time.Now().Add(time.Hour),

			},

			assert: func(r *httptest.ResponseRecorder, inp apiv1NoteCreateRequest) {

				e.Equal(r.Code, http.StatusCreated)

				var body apiv1NoteCreateResponse

				e.readBodyAndUnjsonify(r.Body, &body)

				dbNote := e.getNoteFromDBbySlug(body.Slug)

				e.NotEmpty(dbNote)

				e.Equal(dbNote.Content, inp.Content)

				e.Equal(dbNote.BurnBeforeExpiration, inp.BurnBeforeExpiration)

				e.Equal(dbNote.ExpiresAt.Unix(), inp.ExpiresAt.Unix())

			},

		},

	}

	for _, tt := range tests {

		httpResp := e.httpRequest(http.MethodPost, "/api/v1/note", e.jsonify(tt.inp))

		tt.assert(httpResp, tt.inp)

	}

}

type apiv1NoteGetResponse struct {

	Content   string     `json:"content"`

	ReadAt    *time.Time `json:"read_at"`

	CreatedAt time.Time  `json:"created_at"`

	ExpiresAt time.Time  `json:"expires_at"`

}

func (e *AppTestSuite) TestNoteV1_Get() {

	content := e.uuid()

	httpResp := e.httpRequest(

		http.MethodPost,

		"/api/v1/note",

		e.jsonify(apiv1NoteCreateRequest{ //nolint:exhaustruct

			Content: content,

		}),

	)

	e.Equal(http.StatusCreated, httpResp.Code)

	var bodyCreated apiv1NoteCreateResponse

	e.readBodyAndUnjsonify(httpResp.Body, &bodyCreated)

	httpResp = e.httpRequest(http.MethodGet, "/api/v1/note/"+bodyCreated.Slug, nil)

	e.Equal(httpResp.Code, http.StatusOK)

	var body apiv1NoteGetResponse

	e.readBodyAndUnjsonify(httpResp.Body, &body)

	e.Equal(content, body.Content)

	dbNote := e.getNoteFromDBbySlug(bodyCreated.Slug)

	e.Equal(dbNote.Content, "")

	e.Equal(dbNote.ReadAt.IsZero(), false)

}

type apiv1NoteGetRequest struct {

	Password string `json:"password"`

}

func (e *AppTestSuite) TestNoteV1_GetWithPassword() {

	content := e.uuid()

	passwd := e.uuid()

	httpResp := e.httpRequest(

		http.MethodPost,

		"/api/v1/note",

		e.jsonify(apiv1NoteCreateRequest{ //nolint:exhaustruct

			Content:  content,

			Password: passwd,

		}),

	)

	e.Equal(http.StatusCreated, httpResp.Code)

	var bodyCreated apiv1NoteCreateResponse

	e.readBodyAndUnjsonify(httpResp.Body, &bodyCreated)

	httpResp = e.httpRequest(

		http.MethodGet,

		"/api/v1/note/"+bodyCreated.Slug,

		e.jsonify(apiv1NoteGetRequest{

			Password: passwd,

		}),

	)

	e.Equal(httpResp.Code, http.StatusOK)

	var body apiv1NoteGetResponse

	e.readBodyAndUnjsonify(httpResp.Body, &body)

	e.Equal(content, body.Content)

	dbNote := e.getNoteFromDBbySlug(bodyCreated.Slug)

	e.Equal(dbNote.Content, "")

	e.Equal(dbNote.ReadAt.IsZero(), false)

}

func (e *AppTestSuite) TestNoteV1_GetWithPassword_wrongNoPassword() {

	content := e.uuid()

	passwd := e.uuid()

	httpResp := e.httpRequest(

		http.MethodPost,

		"/api/v1/note",

		e.jsonify(apiv1NoteCreateRequest{ //nolint:exhaustruct

			Content:  content,

			Password: passwd,

		}),

	)

	e.Equal(http.StatusCreated, httpResp.Code)

	var bodyCreated apiv1NoteCreateResponse

	e.readBodyAndUnjsonify(httpResp.Body, &bodyCreated)

	httpResp = e.httpRequest(http.MethodGet, "/api/v1/note/"+bodyCreated.Slug, nil)

	e.Equal(httpResp.Code, http.StatusNotFound)

}

func (e *AppTestSuite) TestNoteV1_GetWithPassword_wrong() {

	content := e.uuid()

	httpResp := e.httpRequest(

		http.MethodPost,

		"/api/v1/note",

		e.jsonify(apiv1NoteCreateRequest{ //nolint:exhaustruct

			Content:  content,

			Password: e.uuid(),

		}),

	)

	e.Equal(http.StatusCreated, httpResp.Code)

	var bodyCreated apiv1NoteCreateResponse

	e.readBodyAndUnjsonify(httpResp.Body, &bodyCreated)

	httpResp = e.httpRequest(

		http.MethodGet,

		"/api/v1/note/"+bodyCreated.Slug,

		e.jsonify(apiv1NoteGetRequest{

			Password: e.uuid(),

		}),

	)

	e.Equal(httpResp.Code, http.StatusNotFound)

}