122 lines
3.6 KiB
Go
122 lines
3.6 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"palette/internal/store"
|
|
)
|
|
|
|
// #95: HTML paste views dedupe per viewer (vwr cookie) within the burn viewer
|
|
// window, reusing the paste_views table. Repeated views by the same viewer do
|
|
// not inflate view_count; a different viewer counts.
|
|
func TestHTMLViewsDedupedPerViewer(t *testing.T) {
|
|
s := testServer(t)
|
|
h := s.routes()
|
|
|
|
req := httptest.NewRequest("POST", "/api/pastes", strings.NewReader(`{"content":"vd"}`))
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
id := jsonField(t, rec.Body.String(), "id")
|
|
|
|
view := func(vwr string) {
|
|
req := httptest.NewRequest("GET", "/"+id, nil)
|
|
if vwr != "" {
|
|
req.AddCookie(&http.Cookie{Name: "vwr", Value: vwr})
|
|
}
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
if rec.Code != 200 {
|
|
t.Fatalf("view: got %d", rec.Code)
|
|
}
|
|
}
|
|
count := func() string {
|
|
req := httptest.NewRequest("GET", "/api/pastes/"+id, nil)
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
return rec.Body.String()
|
|
}
|
|
|
|
// same viewer (same cookie) renders twice: still 1 view
|
|
view("v-a")
|
|
view("v-a")
|
|
if got := count(); !strings.Contains(got, `"view_count":1`) {
|
|
t.Fatalf("expected view_count 1 after duplicate viewer views, got: %s", got)
|
|
}
|
|
// a different viewer counts
|
|
view("v-b")
|
|
if got := count(); !strings.Contains(got, `"view_count":2`) {
|
|
t.Fatalf("expected view_count 2 after second viewer, got: %s", got)
|
|
}
|
|
// viewer without cookie counts every time (plain API clients, #49 decision)
|
|
view("")
|
|
view("")
|
|
if got := count(); !strings.Contains(got, `"view_count":4`) {
|
|
t.Fatalf("expected view_count 4 after uncoookied views, got: %s", got)
|
|
}
|
|
}
|
|
|
|
// #95: raw views always count, even from a cookie-holding viewer (#49 decision).
|
|
func TestRawViewsAlwaysCount(t *testing.T) {
|
|
s := testServer(t)
|
|
h := s.routes()
|
|
|
|
req := httptest.NewRequest("POST", "/api/pastes", strings.NewReader(`{"content":"rv"}`))
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
id := jsonField(t, rec.Body.String(), "id")
|
|
|
|
for i := 0; i < 3; i++ {
|
|
req := httptest.NewRequest("GET", "/raw/"+id, nil)
|
|
req.AddCookie(&http.Cookie{Name: "vwr", Value: "same-viewer"})
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
if rec.Code != 200 {
|
|
t.Fatalf("raw view: got %d", rec.Code)
|
|
}
|
|
}
|
|
req = httptest.NewRequest("GET", "/api/pastes/"+id, nil)
|
|
rec = httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
if got := rec.Body.String(); !strings.Contains(got, `"view_count":3`) {
|
|
t.Fatalf("expected view_count 3 after 3 raw views by one viewer, got: %s", got)
|
|
}
|
|
}
|
|
|
|
// #95: after the dedupe window passes, the same viewer counts again.
|
|
func TestViewDedupeWindowExpires(t *testing.T) {
|
|
s := testServer(t)
|
|
h := s.routes()
|
|
|
|
req := httptest.NewRequest("POST", "/api/pastes", strings.NewReader(`{"content":"we"}`))
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
id := jsonField(t, rec.Body.String(), "id")
|
|
|
|
view := func() {
|
|
req := httptest.NewRequest("GET", "/"+id, nil)
|
|
req.AddCookie(&http.Cookie{Name: "vwr", Value: "v-w"})
|
|
rec := httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
if rec.Code != 200 {
|
|
t.Fatalf("view: got %d", rec.Code)
|
|
}
|
|
}
|
|
view()
|
|
view()
|
|
// jump past the 15-minute window
|
|
realNow := store.TimeNow
|
|
store.TimeNow = func() time.Time { return realNow().Add(16 * time.Minute) }
|
|
defer func() { store.TimeNow = realNow }()
|
|
view()
|
|
req = httptest.NewRequest("GET", "/api/pastes/"+id, nil)
|
|
rec = httptest.NewRecorder()
|
|
h.ServeHTTP(rec, req)
|
|
if got := rec.Body.String(); !strings.Contains(got, `"view_count":2`) {
|
|
t.Fatalf("expected view_count 2 after window expiry, got: %s", got)
|
|
}
|
|
}
|