core-old/internal/bundb/tribe_snapshot_test.go

package bundb_test

import (
	"context"
	"testing"
	"time"

	"gitea.dwysokinski.me/twhelp/core/internal/bundb"
	"gitea.dwysokinski.me/twhelp/core/internal/domain"
	"github.com/jackc/pgerrcode"
	"github.com/stretchr/testify/assert"
	"github.com/uptrace/bun/driver/pgdriver"
)

func TestTribeSnapshot_Create(t *testing.T) {
	t.Parallel()

	db := newDB(t)
	fixture := loadFixtures(t, db)
	repo := bundb.NewTribeSnapshot(db)

	t.Run("OK", func(t *testing.T) {
		t.Parallel()

		tribeTT := fixture.tribe(t, "pl169-tt")
		tribeCSA := fixture.tribe(t, "pl169-csa")

		params := []domain.CreateTribeSnapshotParams{
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID,
				ServerKey:         tribeTT.ServerKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-15 * time.Hour),
			},
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID,
				ServerKey:         tribeTT.ServerKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-48 * time.Hour),
			},
			{
				OpponentsDefeated: tribeCSA.OpponentsDefeated,
				TribeID:           tribeCSA.ID,
				ServerKey:         tribeCSA.ServerKey,
				NumMembers:        tribeCSA.NumMembers,
				NumVillages:       tribeCSA.NumVillages,
				Points:            tribeCSA.Points,
				AllPoints:         tribeCSA.AllPoints,
				Rank:              tribeCSA.Rank,
				Dominance:         tribeCSA.Dominance,
				Date:              time.Now().Add(-15 * time.Hour),
			},
		}

		assert.NoError(t, repo.Create(context.Background(), params...))
	})

	t.Run("OK: len(params) == 0", func(t *testing.T) {
		t.Parallel()

		assert.NoError(t, repo.Create(context.Background()))
	})

	t.Run("ERR: snapshot already exists", func(t *testing.T) {
		t.Parallel()

		tribeTT := fixture.tribe(t, "pl169-tt")

		params := []domain.CreateTribeSnapshotParams{
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID,
				ServerKey:         tribeTT.ServerKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-120 * time.Hour),
			},
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID,
				ServerKey:         tribeTT.ServerKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-120 * time.Hour),
			},
		}

		err := repo.Create(context.Background(), params...)
		var pgErr pgdriver.Error
		assert.ErrorAs(t, err, &pgErr)
		assert.Equal(t, pgerrcode.UniqueViolation, pgErr.Field('C'))
		assert.Equal(t, "tribe_snapshots_tribe_id_server_key_date_key", pgErr.Field('n'))
	})

	t.Run("ERR: server must exist in the db", func(t *testing.T) {
		t.Parallel()

		serverKey := "random"
		tribeTT := fixture.tribe(t, "pl169-tt")

		params := []domain.CreateTribeSnapshotParams{
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID,
				ServerKey:         serverKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-168 * time.Hour),
			},
		}

		err := repo.Create(context.Background(), params...)
		var pgErr pgdriver.Error
		assert.ErrorAs(t, err, &pgErr)
		assert.Equal(t, pgerrcode.ForeignKeyViolation, pgErr.Field('C'))
		assert.Equal(t, "tribe_snapshots_server_key_fkey", pgErr.Field('n'))
	})

	t.Run("ERR: tribe must exist in the db", func(t *testing.T) {
		t.Parallel()

		tribeTT := fixture.tribe(t, "pl169-tt")

		params := []domain.CreateTribeSnapshotParams{
			{
				OpponentsDefeated: tribeTT.OpponentsDefeated,
				TribeID:           tribeTT.ID + 123123123131,
				ServerKey:         tribeTT.ServerKey,
				NumMembers:        tribeTT.NumMembers,
				NumVillages:       tribeTT.NumVillages,
				Points:            tribeTT.Points,
				AllPoints:         tribeTT.AllPoints,
				Rank:              tribeTT.Rank,
				Dominance:         tribeTT.Dominance,
				Date:              time.Now().Add(-168 * time.Hour),
			},
		}

		err := repo.Create(context.Background(), params...)
		var pgErr pgdriver.Error
		assert.ErrorAs(t, err, &pgErr)
		assert.Equal(t, pgerrcode.ForeignKeyViolation, pgErr.Field('C'))
		assert.Equal(t, "tribe_snapshots_tribe_id_server_key_fkey", pgErr.Field('n'))
	})
}