. "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() {