43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
|
package transcodeproxy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAcknowledgeSurvivesCompletedDownstreamRequest(t *testing.T) {
|
||
|
|
const (
|
||
|
|
jwtSecret = "test-secret"
|
||
|
|
generation = "incarnation:generation"
|
||
|
|
)
|
||
|
|
|
||
|
|
acknowledged := make(chan struct{}, 1)
|
||
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if r.Method != http.MethodPost {
|
||
|
|
t.Errorf("method = %q, want POST", r.Method)
|
||
|
|
}
|
||
|
|
if got := r.Header.Get("Authorization"); got != "Bearer "+jwtSecret {
|
||
|
|
t.Errorf("Authorization = %q", got)
|
||
|
|
}
|
||
|
|
if got := r.Header.Get(GenerationHeader); got != generation {
|
||
|
|
t.Errorf("generation = %q, want %q", got, generation)
|
||
|
|
}
|
||
|
|
acknowledged <- struct{}{}
|
||
|
|
w.WriteHeader(http.StatusNoContent)
|
||
|
|
}))
|
||
|
|
t.Cleanup(server.Close)
|
||
|
|
|
||
|
|
requestCtx, cancelRequest := context.WithCancel(context.Background())
|
||
|
|
cancelRequest()
|
||
|
|
if err := Acknowledge(requestCtx, server.Client(), server.URL+"/segment", jwtSecret, generation); err != nil {
|
||
|
|
t.Fatalf("Acknowledge() error = %v", err)
|
||
|
|
}
|
||
|
|
select {
|
||
|
|
case <-acknowledged:
|
||
|
|
default:
|
||
|
|
t.Fatal("acknowledgement request was not delivered")
|
||
|
|
}
|
||
|
|
}
|