go.charczuk.com@v0.0.0-20240327042549-bc490516bd1a/projects/chirp/pkg/model/manager_test.go

/*

Copyright (c) 2023 - Present. Will Charczuk. All rights reserved.
Use of this source code is governed by a MIT license that can be found in the LICENSE file at the root of the repository.

*/

package model

import (
	"context"
	"testing"
	"time"

	"go.charczuk.com/sdk/apputil"
	"go.charczuk.com/sdk/assert"
	"go.charczuk.com/sdk/uuid"
)

func Test_Manager_User(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	user, found, err := mgr.User(context.Background(), testData.users[0].ID, testData.users[1].ID)
	assert.ItsNil(t, err)
	assert.ItsTrue(t, found)
	assert.ItsEqual(t, testData.users[1].ID, user.ID)
	assert.ItsEqual(t, false, user.Status.IsSelf)
	assert.ItsEqual(t, true, user.Status.IsFollowing)
}

func Test_Manager_User_self(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	user, found, err := mgr.User(context.Background(), testData.users[0].ID, testData.users[0].ID)
	assert.ItsNil(t, err)
	assert.ItsTrue(t, found)
	assert.ItsEqual(t, testData.users[0].ID, user.ID)
	assert.ItsEqual(t, true, user.Status.IsSelf)
	assert.ItsEqual(t, false, user.Status.IsFollowing)
}

func Test_Manager_User_notFound(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	seedTestData(t, mgr)

	user, found, err := mgr.User(context.Background(), uuid.V4(), uuid.V4())
	assert.ItsNil(t, err)
	assert.ItsFalse(t, found)
	assert.ItsTrue(t, user.ID.IsZero())
}

func Test_Manager_Chirp(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	chirp, found, err := mgr.Chirp(context.Background(), testData.users[0].ID, testData.userChirps[0][0])
	assert.ItsNil(t, err)
	assert.ItsTrue(t, found)
	assert.ItsEqual(t, testData.userChirps[0][0], chirp.Chirp.ID)
	assert.ItsEqual(t, testData.users[0].ID, chirp.Chirp.UserID)
}

func Test_Manager_Chirp_notFound(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	chirp, found, err := mgr.Chirp(context.Background(), testData.users[0].ID, uuid.V4())
	assert.ItsNil(t, err)
	assert.ItsFalse(t, found)
	assert.ItsTrue(t, chirp.Chirp.ID.IsZero())
}

func Test_Manager_SearchUsers(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	users, err := mgr.SearchUsers(context.Background(), testData.users[0].ID, "user")
	assert.ItsNil(t, err)
	assert.ItsNotEmpty(t, users)
}

func Test_Manager_FeedForUserID(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	userID := testData.users[0].ID
	feed, err := mgr.FeedForUserID(context.Background(), userID, 8, nil)
	assert.ItsNil(t, err)
	assert.ItsNotEmpty(t, feed)

	assert.ItsAll(t, feed, func(c ChirpFull) bool {
		return c.User.PictureURL != ""
	})
}

func Test_Manager_FeedForUserID_after(t *testing.T) {
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	testData := seedTestData(t, mgr)

	userID := testData.users[0].ID
	after := testData.userChirps[1][16]
	feed, err := mgr.FeedForUserID(context.Background(), userID, 8, &after)
	assert.ItsNil(t, err)
	assert.ItsNotEmpty(t, feed)

	assert.ItsAny(t, feed, func(c ChirpFull) bool {
		return c.Chirp.ID == testData.userChirps[1][15]
	})
	assert.ItsEqual(t, testData.userChirps[1][15], feed[0].Chirp.ID)
}

func Test_Manager_ProfileLikes(t *testing.T) {
	ctx := context.TODO()
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	u0 := createTestUser(t, mgr, 0)
	u1 := createTestUser(t, mgr, 1)

	u0Chirps := createTestChirps(t, mgr, u0.Username(), u0.ID, time.Duration(0)*time.Minute)
	u1Chirps := createTestChirps(t, mgr, u1.Username(), u1.ID, time.Duration(0)*time.Minute)

	u0liked0 := u1Chirps[0]
	u0liked1 := u1Chirps[1]
	u1liked0 := u0Chirps[0]

	var err error
	err = mgr.LikeChirp(ctx, u0.ID, Chirp{ID: u0liked0, UserID: u0.ID})
	assert.ItsNil(t, err)
	err = mgr.LikeChirp(ctx, u0.ID, Chirp{ID: u0liked1, UserID: u0.ID})
	assert.ItsNil(t, err)
	err = mgr.LikeChirp(ctx, u1.ID, Chirp{ID: u1liked0, UserID: u1.ID})
	assert.ItsNil(t, err)

	u0likes, err := mgr.ProfileLikes(ctx, u0.ID, u0.ID, 10, nil)
	assert.ItsNil(t, err)
	assert.ItsNotEmpty(t, u0likes)
	assert.ItsLen(t, u0likes, 2)

	assert.ItsEqual(t, u0liked0, u0likes[1].ID)
	assert.ItsEqual(t, u0liked1, u0likes[0].ID)
}

func Test_ChirpReplies(t *testing.T) {
	ctx := context.TODO()
	appMgr, done := apputil.NewTest(t)
	defer done()

	mgr := &Manager{BaseManager: appMgr.BaseManager}
	u0 := createTestUser(t, mgr, 0)
	u1 := createTestUser(t, mgr, 1)
	u2 := createTestUser(t, mgr, 2)

	u0Chirps := createTestChirps(t, mgr, u0.Username(), u0.ID, time.Duration(0)*time.Minute)

	// create a reply to a u0 chirp by u1 and u2
	replyTo(t, mgr, u1.ID, u0Chirps[0])
	replyTo(t, mgr, u2.ID, u0Chirps[0])

	replies, err := mgr.ChirpReplies(ctx, u0.ID, u0Chirps[0])
	assert.ItsNil(t, err)
	assert.ItsLen(t, replies, 2)
}