j1 fix admin media lifecycle QA defects

1 files changed, 19 insertions, 5 deletions

diff --git a/internal/service/write_test.go b/internal/service/write_test.go
index d303431..122ca78 100644
--- a/internal/service/write_test.go
+++ b/internal/service/write_test.go
@@ -14,17 +14,28 @@ func TestWriteService_RestoreMedia(t *testing.T) {
 	ctx := context.Background()
 
 	tests := []struct {
-		name     string
-		media    *model.Media
-		storeErr error
-		wantErr  bool
-		wantCode error
+		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"),
@@ -45,6 +56,9 @@ func TestWriteService_RestoreMedia(t *testing.T) {
 					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
 					},