package postgres import ( "context" "fmt" "git.dragse.it/anthrove/otter-space-sdk/pkg/models" "git.dragse.it/anthrove/otter-space-sdk/pkg/models/graphModels" "git.dragse.it/anthrove/otter-space-sdk/test" "gorm.io/gorm" "reflect" "testing" ) func TestCreateUser(t *testing.T) { // Setup trow away container ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID } tests := []struct { name string args args wantErr bool }{ { name: "Test 1: Valid AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", }, wantErr: false, }, { name: "Test 2: Invalid AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", }, wantErr: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if err := CreateUser(tt.args.ctx, tt.args.db, tt.args.anthroveUserID); (err != nil) != tt.wantErr { t.Errorf("CreateUser() error = %v, wantErr %v", err, tt.wantErr) } }) } } func TestCreateUserNodeWithSourceRelation(t *testing.T) { // Setup trow away container ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test source := &models.Source{ DisplayName: "e621", Domain: "e621.net", Icon: "icon.e621.net", } err = CreateSourceNode(ctx, gormDB, source) if err != nil { t.Fatal(err) } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID sourceDomain string userID string username string } tests := []struct { name string args args wantErr bool }{ { name: "Test 1: Valid anthroveUserID, sourceDomain, userID, username", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDomain: "e621.net", userID: "e1", username: "marius", }, wantErr: false, }, { name: "Test 2: Invalid anthroveUserID, valid sourceDomain, userID, username", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "2", sourceDomain: "e621.net", userID: "e1", username: "marius", }, wantErr: true, }, { name: "Test 3: Empty anthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", sourceDomain: "e621.net", userID: "e1", username: "marius", }, wantErr: true, }, { name: "Test 4: invalid sourceDomain", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDomain: "fa.net", userID: "e1", username: "marius", }, wantErr: true, }, { name: "Test 5: no userID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDomain: "e621.net", userID: "", username: "marius", }, wantErr: true, }, { name: "Test 6: no username", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDomain: "e621.net", userID: "aa", username: "", }, wantErr: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if err := CreateUserNodeWithSourceRelation(tt.args.ctx, tt.args.db, tt.args.anthroveUserID, tt.args.sourceDomain, tt.args.userID, tt.args.username); (err != nil) != tt.wantErr { t.Errorf("CreateUserNodeWithSourceRelation() error = %v, wantErr %v", err, tt.wantErr) } }) } } func TestGetAllAnthroveUserIDs(t *testing.T) { // Setup trow away container ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test users := []models.AnthroveUserID{"1", "2", "3"} for _, user := range users { err = CreateUser(ctx, gormDB, user) if err != nil { t.Fatal(err) } } // Test type args struct { ctx context.Context db *gorm.DB } tests := []struct { name string args args want []models.AnthroveUserID wantErr bool }{ { name: "Test 1: Get Data", args: args{ ctx: ctx, db: gormDB, }, want: users, wantErr: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetAllAnthroveUserIDs(tt.args.ctx, tt.args.db) if (err != nil) != tt.wantErr { t.Errorf("GetAllAnthroveUserIDs() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetAllAnthroveUserIDs() got = %v, want %v", got, tt.want) } }) } } func TestGetAnthroveUser(t *testing.T) { // Setup trow away container ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test user := graphModels.AnthroveUser{ UserID: "1", } err = CreateUser(ctx, gormDB, user.UserID) if err != nil { t.Fatal(err) } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID } tests := []struct { name string args args want *graphModels.AnthroveUser wantErr bool }{ { name: "Test 1: Valid AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", }, want: &user, wantErr: false, }, { name: "Test 2: Invalid AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "2", }, want: nil, wantErr: true, }, { name: "Test 3: No AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", }, want: nil, wantErr: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetAnthroveUser(tt.args.ctx, tt.args.db, tt.args.anthroveUserID) if (err != nil) != tt.wantErr { t.Errorf("GetAnthroveUser() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetAnthroveUser() got = %v, want %v", got, tt.want) } }) } } func TestGetSpecifiedUserSourceLink(t *testing.T) { // Setup trow away container ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test source := &models.Source{ DisplayName: "e621", Domain: "e621.net", } err = CreateSourceNode(ctx, gormDB, source) if err != nil { t.Fatal(err) } expectedResult := make(map[string]graphModels.AnthroveUserRelationship) expectedResult["e621"] = graphModels.AnthroveUserRelationship{ UserID: "e1", Username: "euser", Source: graphModels.AnthroveSource{ DisplayName: source.DisplayName, Domain: source.Domain, }, } err = CreateUserNodeWithSourceRelation(ctx, gormDB, "1", source.Domain, expectedResult["e621"].UserID, expectedResult["e621"].Username) if err != nil { t.Fatal(err) } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID sourceDisplayName string } tests := []struct { name string args args want map[string]graphModels.AnthroveUserRelationship wantErr bool }{ { name: "Test 1: Valid AnthroveUserID and SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDisplayName: "e621", }, want: expectedResult, wantErr: false, }, { name: "Test 2: Invalid AnthroveUserID and valid SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "22", sourceDisplayName: "e621", }, want: nil, wantErr: true, }, { name: "Test 3: Valid AnthroveUserID and invalid SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDisplayName: "fa", }, want: nil, wantErr: true, }, { name: "Test 4: No AnthroveUserID and Valid SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", sourceDisplayName: "e621", }, want: nil, wantErr: true, }, { name: "Test 5: Valid AnthroveUserID and No SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", sourceDisplayName: "", }, want: nil, wantErr: true, }, { name: "Test 6: No AnthroveUserID and No SourceDisplayName", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", sourceDisplayName: "", }, want: nil, wantErr: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetSpecifiedUserSourceLink(tt.args.ctx, tt.args.db, tt.args.anthroveUserID, tt.args.sourceDisplayName) if (err != nil) != tt.wantErr { t.Errorf("GetSpecifiedUserSourceLink() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetSpecifiedUserSourceLink() got = %v, expectedResult %v", got, tt.want) } }) } } func TestGetUserFavoriteNodeWithPagination(t *testing.T) { // Setup trow away containert ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test expectedResultPostsGraph := []graphModels.FavoritePost{ { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post1")), Rating: "safe", }, }, { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post2")), Rating: "safe", }, }, { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post3")), Rating: "explicit", }, }, { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post4")), Rating: "explicit", }, }, { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post5")), Rating: "questionable", }, }, { AnthrovePost: graphModels.AnthrovePost{ PostID: models.AnthrovePostID(fmt.Sprintf("%-25s", "Post6")), Rating: "safe", }, }, } expectedResult := &graphModels.FavoriteList{ Posts: expectedResultPostsGraph, } expectedResult2 := &graphModels.FavoriteList{ Posts: expectedResultPostsGraph[2:], } expectedResult3 := &graphModels.FavoriteList{ Posts: expectedResultPostsGraph[:3], } err = CreateUser(ctx, gormDB, "1") if err != nil { t.Fatal(err) } expectedResultPostsPg := []models.Post{ { BaseModel: models.BaseModel{ID: "Post1"}, Rating: "safe", }, { BaseModel: models.BaseModel{ID: "Post2"}, Rating: "safe", }, { BaseModel: models.BaseModel{ID: "Post3"}, Rating: "explicit", }, { BaseModel: models.BaseModel{ID: "Post4"}, Rating: "explicit", }, { BaseModel: models.BaseModel{ID: "Post5"}, Rating: "questionable", }, { BaseModel: models.BaseModel{ID: "Post6"}, Rating: "safe", }, } for _, expectedResultPost := range expectedResultPostsPg { err = CreateAnthrovePostNode(ctx, gormDB, models.AnthrovePostID(expectedResultPost.ID), expectedResultPost.Rating) if err != nil { t.Fatal(err) } err = EstablishUserToPostLink(ctx, gormDB, "1", models.AnthrovePostID(expectedResultPost.ID)) if err != nil { t.Fatal(err) } } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID skip int limit int } tests := []struct { name string args args want *graphModels.FavoriteList wantErr bool }{ { name: "Test 1: Valid AnthroveUserID", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", skip: 0, limit: 2000, }, want: expectedResult, wantErr: false, }, { name: "Test 2: Skip first two", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", skip: 2, limit: 2000, }, want: expectedResult2, wantErr: false, }, { name: "Test 3: Limit of 3", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", skip: 0, limit: 3, }, want: expectedResult3, wantErr: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetUserFavoriteNodeWithPagination(tt.args.ctx, tt.args.db, tt.args.anthroveUserID, tt.args.skip, tt.args.limit) if (err != nil) != tt.wantErr { t.Errorf("GetUserFavoriteNodeWithPagination() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetUserFavoriteNodeWithPagination() got = %v, want %v", got, tt.want) } }) } } func TestGetUserFavoritesCount(t *testing.T) { // Setup trow away containert ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test err = CreateUser(ctx, gormDB, "1") if err != nil { t.Fatal(err) } expectedResultPostsPg := []models.Post{ { BaseModel: models.BaseModel{ID: "Post1"}, Rating: "safe", }, { BaseModel: models.BaseModel{ID: "Post2"}, Rating: "safe", }, { BaseModel: models.BaseModel{ID: "Post3"}, Rating: "explicit", }, { BaseModel: models.BaseModel{ID: "Post4"}, Rating: "explicit", }, { BaseModel: models.BaseModel{ID: "Post5"}, Rating: "questionable", }, { BaseModel: models.BaseModel{ID: "Post6"}, Rating: "safe", }, } for _, expectedResultPost := range expectedResultPostsPg { err = CreateAnthrovePostNode(ctx, gormDB, models.AnthrovePostID(expectedResultPost.ID), expectedResultPost.Rating) if err != nil { t.Fatal(err) } err = EstablishUserToPostLink(ctx, gormDB, "1", models.AnthrovePostID(expectedResultPost.ID)) if err != nil { t.Fatal(err) } } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID } tests := []struct { name string args args want int64 wantErr bool }{ { name: "Test 1: Valid anthroveUserID and 6 favorite posts", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", }, want: 6, wantErr: false, }, { name: "Test 2: Invalid anthroveUserID and 6 favorite posts", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "2", }, want: 0, wantErr: false, }, { name: "Test 3: no anthroveUserID and 6 favorite posts", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "", }, want: 0, wantErr: true, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetUserFavoritesCount(tt.args.ctx, tt.args.db, tt.args.anthroveUserID) if (err != nil) != tt.wantErr { t.Errorf("GetUserFavoritesCount() error = %v, wantErr %v", err, tt.wantErr) return } if got != tt.want { t.Errorf("GetUserFavoritesCount() got = %v, want %v", got, tt.want) } }) } } func TestGetUserSourceLink(t *testing.T) { // Setup trow away containert ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test esource := &models.Source{ DisplayName: "e621", Domain: "e621.net", } err = CreateSourceNode(ctx, gormDB, esource) if err != nil { t.Fatal(err) } fasource := &models.Source{ DisplayName: "fa", Domain: "fa.net", } err = CreateSourceNode(ctx, gormDB, fasource) if err != nil { t.Fatal(err) } expectedResult := make(map[string]graphModels.AnthroveUserRelationship) expectedResult["e621"] = graphModels.AnthroveUserRelationship{ UserID: "e1", Username: "euser", Source: graphModels.AnthroveSource{ DisplayName: esource.DisplayName, Domain: esource.Domain, }, } expectedResult["fa"] = graphModels.AnthroveUserRelationship{ UserID: "fa1", Username: "fauser", Source: graphModels.AnthroveSource{ DisplayName: fasource.DisplayName, Domain: fasource.Domain, }, } err = CreateUserNodeWithSourceRelation(ctx, gormDB, "1", esource.Domain, expectedResult["e621"].UserID, expectedResult["e621"].Username) if err != nil { t.Fatal(err) } err = CreateUserNodeWithSourceRelation(ctx, gormDB, "1", fasource.Domain, expectedResult["fa"].UserID, expectedResult["fa"].Username) if err != nil { t.Fatal(err) } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID } tests := []struct { name string args args want map[string]graphModels.AnthroveUserRelationship wantErr bool }{ { name: "Test 1: Get Data", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", }, want: expectedResult, wantErr: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetUserSourceLink(tt.args.ctx, tt.args.db, tt.args.anthroveUserID) if (err != nil) != tt.wantErr { t.Errorf("GetUserSourceLink() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetUserSourceLink() got = %v, want %v", got, tt.want) } }) } } func TestGetUserTagNodeWitRelationToFavedPosts(t *testing.T) { // Setup trow away containert ctx := context.Background() container, gormDB, err := test.StartPostgresContainer(ctx) if err != nil { t.Fatalf("Could not start PostgreSQL container: %v", err) } defer container.Terminate(ctx) // Setup Test err = CreateUser(ctx, gormDB, "1") if err != nil { t.Fatal(err) } posts := []models.Post{ {BaseModel: models.BaseModel{ID: fmt.Sprintf("%-25s", "Post1")}, Rating: "safe"}, {BaseModel: models.BaseModel{ID: fmt.Sprintf("%-25s", "Post2")}, Rating: "safe"}, {BaseModel: models.BaseModel{ID: fmt.Sprintf("%-25s", "Post3")}, Rating: "explicit"}, } for _, post := range posts { err = CreateAnthrovePostNode(ctx, gormDB, models.AnthrovePostID(post.ID), post.Rating) if err != nil { t.Fatal(err) } err = EstablishUserToPostLink(ctx, gormDB, "1", models.AnthrovePostID(post.ID)) if err != nil { t.Fatal(err) } } tags := []models.Tag{ {Name: "JayTheFerret", Type: "artist"}, {Name: "Ferret", Type: "species"}, {Name: "Jay", Type: "character"}, } for i, tag := range tags { err = CreateTagNodeWitRelation(ctx, gormDB, models.AnthrovePostID(posts[i].ID), &tag) if err != nil { t.Fatal(err) } } expectedResult := []graphModels.TagsWithFrequency{ { Frequency: 1, Tags: graphModels.AnthroveTag{ Name: tags[0].Name, Type: string(tags[0].Type), }, }, { Frequency: 1, Tags: graphModels.AnthroveTag{ Name: tags[1].Name, Type: string(tags[1].Type), }, }, { Frequency: 1, Tags: graphModels.AnthroveTag{ Name: tags[2].Name, Type: string(tags[2].Type), }, }, } // Test type args struct { ctx context.Context db *gorm.DB anthroveUserID models.AnthroveUserID } tests := []struct { name string args args want []graphModels.TagsWithFrequency wantErr bool }{ { name: "", args: args{ ctx: ctx, db: gormDB, anthroveUserID: "1", }, want: expectedResult, wantErr: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { got, err := GetUserTagNodeWitRelationToFavedPosts(tt.args.ctx, tt.args.db, tt.args.anthroveUserID) if (err != nil) != tt.wantErr { t.Errorf("GetUserTagNodeWitRelationToFavedPosts() error = %v, wantErr %v", err, tt.wantErr) return } if !reflect.DeepEqual(got, tt.want) { t.Errorf("GetUserTagNodeWitRelationToFavedPosts() got = %v, want %v", got, tt.want) } }) } }