Files
bare/internal/api/push_test.go
T
mayatnikovandClaude Opus 5 8f67f4aa4d Этап 4: PWA и пуши — service worker, Web Push с VAPID, баннер установки
Сервер: пакет push на webpush-go (третья и последняя прямая зависимость),
подписка устройства, правило ADR-023 «пуш только молчащему устройству и только
один» через атомарный захват push_pending, удаление подписки на 404 и 410,
TTL сутки, urgency normal. В нагрузке только {title, body, chat} — тело
собирается из константы, плейнтекст туда не попадает даже по ошибке.

Клиент: service worker с версионированным кэшем оболочки и никогда — /api/*,
push и notificationclick, подписка на VAPID-ключ сервера, запрос разрешения
после первого отправленного сообщения, разделы настроек «уведомления»
и «установить приложение», баннер установки на iOS.

ADR-045: пуш адресован получателю — по букве ADR-023 он уходил бы и молчащему
устройству отправителя с бессмысленным заголовком из собственного ника.
ADR-047: сервер ходит на endpoint подписки, который выбирает браузер. Проверка
«только https» обходилась редиректом, а имя могло смотреть внутрь сети — теперь
запрет редиректов и проверка разрешённого адреса на уровне сокета.
ADR-048: пределы отправки — недоступный push-сервис одного аккаунта больше
не съедает пуши всего сервера.
ADR-049: явно выключенные уведомления сами не включаются обратно.

Попутно: webpush-go дописывает набивку в переданный срез, а одна нагрузка
уходила всем устройствам сообщения — гонка, пойманная go test -race.
Теперь у каждого задания своя копия.

Приёмка на боевом: подписки, hasPush, чужое устройство, Origin, оболочка
из девяти файлов, Service-Worker-Allowed. Отдельно шесть непубличных адресов
и endpoint на 3 КиБ — все отбиты.

Чеклист ручной проверки на iPhone и Android — в docs/plan.md.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_015DbCjVfTFq4ZFG8juD45YJ
2026-08-23 00:50:58 +03:00

807 lines
30 KiB
Go

package api_test
import (
"bytes"
"context"
"crypto/aes"
"crypto/cipher"
"crypto/ecdh"
"crypto/hkdf"
"crypto/rand"
"crypto/sha256"
"encoding/base64"
"encoding/json"
"io"
"net/http"
"net/http/httptest"
"strings"
"sync"
"sync/atomic"
"testing"
"time"
"github.com/xmatic-squad/bare/internal/config"
)
// quiet — сколько ждём, чтобы убедиться, что пуша нет. Всё локально,
// задержек быть не должно.
const quiet = 300 * time.Millisecond
// pushEnv — сервер с настоящей парой VAPID-ключей: без неё пуши выключены
// (docs/deploy.md).
func pushEnv(t *testing.T) *env { return pushEnvWith(t, true) }
// pushEnvWith — то же; local разрешает отправку на 127.0.0.1, где живёт
// подменный push-сервис. Настоящий сервер ходит только по публичным
// адресам (ADR-047), и это проверяется отдельно.
func pushEnvWith(t *testing.T, local bool) *env {
t.Helper()
key, err := ecdh.P256().GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("vapid: %v", err)
}
return envWith(t, func(cfg *config.Config) {
cfg.VAPIDPublic = raw64(key.PublicKey().Bytes())
cfg.VAPIDPrivate = raw64(key.Bytes())
cfg.VAPIDSubject = "mailto:bare@bare.test"
// Push-сервис вендора подменён сервером на 127.0.0.1: в работе
// отправщик ходит только по публичным адресам (ADR-047).
cfg.PushLocal = local
})
}
func raw64(b []byte) string { return base64.RawURLEncoding.EncodeToString(b) }
// padded64 — то же, что bytesOf, но с паддингом: браузер вправе прислать
// ключи подписки и в такой форме.
func padded64(n int, seed byte) string {
raw := make([]byte, n)
for i := range raw {
raw[i] = seed + byte(i)
}
return base64.URLEncoding.EncodeToString(raw)
}
// point65 — p256dh настоящей подписки: несжатая точка P-256 в 65 байтах.
// Случайные байты той же длины точкой не являются, и сервер их не примет
// (docs/protocol.md, «Устройства»).
func point65(t *testing.T) []byte {
t.Helper()
key, err := ecdh.P256().GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("ключ подписки: %v", err)
}
return key.PublicKey().Bytes()
}
// pushService — push-сервис вендора в тесте. Настоящий FCM тестам не нужен
// и не годится: проверяется, что уходит и что сервер делает с ответом.
type pushService struct {
t *testing.T
url string
got chan delivered
status atomic.Int32
}
// delivered — то, что увидел push-сервис.
type delivered struct {
device string // хвост endpoint: по нему видно, чей это пуш
ttl string
urgency string
encoding string
auth string
record []byte
}
func newPushService(t *testing.T) *pushService {
t.Helper()
open := make(chan struct{})
close(open)
return pushServiceWith(t, open)
}
// newSlowPushService — push-сервис, который принимает запрос и молчит,
// пока тест не отпустит его. Так видно, что делает сервер, пока отправка
// ещё идёт. Отпускать обязательно: иначе остановка сервера ждёт таймаута.
func newSlowPushService(t *testing.T) (*pushService, func()) {
t.Helper()
gate := make(chan struct{})
var once sync.Once
return pushServiceWith(t, gate), func() { once.Do(func() { close(gate) }) }
}
// pushServiceWith — push-сервис, отвечающий не раньше, чем закроется gate.
func pushServiceWith(t *testing.T, gate <-chan struct{}) *pushService {
t.Helper()
p := &pushService{t: t, got: make(chan delivered, 512)}
p.status.Store(http.StatusCreated)
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
record, _ := io.ReadAll(r.Body)
got := delivered{
device: strings.TrimPrefix(r.URL.Path, "/push/"),
ttl: r.Header.Get("TTL"),
urgency: r.Header.Get("Urgency"),
encoding: r.Header.Get("Content-Encoding"),
auth: r.Header.Get("Authorization"),
record: record,
}
select {
case p.got <- got:
default:
}
select {
case <-gate:
case <-r.Context().Done():
return
}
w.WriteHeader(int(p.status.Load()))
}))
t.Cleanup(srv.Close)
p.url = srv.URL
return p
}
// next — следующий пуш; его отсутствие — ошибка теста.
func (p *pushService) next() delivered {
p.t.Helper()
select {
case got := <-p.got:
return got
case <-time.After(wait):
p.t.Fatal("пуш не пришёл")
}
return delivered{}
}
// silent требует, чтобы других пушей не было.
func (p *pushService) silent() {
p.t.Helper()
select {
case got := <-p.got:
p.t.Fatalf("лишний пуш устройству %s", got.device)
case <-time.After(quiet):
}
}
// subscriber — устройство с push-подпиской. Ключи настоящие: тест
// расшифровывает пуш ровно так, как это сделал бы браузер (RFC 8291),
// и потому видит, что в нём лежит.
type subscriber struct {
device string
key *ecdh.PrivateKey
auth []byte
}
// subscribe кладёт подписку устройства прямо в базу. Через PUT её сюда
// не поставить: тестовый push-сервис живёт на http, а эндпоинт принимает
// только https. Форму подписки проверяют TestPushSubscription
// и TestPushSubscriptionForm, правила отправки от неё не зависят.
func (e *env) subscribe(nick, device string, p *pushService) *subscriber {
e.t.Helper()
key, err := ecdh.P256().GenerateKey(rand.Reader)
if err != nil {
e.t.Fatalf("ключ подписки: %v", err)
}
auth := make([]byte, 16)
if _, err := rand.Read(auth); err != nil {
e.t.Fatalf("секрет подписки: %v", err)
}
raw, err := json.Marshal(map[string]any{
"endpoint": p.url + "/push/" + device,
"keys": map[string]string{
"p256dh": raw64(key.PublicKey().Bytes()),
"auth": raw64(auth),
},
})
if err != nil {
e.t.Fatalf("подписка: %v", err)
}
set, err := e.st.SetPush(context.Background(), device, nick, string(raw))
if err != nil || !set {
e.t.Fatalf("SetPush: %v (поставлена: %v)", err, set)
}
return &subscriber{device: device, key: key, auth: auth}
}
// open расшифровывает пуш: aes128gcm по RFC 8291, как это делает браузер.
// Без расшифровки нельзя утверждать, что в пуше нет ничего лишнего.
func (s *subscriber) open(t *testing.T, record []byte) map[string]string {
t.Helper()
check := func(what string, err error) {
t.Helper()
if err != nil {
t.Fatalf("%s: %v", what, err)
}
}
// Заголовок записи: соль, размер записи, длина открытого ключа.
const header = 16 + 4 + 1
if len(record) < header {
t.Fatalf("запись короче заголовка: %d байт", len(record))
}
salt := record[:16]
keyLen := int(record[20])
if len(record) < header+keyLen {
t.Fatalf("запись короче ключа отправителя: %d байт", len(record))
}
sender, ct := record[header:header+keyLen], record[header+keyLen:]
remote, err := ecdh.P256().NewPublicKey(sender)
check("ключ отправителя", err)
shared, err := s.key.ECDH(remote)
check("ecdh", err)
info := append([]byte("WebPush: info\x00"), s.key.PublicKey().Bytes()...)
info = append(info, sender...)
ikm, err := hkdf.Key(sha256.New, shared, s.auth, string(info), 32)
check("ikm", err)
cek, err := hkdf.Key(sha256.New, ikm, salt, "Content-Encoding: aes128gcm\x00", 16)
check("ключ записи", err)
nonce, err := hkdf.Key(sha256.New, ikm, salt, "Content-Encoding: nonce\x00", 12)
check("nonce", err)
block, err := aes.NewCipher(cek)
check("aes", err)
gcm, err := cipher.NewGCM(block)
check("gcm", err)
plain, err := gcm.Open(nil, nonce, ct, nil)
check("расшифровка", err)
// Хвост записи — набивка: нули после разделителя 0x02.
plain = bytes.TrimSuffix(bytes.TrimRight(plain, "\x00"), []byte{2})
var out map[string]string
if err := json.Unmarshal(plain, &out); err != nil {
t.Fatalf("нагрузка %q: %v", plain, err)
}
return out
}
// hasPush — что о подписке устройства говорит GET /api/devices.
func (e *env) hasPush(c *http.Cookie, device string) bool {
e.t.Helper()
rec := e.do(http.MethodGet, "/api/devices", nil, with(c))
expect(e.t, rec, http.StatusOK, "")
var list []struct {
ID string `json:"id"`
HasPush bool `json:"hasPush"`
}
decodeBody(e.t, rec, &list)
for _, got := range list {
if got.ID == device {
return got.HasPush
}
}
e.t.Fatalf("устройства %s нет в списке", device)
return false
}
// waitPushGone ждёт, пока подписка исчезнет: снимает её отправщик, уже
// после того, как push-сервис ответил.
func (e *env) waitPushGone(c *http.Cookie, device string) {
e.t.Helper()
for deadline := time.Now().Add(wait); time.Now().Before(deadline); {
if !e.hasPush(c, device) {
return
}
time.Sleep(5 * time.Millisecond)
}
e.t.Fatalf("подписка устройства %s не снята", device)
}
// send — обычная отправка личного сообщения.
func (e *env) send(c *http.Cookie, device, to string, seed byte) {
e.t.Helper()
expect(e.t, e.do(http.MethodPost, "/api/messages", message(ulid(nowMillis(), seed), to),
with(c), withDevice(device)), http.StatusAccepted, "")
}
// pushEventually шлёт сообщения, пока не придёт пуш. И разрыв потока,
// и возврат права на пуш случаются после ответа на запрос: момент их
// наступления не назначить, поэтому попытка повторяется.
func (e *env) pushEventually(p *pushService, c *http.Cookie, device, to string) delivered {
e.t.Helper()
for i := 0; i < 8; i++ {
e.send(c, device, to, byte(50+i))
select {
case got := <-p.got:
return got
case <-time.After(200 * time.Millisecond):
}
}
e.t.Fatal("пуш так и не пришёл")
return delivered{}
}
// subscription — тело PUT /api/devices/{id}/push в форме
// PushSubscription.toJSON().
func subscription(t *testing.T) map[string]any {
t.Helper()
return map[string]any{
"endpoint": "https://push.example/one",
"expirationTime": nil,
"keys": map[string]string{"p256dh": raw64(point65(t)), "auth": bytesOf(16, 7)},
}
}
// Подписка ставится и снимается, hasPush честный, чужое устройство — 403
// (docs/protocol.md, «Устройства», «Общие правила»).
func TestPushSubscription(t *testing.T) {
e := newEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
body := map[string]any{"subscription": subscription(t)}
if e.hasPush(marta, m1) {
t.Error("hasPush до подписки: true")
}
expect(t, e.do(http.MethodPut, "/api/devices/"+m1+"/push", body, with(marta), withDevice(m1)),
http.StatusNoContent, "")
if !e.hasPush(marta, m1) {
t.Error("hasPush после подписки: false")
}
// Подписка принадлежит устройству: у соседа её не появилось.
if e.hasPush(petya, p1) {
t.Error("подписка досталась чужому устройству")
}
expect(t, e.do(http.MethodDelete, "/api/devices/"+m1+"/push", nil, with(marta), withDevice(m1)),
http.StatusNoContent, "")
if e.hasPush(marta, m1) {
t.Error("hasPush после снятия: true")
}
// Снимать нечего — тот же 204.
expect(t, e.do(http.MethodDelete, "/api/devices/"+m1+"/push", nil, with(marta), withDevice(m1)),
http.StatusNoContent, "")
// Чужое устройство в пути — 403, и подписки у него не появилось.
expect(t, e.do(http.MethodPut, "/api/devices/"+p1+"/push", body, with(marta), withDevice(m1)),
http.StatusForbidden, "unknown_device")
expect(t, e.do(http.MethodDelete, "/api/devices/"+p1+"/push", nil, with(marta), withDevice(m1)),
http.StatusForbidden, "unknown_device")
if e.hasPush(petya, p1) {
t.Error("подписка поставлена чужому устройству")
}
// X-Device обязателен и обязан быть своим.
for _, opts := range [][]func(*http.Request){
{with(marta)},
{with(marta), withDevice(p1)},
{with(marta), withDevice("мусор")},
{with(marta), withDevice(deviceOf(9))},
} {
expect(t, e.do(http.MethodPut, "/api/devices/"+m1+"/push", body, opts...),
http.StatusForbidden, "unknown_device")
expect(t, e.do(http.MethodDelete, "/api/devices/"+m1+"/push", nil, opts...),
http.StatusForbidden, "unknown_device")
}
if e.hasPush(marta, m1) {
t.Error("подписка появилась после отказа")
}
}
// Форма подписки: абсолютный https-адрес и два ключа нужной длины.
func TestPushSubscriptionForm(t *testing.T) {
e := newEnv(t)
marta, m1 := e.join("marta", 1)
_, p1 := e.join("petya", 2)
cases := []struct {
name string
change func(map[string]any)
}{
{"нет endpoint", func(s map[string]any) { delete(s, "endpoint") }},
{"endpoint без tls", func(s map[string]any) { s["endpoint"] = "http://push.example/one" }},
{"endpoint без хоста", func(s map[string]any) { s["endpoint"] = "https:///one" }},
{"endpoint не url", func(s map[string]any) { s["endpoint"] = "какой же это url" }},
{"нет ключей", func(s map[string]any) { delete(s, "keys") }},
{"endpoint на loopback", func(s map[string]any) { s["endpoint"] = "https://127.0.0.1:9/push" }},
{"endpoint на link-local", func(s map[string]any) {
s["endpoint"] = "https://169.254.169.254/latest/meta-data/"
}},
{"endpoint в приватной сети", func(s map[string]any) { s["endpoint"] = "https://10.0.0.1/push" }},
{"endpoint на ::1", func(s map[string]any) { s["endpoint"] = "https://[::1]:8411/api/me" }},
{"endpoint длиннее 2 КиБ", func(s map[string]any) {
s["endpoint"] = "https://push.example/" + strings.Repeat("a", 2048)
}},
{"p256dh не 65 байт", func(s map[string]any) {
s["keys"] = map[string]string{"p256dh": bytesOf(32, 5), "auth": bytesOf(16, 7)}
}},
{"p256dh не точка на кривой", func(s map[string]any) {
s["keys"] = map[string]string{"p256dh": bytesOf(65, 5), "auth": bytesOf(16, 7)}
}},
{"auth не 16 байт", func(s map[string]any) {
s["keys"] = map[string]string{"p256dh": bytesOf(65, 5), "auth": bytesOf(32, 7)}
}},
{"ключ не base64url", func(s map[string]any) {
s["keys"] = map[string]string{"p256dh": strings.Repeat("!", 87), "auth": bytesOf(16, 7)}
}},
}
for _, c := range cases {
t.Run(c.name, func(t *testing.T) {
sub := subscription(t)
c.change(sub)
rec := e.do(http.MethodPut, "/api/devices/"+m1+"/push",
map[string]any{"subscription": sub}, with(marta), withDevice(m1))
expect(t, rec, http.StatusBadRequest, "invalid")
var field struct {
Field string `json:"field"`
}
decodeBody(t, rec, &field)
if field.Field != "subscription" {
t.Errorf("field: получено %q, ожидалось \"subscription\"", field.Field)
}
})
}
if e.hasPush(marta, m1) {
t.Error("подписка не по форме поставилась")
}
// Паддинг в base64url тоже принимается: браузер вправе его добавить.
padded := subscription(t)
padded["keys"] = map[string]string{
"p256dh": base64.URLEncoding.EncodeToString(point65(t)),
"auth": padded64(16, 7),
}
expect(t, e.do(http.MethodPut, "/api/devices/"+m1+"/push",
map[string]any{"subscription": padded}, with(marta), withDevice(m1)), http.StatusNoContent, "")
// Форма проверяется раньше прав: на запрос к чужому устройству
// приходит отказ по форме, а не по правам (ADR-043).
expect(t, e.do(http.MethodPut, "/api/devices/"+p1+"/push", "не json", with(marta), withDevice(m1)),
http.StatusBadRequest, "bad_json")
broken := subscription(t)
broken["endpoint"] = "http://push.example/one"
expect(t, e.do(http.MethodPut, "/api/devices/"+p1+"/push",
map[string]any{"subscription": broken}, with(marta), withDevice(m1)), http.StatusBadRequest, "invalid")
}
// Пуш уходит только отключённому устройству с подпиской (ADR-023).
// Он несёт заголовок, «новое сообщение» и адрес чата — и ничего больше.
func TestPushToSilentDevice(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
p2 := e.addDevice(petya, deviceOf(3))
e.subscribe("petya", p1, svc) // подключено по SSE
silent := e.subscribe("petya", p2, svc) // молчит
stream := e.open(p1, petya)
stream.untilReady()
e.send(marta, m1, "petya", 4)
got := svc.next()
if got.device != p2 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p2)
}
// TTL сутки, urgency normal (ADR-023).
if got.ttl != "86400" {
t.Errorf("TTL: получено %q, ожидалось \"86400\"", got.ttl)
}
if got.urgency != "normal" {
t.Errorf("Urgency: получено %q, ожидалось \"normal\"", got.urgency)
}
if got.encoding != "aes128gcm" {
t.Errorf("Content-Encoding: получено %q, ожидалось \"aes128gcm\"", got.encoding)
}
if !strings.HasPrefix(got.auth, "vapid t=") {
t.Errorf("Authorization: получено %q, ожидался vapid", got.auth)
}
payload := silent.open(t, got.record)
if len(payload) != 3 {
t.Errorf("поля нагрузки: %v", payload)
}
if payload["title"] != "@marta" {
t.Errorf("title: получено %q, ожидалось \"@marta\"", payload["title"])
}
if payload["body"] != "новое сообщение" {
t.Errorf("body: получено %q, ожидалось \"новое сообщение\"", payload["body"])
}
if payload["chat"] != "dm:marta" {
t.Errorf("chat: получено %q, ожидалось \"dm:marta\"", payload["chat"])
}
// Шифротекста сообщения в пуше нет ни в каком виде: сервер его
// не пересылает, а плейнтекста он и не знает (ADR-011).
if bytes.Contains(got.record, []byte(bytesOf(48, 23))) {
t.Error("шифротекст сообщения попал в пуш")
}
svc.silent()
}
// Одно молчащее устройство получает один пуш, а не ленту (ADR-023).
func TestPushOncePerSilentDevice(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
_, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
if got := svc.next(); got.device != p1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p1)
}
// Второе и третье сообщение подряд пуша не порождают.
e.send(marta, m1, "petya", 4)
e.send(marta, m1, "petya", 5)
svc.silent()
}
// Подключение по SSE сбрасывает неотработанный пуш: следующее сообщение
// молчащему устройству снова даёт пуш (ADR-023).
func TestPushAgainAfterStream(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
svc.next()
e.send(marta, m1, "petya", 4)
svc.silent()
stream := e.open(p1, petya)
stream.untilReady()
stream.close()
if got := e.pushEventually(svc, marta, m1, "petya"); got.device != p1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p1)
}
}
// 404 и 410 от push-сервиса означают, что подписки больше нет (ADR-011).
func TestPushDeadSubscription(t *testing.T) {
for _, status := range []int{http.StatusNotFound, http.StatusGone} {
t.Run(http.StatusText(status), func(t *testing.T) {
svc := newPushService(t)
svc.status.Store(int32(status))
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
svc.next()
e.waitPushGone(petya, p1)
})
}
}
// Прочие отказы push-сервиса подписку не трогают и доставку сообщения
// не роняют. Право на пуш при этом возвращается: иначе одна ошибка
// затыкала бы уведомления устройства до самого подключения.
func TestPushServiceError(t *testing.T) {
svc := newPushService(t)
svc.status.Store(http.StatusInternalServerError)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
svc.next()
if !e.hasPush(petya, p1) {
t.Error("подписка снята по ответу 500")
}
svc.status.Store(http.StatusCreated)
if got := e.pushEventually(svc, marta, m1, "petya"); got.device != p1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p1)
}
if !e.hasPush(petya, p1) {
t.Error("подписка снята после успешного пуша")
}
}
// Отправитель пуша о собственном сообщении не получает — ни на то
// устройство, с которого писал, ни на остальные свои (ADR-045).
func TestPushNotToSender(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
m2 := e.addDevice(marta, deviceOf(2))
_, p1 := e.join("petya", 3)
e.subscribe("marta", m1, svc)
e.subscribe("marta", m2, svc)
to := e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 4)
got := svc.next()
if got.device != p1 {
t.Fatalf("пуш ушёл устройству отправителя %s", got.device)
}
svc.silent()
if payload := to.open(t, got.record); payload["chat"] != "dm:marta" {
t.Errorf("chat: получено %q, ожидалось \"dm:marta\"", payload["chat"])
}
}
// Пуш из комнаты: заголовок — имя комнаты, адрес чата — её идентификатор
// (ADR-023).
func TestPushFromRoom(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
_, p1 := e.join("petya", 2)
room := e.makeRoom(marta, "marta", "общая", 40, withDevice(m1))
expect(t, e.changeMembers(marta, room.ID, []string{"petya"}, nil, []string{"marta", "petya"}, 41),
http.StatusOK, "")
member := e.subscribe("petya", p1, svc)
expect(t, e.do(http.MethodPost, "/api/messages",
roomMessage(ulid(nowMillis(), 5), room.ID, keyID(41)), with(marta), withDevice(m1)),
http.StatusAccepted, "")
got := svc.next()
if got.device != p1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p1)
}
payload := member.open(t, got.record)
if payload["title"] != "#общая" {
t.Errorf("title: получено %q, ожидалось \"#общая\"", payload["title"])
}
if payload["chat"] != "room:"+room.ID {
t.Errorf("chat: получено %q, ожидалось %q", payload["chat"], "room:"+room.ID)
}
if payload["body"] != "новое сообщение" {
t.Errorf("body: получено %q", payload["body"])
}
svc.silent()
}
// Без VAPID-ключей пуши выключены: подписка ставится, отправки нет.
func TestPushOffWithoutKeys(t *testing.T) {
svc := newPushService(t)
e := newEnv(t)
marta, m1 := e.join("marta", 1)
_, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
svc.silent()
}
// Аккаунт с молчащим push-сервисом не отбирает отправку у остальных:
// доля одного аккаунта в отправщиках ограничена (ADR-048).
func TestPushShareBetweenAccounts(t *testing.T) {
e := pushEnv(t)
stuck, release := newSlowPushService(t)
defer release()
live := newPushService(t)
marta, m1 := e.join("marta", 1)
greedy, g1 := e.join("greedy", 2)
e.subscribe("greedy", g1, stuck)
for seed := byte(10); seed < 30; seed++ {
e.subscribe("greedy", e.addDevice(greedy, deviceOf(seed)), stuck)
}
_, c1 := e.join("carol", 3)
e.subscribe("carol", c1, live)
// Двадцать одно молчащее устройство одного аккаунта: часть заданий
// отбрасывается сразу, остальные занимают не больше своей доли.
e.send(marta, m1, "greedy", 40)
e.send(marta, m1, "carol", 41)
select {
case got := <-live.got:
if got.device != c1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, c1)
}
case <-time.After(wait):
t.Fatal("пуш постороннему аккаунту не ушёл: отправщики заняты чужим")
}
}
// Устройство, подключившееся по SSE во время отправки, не остаётся
// с неотработанным пушем: право возвращается, и следующее сообщение
// после ухода в офлайн снова даёт пуш (ADR-023).
func TestPushReleasedWhenDeviceConnects(t *testing.T) {
e := pushEnv(t)
svc, release := newSlowPushService(t)
defer release()
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
// Отправка уже началась: push-сервис получил запрос и держит его.
if got := svc.next(); got.device != p1 {
t.Fatalf("пуш ушёл устройству %s, ожидалось %s", got.device, p1)
}
// Пока пуш в пути, устройство подключилось: подключение сбрасывает
// неотработанный пуш, а отправщик поставил его позже.
stream := e.open(p1, petya)
stream.untilReady()
release()
// Право на пуш свободно: захват удаётся.
for deadline := time.Now().Add(wait); ; {
claimed, ok, err := e.st.ClaimPush(context.Background(), p1)
if err != nil {
t.Fatalf("ClaimPush: %v", err)
}
if ok {
if claimed == "" {
t.Error("подписка пуста")
}
return
}
if time.Now().After(deadline) {
t.Fatal("неотработанный пуш остался висеть на подключённом устройстве")
}
time.Sleep(5 * time.Millisecond)
}
}
// Пуш на непубличный адрес не уходит вовсе: соединения не случается,
// право на пуш возвращается, а адрес подписки в журнал не попадает
// (ADR-047, docs/deploy.md, «Логи»).
func TestPushSkipsLocalEndpoint(t *testing.T) {
svc := newPushService(t)
e := pushEnvWith(t, false)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
e.subscribe("petya", p1, svc)
e.send(marta, m1, "petya", 3)
svc.silent()
if !e.hasPush(petya, p1) {
t.Error("подписка снята, хотя push-сервис не отвечал")
}
// Право на пуш вернулось: следующее сообщение попробует снова.
claimed, ok, err := e.st.ClaimPush(context.Background(), p1)
if err != nil {
t.Fatalf("ClaimPush: %v", err)
}
if !ok || claimed == "" {
t.Error("право на пуш осталось захваченным")
}
log := e.log.String()
if !strings.Contains(log, "адрес подписки не публичный") {
t.Errorf("в журнале нет причины отказа: %q", log)
}
if strings.Contains(log, "127.0.0.1") || strings.Contains(log, strings.TrimPrefix(svc.url, "http://")) {
t.Errorf("адрес подписки попал в журнал: %q", log)
}
}
// Одно сообщение — несколько молчащих устройств: каждое получает свою
// расшифровываемую нагрузку. Нагрузка на всех одна (ADR-045), но
// шифруется она для каждой подписки отдельно.
func TestPushPayloadPerDevice(t *testing.T) {
svc := newPushService(t)
e := pushEnv(t)
marta, m1 := e.join("marta", 1)
petya, p1 := e.join("petya", 2)
p2 := e.addDevice(petya, deviceOf(3))
p3 := e.addDevice(petya, deviceOf(4))
subs := map[string]*subscriber{
p1: e.subscribe("petya", p1, svc),
p2: e.subscribe("petya", p2, svc),
p3: e.subscribe("petya", p3, svc),
}
e.send(marta, m1, "petya", 5)
seen := make(map[string]bool)
for i := 0; i < len(subs); i++ {
got := svc.next()
to, ok := subs[got.device]
if !ok {
t.Fatalf("пуш ушёл неизвестному устройству %s", got.device)
}
if seen[got.device] {
t.Fatalf("устройство %s получило второй пуш", got.device)
}
seen[got.device] = true
payload := to.open(t, got.record)
if payload["title"] != "@marta" || payload["chat"] != "dm:marta" || payload["body"] != "новое сообщение" {
t.Errorf("нагрузка устройства %s: %v", got.device, payload)
}
}
svc.silent()
}