pipu-music/services/api/internal/appupdate/appupdate_test.go

package appupdate

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"encoding/json"
	"log/slog"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestPickClientFiles(t *testing.T) {
	files := pickClientFiles([]asset{
		{Name: "pipu-music-1.3.0.apk", Size: 10},
		{Name: "pipu-music-1.3.0.apk.sha256", Size: 65, URL: "https://gitea/sum"},
		{Name: "pipu-music_1.3.0_x64-setup.exe", Size: 20},
		{Name: "pipu-music_1.3.0_amd64.deb", Size: 11},
		{Name: "pipu-music_1.3.0_amd64.AppImage", Size: 30},
		{Name: "pipu-music_1.3.0_aarch64.dmg", Size: 40},
	})
	assert.Equal(t, "pipu-music-1.3.0.apk", files["android"].Name)
	assert.Equal(t, "https://gitea/sum", files["android"].ChecksumURL)
	assert.Equal(t, "pipu-music_1.3.0_x64-setup.exe", files["windows"].Name)
	assert.Equal(t, "pipu-music_1.3.0_amd64.AppImage", files["linux"].Name)
	assert.Equal(t, "pipu-music_1.3.0_aarch64.dmg", files["macos"].Name)
}

func TestVersionCode(t *testing.T) {
	cases := map[string]int{"v0.6.1": 60_199, "1.2.0-rc.3": 1_020_003, "v10.0.0": 10_000_099}
	for tag, want := range cases {
		_, got, err := VersionCode(tag)
		require.NoError(t, err, tag)
		assert.Equal(t, want, got, tag)
	}
	for _, bad := range []string{"latest", "v1.100.0", "1.2"} {
		_, _, err := VersionCode(bad)
		assert.Error(t, err, bad)
	}
}

func TestLatestAndDownload(t *testing.T) {
	apk := []byte("fake apk bytes")
	sum := sha256.Sum256(apk)
	var srv *httptest.Server
	srv = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		assert.Equal(t, "token secret", r.Header.Get("Authorization"))
		switch r.URL.Path {
		case "/api/v1/repos/o/r/releases/latest":
			_ = json.NewEncoder(w).Encode(map[string]any{
				"tag_name": "v0.6.1", "body": "## Новое",
				"assets": []map[string]any{
					{"name": "pipu-music-0.6.1.apk", "size": len(apk), "browser_download_url": srv.URL + "/apk"},
					{"name": "pipu-music-0.6.1.apk.sha256", "size": 65, "browser_download_url": srv.URL + "/sum"},
				},
			})
		case "/apk":
			_, _ = w.Write(apk)
		case "/sum":
			_, _ = w.Write([]byte(hex.EncodeToString(sum[:]) + "\n"))
		default:
			http.NotFound(w, r)
		}
	}))
	defer srv.Close()

	dir := t.TempDir()
	h := NewHandler(Config{GiteaURL: srv.URL, Repo: "o/r", Token: "secret", Dir: dir, MinSupported: 60_000}, slog.Default())
	l, err := h.latest(context.Background())
	require.NoError(t, err)
	assert.Equal(t, "0.6.1", l.Version)
	assert.Equal(t, 60_199, l.VersionCode)
	android := l.Files["android"]
	assert.Equal(t, hex.EncodeToString(sum[:]), android.SHA256)

	path, err := h.ensure(context.Background(), android)
	require.NoError(t, err)
	got, _ := os.ReadFile(path)
	assert.Equal(t, apk, got)

	// Подменённый файл не пройдёт сверку.
	require.NoError(t, os.Remove(path))
	android.SHA256 = "00" + android.SHA256[2:]
	_, err = h.ensure(context.Background(), android)
	assert.ErrorContains(t, err, "checksum mismatch")
	_, statErr := os.Stat(filepath.Join(dir, "pipu-music-0.6.1.apk"))
	assert.True(t, os.IsNotExist(statErr))
}