61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
|
package proxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/Silo-Server/silo-server/internal/config"
|
||
|
|
"github.com/Silo-Server/silo-server/internal/httpstream"
|
||
|
|
"github.com/Silo-Server/silo-server/internal/nodeconfig"
|
||
|
|
"github.com/Silo-Server/silo-server/internal/nodesessions"
|
||
|
|
"github.com/Silo-Server/silo-server/internal/streamtoken"
|
||
|
|
)
|
||
|
|
|
||
|
|
type latchCapturingRevocationStore struct {
|
||
|
|
latch *httpstream.CutLatch
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *latchCapturingRevocationStore) IsRevoked(string, int, time.Time) bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *latchCapturingRevocationStore) Refuse(http.ResponseWriter, string, int, time.Time) bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *latchCapturingRevocationStore) WatchAndCutContext(ctx context.Context, _ http.ResponseWriter, _ string, _ int, _ time.Time) func() {
|
||
|
|
s.latch = httpstream.CutLatchFrom(ctx)
|
||
|
|
return func() {}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandleRemuxCarriesCutLatchToRevocationWatcher(t *testing.T) {
|
||
|
|
const secret = "proxy-revocation-test-secret"
|
||
|
|
watcher := nodeconfig.NewWatcher(nil, nil, nil, nodeconfig.BootstrapOverrides{})
|
||
|
|
cfg := &config.Config{}
|
||
|
|
cfg.Auth.JWTSecret = secret
|
||
|
|
watcher.SetConfigForTest(cfg)
|
||
|
|
|
||
|
|
server := NewServer(watcher, nodesessions.NewTracker(nil, "http://proxy", "proxy", "proxy"))
|
||
|
|
revocation := &latchCapturingRevocationStore{}
|
||
|
|
server.SetRevocationStore(revocation)
|
||
|
|
|
||
|
|
token, err := streamtoken.Sign(streamtoken.Claims{
|
||
|
|
SessionID: "remux-latch",
|
||
|
|
MediaPath: t.TempDir() + "/missing.mkv",
|
||
|
|
}, secret, time.Minute)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
req := httptest.NewRequest(http.MethodGet, "/stream/remux/"+token, nil)
|
||
|
|
rr := httptest.NewRecorder()
|
||
|
|
|
||
|
|
server.Handler().ServeHTTP(rr, req)
|
||
|
|
|
||
|
|
if revocation.latch == nil {
|
||
|
|
t.Fatal("remux revocation watcher request context has no cut latch")
|
||
|
|
}
|
||
|
|
}
|