onasty @ a705387

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.getNoteBySlug(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.getNoteBySlug(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.getNoteBySlug(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.getNoteBySlug(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.getNoteBySlug(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)
}