cmsp/app_upload_test.go

package main

import (
	"context"
	"fmt"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"cmsp/internal/downloader"
	"cmsp/internal/erpgo"
	"cmsp/internal/store"
)

func TestVideoUploadUsesERPGoAndPersistsOutcome(t *testing.T) {
	for _, status := range []string{"succeeded", "unknown"} {
		t.Run(status, func(t *testing.T) {
			putCount := 0
			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.Header().Set("Content-Type", "application/json")
				if r.Header.Get("X-API-Key") != "fictional-key" {
					t.Error("missing API key")
				}
				switch {
				case r.Method == "GET" && r.URL.Path == "/api/v1/shopee/products/123/video":
					fmt.Fprint(w, `{"code":200,"data":{"shopeeId":"123","source":"huohanhan","fetchedAt":"2026-09-30T00:00:00Z","video":[]}}`)
				case r.Method == "PUT" && r.URL.Path == "/api/v1/shopee/products/123/video":
					putCount++
					if r.Header.Get("Idempotency-Key") == "" {
						t.Error("missing idempotency key")
					}
					w.WriteHeader(202)
					fmt.Fprint(w, `{"code":202,"data":{"operationId":"op-1","shopeeId":"123","status":"submitted"}}`)
				case r.Method == "GET" && r.URL.Path == "/api/v1/shopee/products/123/video/operation":
					fmt.Fprintf(w, `{"code":200,"data":{"operationId":"op-1","shopeeId":"123","status":%q,"videoUrl":"https://example.invalid/video.mp4"}}`, status)
				default:
					t.Errorf("unexpected request: %s %s", r.Method, r.URL.Path)
				}
			}))
			defer server.Close()
			db, err := store.Open(":memory:")
			if err != nil {
				t.Fatal(err)
			}
			defer db.Close()
			if err := db.UpsertProducts([]store.Product{{ID: "internal-1", ItemID: "123", UploadStatus: store.UploadPending}}, "2026-09-30 00:00:00"); err != nil {
				t.Fatal(err)
			}
			a := NewApp()
			a.db = db
			a.cfg.ERPGo.BaseURL, a.cfg.ERPGo.APIKey = server.URL, "fictional-key"
			a.cfg.Download.VideoDir = t.TempDir()
			a.probe = func(context.Context, string) (downloader.ProbeResult, error) {
				return downloader.ProbeResult{Duration: 20, FormatName: "mp4", Width: 640, Height: 480}, nil
			}
			dir := filepath.Join(a.cfg.Download.VideoDir, "123")
			if err := os.MkdirAll(dir, 0700); err != nil {
				t.Fatal(err)
			}
			if err := os.WriteFile(filepath.Join(dir, "video.mp4"), []byte("fictional-mp4"), 0600); err != nil {
				t.Fatal(err)
			}
			client, err := erpgo.NewClient(a.cfg.ERPGo, nil)
			if err != nil {
				t.Fatal(err)
			}
			a.uploadOneVideo(context.Background(), client, "internal-1", 1)
			p, _, err := db.GetProduct("internal-1")
			if err != nil {
				t.Fatal(err)
			}
			want := store.UploadPending
			if status == "succeeded" {
				want = store.UploadDone
			}
			if p.UploadStatus != want || putCount != 1 {
				t.Fatalf("status=%s put=%d", p.UploadStatus, putCount)
			}
			if status == "succeeded" {
				videos, err := db.ListVideos(p.ID)
				if err != nil || len(videos) != 1 || videos[0].Status != store.VideoStatusUploaded {
					t.Fatalf("videos=%+v err=%v", videos, err)
				}
			} else {
				a.uploadOneVideo(context.Background(), client, "internal-1", 1)
				if putCount != 1 {
					t.Fatalf("unknown outcome repeated PUT: %d", putCount)
				}
			}
		})
	}
}

