1
1
mirror of https://github.com/go-gitea/gitea synced 2025-01-19 14:14:26 +00:00
gitea/modules/auth/openid/discovery_cache_test.go

49 lines
1.1 KiB
Go
Raw Normal View History

2017-03-17 15:16:08 +01:00
// Copyright 2017 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT
2017-03-17 15:16:08 +01:00
package openid
import (
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
2017-03-17 15:16:08 +01:00
)
2017-03-21 01:55:00 +01:00
type testDiscoveredInfo struct{}
2017-03-17 15:16:08 +01:00
func (s *testDiscoveredInfo) ClaimedID() string {
return "claimedID"
}
2017-03-17 15:16:08 +01:00
func (s *testDiscoveredInfo) OpEndpoint() string {
return "opEndpoint"
}
2017-03-17 15:16:08 +01:00
func (s *testDiscoveredInfo) OpLocalID() string {
return "opLocalID"
}
func TestTimedDiscoveryCache(t *testing.T) {
2017-03-21 01:55:00 +01:00
dc := newTimedDiscoveryCache(1 * time.Second)
2017-03-17 15:16:08 +01:00
// Put some initial values
dc.Put("foo", &testDiscoveredInfo{}) // openid.opEndpoint: "a", openid.opLocalID: "b", openid.claimedID: "c"})
2017-03-17 15:16:08 +01:00
// Make sure we can retrieve them
di := dc.Get("foo")
require.NotNil(t, di)
assert.Equal(t, "opEndpoint", di.OpEndpoint())
assert.Equal(t, "opLocalID", di.OpLocalID())
assert.Equal(t, "claimedID", di.ClaimedID())
2017-03-17 15:16:08 +01:00
// Attempt to get a non-existent value
assert.Nil(t, dc.Get("bar"))
2017-03-17 15:16:08 +01:00
2019-06-12 21:41:28 +02:00
// Sleep one second and try retrieve again
2017-03-17 15:16:08 +01:00
time.Sleep(1 * time.Second)
assert.Nil(t, dc.Get("foo"))
2017-03-17 15:16:08 +01:00
}