. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("loggregator logs repository", func() {
var (
fakeConsumer *logsfakes.FakeLoggregatorConsumer
logsRepo *LoggregatorLogsRepository
configRepo coreconfig.ReadWriter
authRepo *authenticationfakes.FakeRepository
)
BeforeEach(func() {
fakeConsumer = new(logsfakes.FakeLoggregatorConsumer)
configRepo = testconfig.NewRepositoryWithDefaults()
configRepo.SetLoggregatorEndpoint("loggregator-server.test.com")
configRepo.SetAccessToken("the-access-token")
authRepo = &authenticationfakes.FakeRepository{}
})
JustBeforeEach(func() {
logsRepo = NewLoggregatorLogsRepository(configRepo, fakeConsumer, authRepo)
})
Describe("RecentLogsFor", func() {
Context("when a noaa_errors.UnauthorizedError occurs", func() {
var recentCalled bool
BeforeEach(func() {
fakeConsumer.RecentStub = func(string, string) ([]*logmessage.LogMessage, error) {
if recentCalled {
return nil, nil
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("logs with noaa repository", func() {
var (
fakeNoaaConsumer *testapi.FakeNoaaConsumer
config coreconfig.ReadWriter
fakeTokenRefresher *authenticationfakes.FakeRepository
repo *logs.NoaaLogsRepository
)
BeforeEach(func() {
fakeNoaaConsumer = &testapi.FakeNoaaConsumer{}
config = testconfig.NewRepositoryWithDefaults()
config.SetLoggregatorEndpoint("loggregator.test.com")
config.SetDopplerEndpoint("doppler.test.com")
config.SetAccessToken("the-access-token")
fakeTokenRefresher = &authenticationfakes.FakeRepository{}
repo = logs.NewNoaaLogsRepository(config, fakeNoaaConsumer, fakeTokenRefresher)
})
Describe("Authentication Token Refresh", func() {
It("sets the noaa token refresher", func() {
Expect(fakeNoaaConsumer.RefreshTokenFromCallCount()).To(Equal(1))
Expect(fakeNoaaConsumer.RefreshTokenFromArgsForCall(0)).To(Equal(fakeTokenRefresher))
})
})
Describe("RecentLogsFor", func() {
Context("when an error does not occur", func() {
