onasty @ 09a22a834f95a9c5a2c710c084cdb039c29283fb

package e2e_test

import (

	"errors"

	"time"

	"github.com/gofrs/uuid/v5"

	"github.com/henvic/pgq"

	"github.com/jackc/pgx/v5"

	"github.com/olexsmir/onasty/internal/models"

)

func (e *AppTestSuite) getUserFromDBByUsername(username string) models.User {

	query, args, err := pgq.

		Select("id", "username", "email", "password", "created_at", "last_login_at").

		From("users").

		Where(pgq.Eq{"username": username}).

		SQL()

	e.require.NoError(err)

	var user models.User

	err = e.postgresDB.QueryRow(e.ctx, query, args...).

		Scan(&user.ID, &user.Username, &user.Email, &user.Password, &user.CreatedAt, &user.LastLoginAt)

	e.require.NoError(err)

	return user

}

func (e *AppTestSuite) insertUserIntoDB(uname, email, passwd string, activated ...bool) uuid.UUID {

	p, err := e.hasher.Hash(passwd)

	e.require.NoError(err)

	var a bool

	if len(activated) == 1 {

		a = activated[0]

	}

	query, args, err := pgq.

		Insert("users").

		Columns("username", "email", "password", "activated", "created_at", "last_login_at").

		Values(uname, email, p, a, time.Now(), time.Now()).

		Returning("id").

		SQL()

	e.require.NoError(err)

	var id uuid.UUID

	err = e.postgresDB.QueryRow(e.ctx, query, args...).Scan(&id)

	e.require.NoError(err)

	return id

}

func (e *AppTestSuite) getLastUserSessionByUserID(uid uuid.UUID) models.Session {

	query, args, err := pgq.

		Select("refresh_token", "expires_at").

		From("sessions").

		Where(pgq.Eq{"user_id": uid.String()}).

		OrderBy("expires_at DESC").

		SQL()

	e.require.NoError(err)

	var session models.Session

	err = e.postgresDB.QueryRow(e.ctx, query, args...).

		Scan(&session.RefreshToken, &session.ExpiresAt)

	if errors.Is(err, pgx.ErrNoRows) {

		return models.Session{} //nolint:exhaustruct

	}

	e.require.NoError(err)

	return session

}

func (e *AppTestSuite) getLastInsertedUserByEmail(em string) models.User {

	query, args, err := pgq.

		Select("id", "username", "activated", "email", "password").

		From("users").

		Where(pgq.Eq{"email": em}).

		OrderBy("created_at DESC").

		Limit(1).

		SQL()

	e.require.NoError(err)

	var u models.User

	err = e.postgresDB.QueryRow(e.ctx, query, args...).

		Scan(&u.ID, &u.Username, &u.Activated, &u.Email, &u.Password)

	if errors.Is(err, pgx.ErrNoRows) {

		return models.User{} //nolint:exhaustruct

	}

	e.require.NoError(err)

	return u

}

type noteModel struct {

	ID                   uuid.UUID

	Content              string

	Slug                 string

	BurnBeforeExpiration bool

	Password             string

	IsRead               bool

	ReadAt               *time.Time

	CreatedAt            time.Time

	ExpiresAt            time.Time

}

func (e *AppTestSuite) getNoteFromDBbySlug(slug string) noteModel {

	query, args, err := pgq.

		Select(

			"id",

			"content",

			"slug",

			"burn_before_expiration",

			"read_at",

			"created_at",

			"expires_at",

		).

		From("notes").

		Where(pgq.Eq{"slug": slug}).

		SQL()

	e.require.NoError(err)

	var note noteModel

	err = e.postgresDB.QueryRow(e.ctx, query, args...).

		Scan(&note.ID, &note.Content, &note.Slug, &note.BurnBeforeExpiration, &note.ReadAt, &note.CreatedAt, &note.ExpiresAt)

	if errors.Is(err, pgx.ErrNoRows) {

		return noteModel{} //nolint:exhaustruct

	}

	e.require.NoError(err)

	return note

}

type noteAuthorModel struct {

	noteID uuid.UUID

	userID uuid.UUID

}

func (e *AppTestSuite) getLastNoteAuthorsRecordByAuthorID(uid uuid.UUID) noteAuthorModel {

	qeuery, args, err := pgq.

		Select("note_id", "user_id").

		From("notes_authors").

		Where(pgq.Eq{"user_id": uid.String()}).

		OrderBy("created_at DESC").

		Limit(1).

		SQL()

	e.require.NoError(err)

	var na noteAuthorModel

	err = e.postgresDB.QueryRow(e.ctx, qeuery, args...).Scan(&na.noteID, &na.userID)

	if errors.Is(err, pgx.ErrNoRows) {

		return noteAuthorModel{} //nolint:exhaustruct

	}

	e.require.NoError(err)

	return na

}

type userVerificationToken struct {

	Token  string

	UsedAt *time.Time

}

func (e *AppTestSuite) getVerificationTokenByUserID(u uuid.UUID) userVerificationToken {

	query, args, err := pgq.

		Select("token", "used_at").

		From("verification_tokens").

		Where(pgq.Eq{"user_id": u.String()}).

		SQL()

	e.require.NoError(err)

	var r userVerificationToken

	err = e.postgresDB.QueryRow(e.ctx, query, args...).Scan(&r.Token, &r.UsedAt)

	e.require.NoError(err)

	return r

}