summaryrefslogtreecommitdiff
path: root/internal/service/write_test.go
blob: 122ca78c597fc0888e896bd034de3200283116c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package service

import (
	"context"
	"errors"
	"testing"

	"codeberg.org/snonux/player/internal/clock"
	"codeberg.org/snonux/player/internal/model"
	"codeberg.org/snonux/player/internal/repository"
)

func TestWriteService_RestoreMedia(t *testing.T) {
	ctx := context.Background()

	tests := []struct {
		name         string
		media        *model.Media
		deletedMedia []model.Media
		deletedErr   error
		storeErr     error
		wantErr      bool
		wantCode     error
	}{
		{
			name:  "ok",
			media: &model.Media{ID: 1, SetID: 1},
		},
		{
			name:         "soft deleted ok",
			deletedMedia: []model.Media{{ID: 1, SetID: 1}},
		},
		{
			name:       "deleted lookup error",
			deletedErr: errors.New("deleted lookup failed"),
			wantErr:    true,
		},
		{
			name:     "store error",
			media:    &model.Media{ID: 1, SetID: 1},
			storeErr: errors.New("boom"),
			wantErr:  true,
		},
		{
			name:     "not found",
			media:    nil,
			wantErr:  true,
			wantCode: ErrNotFound,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			store := &repository.MockStore{
				MediaRepo: repository.MockMediaRepo{
					GetMediaByIDFunc: func(ctx context.Context, id int64) (*model.Media, error) {
						return tt.media, nil
					},
					ListDeletedMediaFunc: func(ctx context.Context) ([]model.Media, error) {
						return tt.deletedMedia, tt.deletedErr
					},
					RestoreMediaFunc: func(ctx context.Context, id int64) error {
						return tt.storeErr
					},
				},
				UserRepo: repository.MockUserRepo{
					GetUserByIDFunc: func(ctx context.Context, id int64) (*model.User, error) {
						return &model.User{ID: id, IsAdmin: true}, nil
					},
				},
				SetRepo: repository.MockSetRepo{
					GetSetByIDFunc: func(ctx context.Context, id int64) (*model.Set, error) {
						return &model.Set{ID: id}, nil
					},
				},
				SetPermissionRepo: repository.MockSetPermissionRepo{
					GetPermissionFunc: func(ctx context.Context, setID, userID int64) (*model.SetPermission, error) {
						return nil, nil
					},
				},
			}
			svc := NewWriteService(store, clock.RealClock{}, "/tmp/media", nil, nil, &accessHelper{store: store})
			err := svc.RestoreMedia(ctx, 1, 1)
			if tt.wantErr {
				if err == nil {
					t.Fatal("expected error")
				}
				if tt.wantCode != nil {
					if !errors.Is(err, tt.wantCode) {
						t.Fatalf("expected error %v, got %v", tt.wantCode, err)
					}
				}
				return
			}
			if err != nil {
				t.Fatalf("unexpected error: %v", err)
			}
		})
	}
}