internal/postgres/user.go

@@ -230,8 +230,8 @@ func GetAllUsers(ctx context.Context, db *gorm.DB) ([]models.User, error) {
 	return users, nil
 }
 
-// TODO: FIX THE TEST
 func GetUserFavoriteWithPagination(ctx context.Context, db *gorm.DB, anthroveUserID models.AnthroveUserID, skip int, limit int) (*models.FavoriteList, error) {
+	var userFavorites []models.UserFavorites
 	var favoritePosts []models.Post
 
 	if anthroveUserID == "" {
@@ -242,7 +242,30 @@ func GetUserFavoriteWithPagination(ctx context.Context, db *gorm.DB, anthroveUse
 		return nil, &otterError.EntityValidationFailed{Reason: otterError.AnthroveUserIDToShort}
 	}
 
-	db.WithContext(ctx).Joins("RIGHT JOIN \"UserFavorites\" AS of ON \"Post\".id = of.post_id AND of.user_id = ?", anthroveUserID).Preload("References").Offset(skip).Limit(limit).Find(&favoritePosts)
+	err := db.WithContext(ctx).Model(&models.UserFavorites{}).Where("user_id = ?", string(anthroveUserID)).Offset(skip).Limit(limit).Find(&userFavorites).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, &otterError.NoDataFound{}
+		}
+		return nil, err
+	}
+
+	for _, userFavorite := range userFavorites {
+		var post models.Post
+		err = db.WithContext(ctx).Model(&models.Post{}).Where("id = ?", userFavorite.PostID).First(&post).Error
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				return nil, &otterError.NoDataFound{}
+			}
+			return nil, err
+		}
+
+		favoritePosts = append(favoritePosts,
+			models.Post{
+				BaseModel: models.BaseModel[models.AnthrovePostID]{ID: post.ID},
+				Rating:    post.Rating,
+			})
+	}
 
 	log.WithFields(log.Fields{
 		"anthrove_user_id":        anthroveUserID,

internal/postgres/user_test.go

@@ -573,30 +573,13 @@ func TestGetUserFavoriteNodeWithPagination(t *testing.T) {
 				t.Errorf("GetAllUserFavoritesWithPagination() error = %v, wantErr %v", err, tt.wantErr)
 				return
 			}
-			if !checkFavoritePosts(got, tt.want) {
+			if !reflect.DeepEqual(got, tt.want) {
 				t.Errorf("GetAllUserFavoritesWithPagination() got = %v, want %v", got, tt.want)
 			}
 		})
 	}
 }
 
-func checkFavoritePosts(got *models.FavoriteList, want *models.FavoriteList) bool {
-	if got == nil && want == nil {
-		return true
-	} else if got == nil || want == nil {
-		return false
-	}
-
-	for i, post := range got.Posts {
-		if post.ID == want.Posts[i].ID {
-		} else {
-			return false
-		}
-	}
-
-	return true
-}
-
 func TestGetUserFavoritesCount(t *testing.T) {
 	// Setup trow away container
 	ctx := context.Background()

pkg/database/postgres_test.go

@@ -1556,7 +1556,7 @@ func Test_postgresqlConnection_GetUserFavoriteWithPagination(t *testing.T) {
 				t.Errorf("GetAllUserFavoritesWithPagination() error = %v, wantErr %v", err, tt.wantErr)
 				return
 			}
-			if !checkFavoritePosts(got, tt.want) {
+			if !reflect.DeepEqual(got, tt.want) {
 				t.Errorf("GetAllUserFavoritesWithPagination() got = %v, want %v", got, tt.want)
 			}
 		})
@@ -3654,20 +3654,3 @@ func Test_postgresqlConnection_CreatePostInBatch(t *testing.T) {
 		})
 	}
 }
-
-func checkFavoritePosts(got *models.FavoriteList, want *models.FavoriteList) bool {
-	if got == nil && want == nil {
-		return true
-	} else if got == nil || want == nil {
-		return false
-	}
-
-	for i, post := range got.Posts {
-		if post.ID == want.Posts[i].ID {
-		} else {
-			return false
-		}
-	}
-
-	return true
-}

pkg/models/orm.go

