fake_token_fetcher "github.com/cloudfoundry-incubator/uaa-token-fetcher/fakes" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" ) var _ = Describe(".UnRegister", func() { var ( client *fake_routing_api.FakeClient tokenFetcher *fake_token_fetcher.FakeTokenFetcher ) BeforeEach(func() { client = &fake_routing_api.FakeClient{} tokenFetcher = &fake_token_fetcher.FakeTokenFetcher{} tokenFetcher.FetchTokenReturns(&token_fetcher.Token{AccessToken: "token"}, nil) }) It("unregisters routes", func() { routes := []db.Route{{}} commands.UnRegister(client, tokenFetcher, routes) Expect(client.DeleteRoutesCallCount()).To(Equal(1)) Expect(client.DeleteRoutesArgsForCall(0)).To(Equal(routes)) }) It("fetches a token", func() { routes := []db.Route{{}} commands.UnRegister(client, tokenFetcher, routes) Expect(tokenFetcher.FetchTokenCallCount()).To(Equal(1)) }) })