func TestUploadPreviewFailureIsLoggedWithoutKey(t *testing.T) {
	const key = "fictional-secret-key"
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusUnauthorized)
		fmt.Fprint(w, `{"code":401,"errorCode":"API_KEY_INVALID","requestId":"fictional-request"}`)
	}))
	defer server.Close()
	db, err := store.Open(":memory:")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	if err := db.UpsertProducts([]store.Product{{ID: "internal-1", ItemID: "123"}}, "2026-09-30 00:00:00"); err != nil {
		t.Fatal(err)
	}
	a := NewApp()
	a.db = db
	a.cfg.ERPGo.BaseURL, a.cfg.ERPGo.APIKey = server.URL, key
	a.cfg.Download.VideoDir = t.TempDir()
	a.probe = func(context.Context, string) (downloader.ProbeResult, error) {
		return downloader.ProbeResult{Duration: 20, FormatName: "mp4", Width: 640, Height: 480}, nil
	}
	dir := filepath.Join(a.cfg.Download.VideoDir, "123")
	if err := os.MkdirAll(dir, 0700); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(filepath.Join(dir, "video.mp4"), []byte("fictional-mp4"), 0600); err != nil {
		t.Fatal(err)
	}
	if _, err := a.GetUploadPreview([]string{"internal-1"}); err == nil {
		t.Fatal("expected preview failure")
	}
	entries := a.GetLogs()
	if len(entries) != 1 || !strings.Contains(entries[0].Message, "API_KEY_INVALID") || strings.Contains(entries[0].Message, key) {
		t.Fatalf("wrong log: %+v", entries)
	}
}

func TestUploadStartFailureIsLogged(t *testing.T) {
	a := NewApp()
	db, err := store.Open(":memory:")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	a.db = db
	if err := a.UploadVideos([]string{"internal-1"}); err == nil {
		t.Fatal("expected unconfigured erpgo")
	}
	entries := a.GetLogs()
	if len(entries) != 1 || !strings.Contains(entries[0].Message, "ERPGo_NOT_CONFIGURED") {
		t.Fatalf("wrong log: %+v", entries)
	}
}

func TestUnknownCurrentVideoDoesNotSubmitUpload(t *testing.T) {
	putCount := 0
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == http.MethodPut {
			putCount++
		}
		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusBadGateway)
		fmt.Fprint(w, `{"code":502,"errorCode":"HHH_UPSTREAM_ERROR","stage":"check","submissionState":"not_submitted","requestId":"fictional-request"}`)
	}))
	defer server.Close()
	db, err := store.Open(":memory:")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	if err := db.UpsertProducts([]store.Product{{ID: "internal-1", ItemID: "123", UploadStatus: store.UploadPending}}, "2026-09-30 00:00:00"); err != nil {
		t.Fatal(err)
	}
	a := NewApp()
	a.db = db
	a.cfg.ERPGo.BaseURL, a.cfg.ERPGo.APIKey = server.URL, "fictional-key"
	a.cfg.Download.VideoDir = t.TempDir()
	a.probe = func(context.Context, string) (downloader.ProbeResult, error) {
		return downloader.ProbeResult{Duration: 20, FormatName: "mp4", Width: 640, Height: 480}, nil
	}
	dir := filepath.Join(a.cfg.Download.VideoDir, "123")
	if err := os.MkdirAll(dir, 0700); err != nil {
		t.Fatal(err)
	}
	videoPath := filepath.Join(dir, "video.mp4")
	if err := os.WriteFile(videoPath, []byte("fictional-mp4"), 0600); err != nil {
		t.Fatal(err)
	}
	if _, err := a.GetUploadPreview([]string{"internal-1"}); err == nil {
		t.Fatal("unknown state must block preview")
	}
	p, _, err := db.GetProduct("internal-1")
	if err != nil || p.UploadStatus != store.UploadPending || putCount != 0 {
		t.Fatalf("preview changed state: %+v %v put=%d", p, err, putCount)
	}
	if _, found, err := db.GetUploadOperation(p.ID); err != nil || found {
		t.Fatalf("preview created operation: %v %v", found, err)
	}
	client, err := erpgo.NewClient(a.cfg.ERPGo, nil)
	if err != nil {
		t.Fatal(err)
	}
	a.uploadOneVideo(context.Background(), client, p.ID, 2)
	p, _, err = db.GetProduct(p.ID)
	if err != nil || p.UploadStatus != store.UploadUnconfirmed || !strings.Contains(p.LastError, "fictional-request") || putCount != 0 {
		t.Fatalf("unknown state was not preserved: %+v %v put=%d", p, err, putCount)
	}
	if _, err := os.Stat(videoPath); err != nil {
		t.Fatalf("local video removed: %v", err)
	}
	if _, found, err := db.GetUploadOperation(p.ID); err != nil || found {
		t.Fatalf("unknown state created operation: %v %v", found, err)
	}
}