@@ -1,10 +1,9 @@
 package models
 
 import (
-	"time"
-
 	gonanoid "github.com/matoous/go-nanoid/v2"
 	"gorm.io/gorm"
+	"time"
 )
 
 type ID interface {
@@ -12,10 +11,10 @@ type ID interface {
 }
 
 type BaseModel[T ID] struct {
-	ID        T              `json:"id" gorm:"primaryKey"`
-	CreatedAt time.Time      `json:"-"`
-	UpdatedAt time.Time      `json:"-"`
-	DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
+	ID        T `gorm:"primaryKey"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	DeletedAt gorm.DeletedAt `gorm:"index"`
 }
 
 func (base *BaseModel[T]) BeforeCreate(db *gorm.DB) error {

pkg/models/post.go

@@ -3,10 +3,10 @@ package models
 // Post model
 type Post struct {
 	BaseModel[AnthrovePostID]
-	Rating     Rating          `json:"rating" gorm:"type:enum('safe','questionable','explicit')"`
-	Tags       []Tag           `json:"-" gorm:"many2many:post_tags;"`
-	Favorites  []UserFavorites `json:"-" gorm:"foreignKey:PostID"`
-	References []PostReference `json:"references" gorm:"foreignKey:PostID"`
+	Rating     Rating          `gorm:"type:enum('safe','questionable','explicit')"`
+	Tags       []Tag           `gorm:"many2many:post_tags;"`
+	Favorites  []UserFavorites `gorm:"foreignKey:PostID"`
+	References []PostReference `gorm:"foreignKey:PostID"`
 }
 
 func (Post) TableName() string {

pkg/models/postReference.go

@@ -1,17 +1,17 @@
 package models
 
 type PostReference struct {
-	PostID   string `json:"post_id" gorm:"primaryKey"`
-	SourceID string `json:"source_id" gorm:"primaryKey"`
-	URL      string `json:"url" gorm:"primaryKey"`
+	PostID   string `gorm:"primaryKey"`
+	SourceID string `gorm:"primaryKey"`
+	URL      string `gorm:"primaryKey"`
 	PostReferenceConfig
 }
 
 type PostReferenceConfig struct {
-	SourcePostID   string `json:"source_post_id"`
-	FullFileURL    string `json:"full_file_url"`
-	PreviewFileURL string `json:"preview_file_url"`
-	SampleFileURL  string `json:"sample_file_url"`
+	SourcePostID   string
+	FullFileURL    string
+	PreviewFileURL string
+	SampleFileURL  string
 }
 
 func (PostReference) TableName() string {

pkg/models/source.go

@@ -3,11 +3,11 @@ package models
 // Source model
 type Source struct {
 	BaseModel[AnthroveSourceID]
-	DisplayName string          `json:"display_name" `
-	Domain      string          `json:"domain" gorm:"not null;unique"`
-	Icon        string          `json:"icon" gorm:"not null"`
-	UserSources []UserSource    `json:"-" gorm:"foreignKey:SourceID"`
-	References  []PostReference `json:"references" gorm:"foreignKey:SourceID"`
+	DisplayName string
+	Domain      string          `gorm:"not null;unique"`
+	Icon        string          `gorm:"not null"`
+	UserSources []UserSource    `gorm:"foreignKey:SourceID"`
+	References  []PostReference `gorm:"foreignKey:SourceID"`
 }
 
 func (Source) TableName() string {

pkg/models/tag.go

@@ -2,11 +2,11 @@ package models
 
 // Tag models
 type Tag struct {
-	Name    string     `json:"name" gorm:"primaryKey"`
-	Type    TagType    `json:"type" gorm:"column:tag_type"`
-	Aliases []TagAlias `json:"aliases" gorm:"foreignKey:TagID"`
-	Groups  []TagGroup `json:"groups" gorm:"foreignKey:TagID"`
-	Posts   []Post     `json:"posts" gorm:"many2many:post_tags;"`
+	Name    string     `gorm:"primaryKey"`
+	Type    TagType    `gorm:"column:tag_type"`
+	Aliases []TagAlias `gorm:"foreignKey:TagID"`
+	Groups  []TagGroup `gorm:"foreignKey:TagID"`
+	Posts   []Post     `gorm:"many2many:post_tags;"`
 }
 
 func (Tag) TableName() string {
@@ -15,8 +15,8 @@ func (Tag) TableName() string {
 
 // TagAlias model
 type TagAlias struct {
-	Name  string `json:"name" gorm:"primaryKey"`
-	TagID string `json:"tag_id"`
+	Name  string `gorm:"primaryKey"`
+	TagID string
 }
 
 func (TagAlias) TableName() string {
@@ -25,8 +25,8 @@ func (TagAlias) TableName() string {
 
 // TagGroup model
 type TagGroup struct {
-	Name  string `json:"name" gorm:"primaryKey"`
-	TagID string `json:"tag_id"`
+	Name  string `gorm:"primaryKey"`
+	TagID string
 }
 
 func (TagGroup) TableName() string {

pkg/models/user.go

@@ -3,8 +3,8 @@ package models
 // User model
 type User struct {
 	BaseModel[AnthroveUserID]
-	Favorites []UserFavorites `json:"-" gorm:"foreignKey:UserID"`
-	Sources   []UserSource    `json:"-" gorm:"foreignKey:UserID"`
+	Favorites []UserFavorites `gorm:"foreignKey:UserID"`
+	Sources   []UserSource    `gorm:"foreignKey:UserID"`
 }
 
 func (User) TableName() string {

pkg/models/userFavorite.go

@@ -3,9 +3,9 @@ package models
 import "time"
 
 type UserFavorites struct {
-	UserID    string    `json:"user_id" gorm:"primaryKey"`
-	PostID    string    `json:"post_id" gorm:"primaryKey"`
-	CreatedAt time.Time `json:"-"`
+	UserID    string `gorm:"primaryKey"`
+	PostID    string `gorm:"primaryKey"`
+	CreatedAt time.Time
 }
 
 func (UserFavorites) TableName() string {

pkg/models/userSource.go

@@ -3,16 +3,16 @@ package models
 import "time"
 
 type UserSource struct {
-	User                 User      `json:"user" gorm:"foreignKey:ID;references:UserID"`
-	UserID               string    `json:"user_id" gorm:"primaryKey"`
-	Source               Source    `json:"source" gorm:"foreignKey:ID;references:SourceID"`
-	SourceID             string    `json:"source_id" gorm:"primaryKey"`
-	ScrapeTimeInterval   string    `json:"scrape_time_interval"`
-	AccountUsername      string    `json:"account_username"`
-	AccountID            string    `json:"account_id"`
-	LastScrapeTime       time.Time `json:"last_scrape_time"`
-	AccountValidate      bool      `json:"account_validate"`
-	AccountValidationKey string    `json:"-"`
+	User                 User   `gorm:"foreignKey:ID;references:UserID"`
+	UserID               string `gorm:"primaryKey"`
+	Source               Source `gorm:"foreignKey:ID;references:SourceID"`
+	SourceID             string `gorm:"primaryKey"`
+	ScrapeTimeInterval   string
+	AccountUsername      string
+	AccountID            string
+	LastScrapeTime       time.Time
+	AccountValidate      bool
+	AccountValidationKey string
 }
 
 func (UserSource) TableName() string {