@@ -2,6 +2,7 @@ package sources
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"path/filepath"
|
||||
@@ -119,6 +120,115 @@ func TestSubscribeEventsBroadcastsToAllSubscribers(t *testing.T) {
|
||||
assertEvent("subscriber B", eventsB)
|
||||
}
|
||||
|
||||
func TestCheckOneResolvesNestedJSONMediaURL(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||
"ok": true,
|
||||
"data": map[string]any{
|
||||
"ignored": "https://example.test/readme.txt",
|
||||
"items": []map[string]any{
|
||||
{"name": "first"},
|
||||
{"cover": "/media/poster.webp"},
|
||||
},
|
||||
},
|
||||
})
|
||||
}))
|
||||
defer server.Close()
|
||||
cfg, store := testStore(t)
|
||||
service := NewService(cfg, store)
|
||||
item, err := store.UpsertSource(db.Source{
|
||||
CategoryID: "image",
|
||||
CategoryName: "Image",
|
||||
SourceID: "json-cover",
|
||||
Name: "JSON Cover",
|
||||
Method: "GET",
|
||||
APIURL: server.URL + "/api/random",
|
||||
TimeoutMS: 3000,
|
||||
CheckIntervalSec: 300,
|
||||
Enabled: true,
|
||||
ClientVisible: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := service.CheckOne(context.Background(), item); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
checked, err := store.GetSourceBySourceID("json-cover")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
meta := parseHealthMeta(checked.LastError)
|
||||
if meta["resolvedUrl"] != server.URL+"/media/poster.webp" {
|
||||
t.Fatalf("resolvedUrl = %#v, want relative media URL", meta["resolvedUrl"])
|
||||
}
|
||||
if meta["resolvedKey"] != "data.items.cover" {
|
||||
t.Fatalf("resolvedKey = %#v", meta["resolvedKey"])
|
||||
}
|
||||
if meta["mediaType"] != "image" {
|
||||
t.Fatalf("mediaType = %#v, want image", meta["mediaType"])
|
||||
}
|
||||
catalog, err := service.Catalog(false)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
categories := catalog["categories"].([]map[string]any)
|
||||
sub := categories[0]["subcategories"].([]map[string]any)[0]
|
||||
if sub["resolvedUrl"] != server.URL+"/media/poster.webp" {
|
||||
t.Fatalf("catalog resolvedUrl = %#v", sub["resolvedUrl"])
|
||||
}
|
||||
endpoints, err := service.Endpoints(false)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if endpoints[0]["resolvedUrl"] != server.URL+"/media/poster.webp" {
|
||||
t.Fatalf("endpoint resolvedUrl = %#v", endpoints[0]["resolvedUrl"])
|
||||
}
|
||||
if endpoints[0]["urlTemplate"] != server.URL+"/api/random" {
|
||||
t.Fatalf("urlTemplate changed: %#v", endpoints[0]["urlTemplate"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestCheckOneResolvesTextMediaURL(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
|
||||
_, _ = w.Write([]byte(`play: https://cdn.example.test/video/sample.mp4`))
|
||||
}))
|
||||
defer server.Close()
|
||||
cfg, store := testStore(t)
|
||||
service := NewService(cfg, store)
|
||||
item, err := store.UpsertSource(db.Source{
|
||||
CategoryID: "video",
|
||||
CategoryName: "Video",
|
||||
SourceID: "text-video",
|
||||
Name: "Text Video",
|
||||
Method: "GET",
|
||||
APIURL: server.URL,
|
||||
TimeoutMS: 3000,
|
||||
CheckIntervalSec: 300,
|
||||
Enabled: true,
|
||||
ClientVisible: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := service.CheckOne(context.Background(), item); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
checked, err := store.GetSourceBySourceID("text-video")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
meta := parseHealthMeta(checked.LastError)
|
||||
if meta["resolvedUrl"] != "https://cdn.example.test/video/sample.mp4" {
|
||||
t.Fatalf("resolvedUrl = %#v", meta["resolvedUrl"])
|
||||
}
|
||||
if meta["mediaType"] != "video" {
|
||||
t.Fatalf("mediaType = %#v, want video", meta["mediaType"])
|
||||
}
|
||||
}
|
||||
|
||||
func testStore(t *testing.T) (*config.Config, *db.Store) {
|
||||
t.Helper()
|
||||
dir := t.TempDir()
|
||||
|
||||
Reference in New Issue
Block a user