mirror of https://github.com/ory/hydra
38 lines
789 B
Go
38 lines
789 B
Go
// Copyright © 2022 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/ory/x/pagination/tokenpagination"
|
|
)
|
|
|
|
func TestFatal(t *testing.T) {
|
|
oldOsExit := osExit
|
|
defer func() { osExit = oldOsExit }()
|
|
|
|
var got int
|
|
myExit := func(code int) {
|
|
got = code
|
|
}
|
|
|
|
osExit = myExit
|
|
fatal("Fatal message")
|
|
if exp := 1; got != exp {
|
|
t.Errorf("Expected exit code: %d, got: %d", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestGetPageToken(t *testing.T) {
|
|
u, _ := url.Parse("https://example.com/foobar")
|
|
rec := httptest.NewRecorder()
|
|
tokenpagination.PaginationHeader(rec, u, 100, 3, 10)
|
|
assert.Equal(t, `eyJvZmZzZXQiOiI0MCIsInYiOjJ9`, getPageToken(rec.Result()), rec.Result().Header.Get("Link"))
|
|
}
|