35 lines
898 B
Go
35 lines
898 B
Go
package response
|
|
|
|
import "testing"
|
|
|
|
func TestWithDataWrapsSlicesAndMergesExtra(t *testing.T) {
|
|
resp := WithData([]string{"a", "b"}, map[string]interface{}{
|
|
"total": 2,
|
|
"page": 1,
|
|
})
|
|
|
|
data, ok := resp.Data.(map[string]interface{})
|
|
if !ok {
|
|
t.Fatalf("expected map payload, got %T", resp.Data)
|
|
}
|
|
if data["total"] != 2 {
|
|
t.Fatalf("expected total=2, got %v", data["total"])
|
|
}
|
|
items, ok := data["items"].([]string)
|
|
if !ok || len(items) != 2 {
|
|
t.Fatalf("expected items slice to be preserved, got %#v", data["items"])
|
|
}
|
|
}
|
|
|
|
func TestWithDataPreservesMapPayload(t *testing.T) {
|
|
resp := WithData(map[string]interface{}{"user": "alice"}, map[string]interface{}{"page": 1})
|
|
|
|
data := resp.Data.(map[string]interface{})
|
|
if data["user"] != "alice" {
|
|
t.Fatalf("expected user=alice, got %v", data["user"])
|
|
}
|
|
if data["page"] != 1 {
|
|
t.Fatalf("expected page=1, got %v", data["page"])
|
|
}
|
|
}
|