Пример #1
0
func main() {
	fmt.Println()
	app := cli.NewApp()
	app.Name = "rtr"
	app.Usage = "A CLI for the Router API server."
	authors := []cli.Author{cli.Author{Name: "Cloud Foundry Routing Team", Email: "*****@*****.**"}}
	app.Authors = authors
	app.Commands = cliCommands
	app.CommandNotFound = commandNotFound
	app.Version = "2.1.0"

	cli.AppHelpTemplate = cli.AppHelpTemplate + environmentVariableHelp + "\n"

	trace.NewLogger(os.Getenv(RTR_TRACE))

	app.Run(os.Args)
	os.Exit(0)
}
Пример #2
0
					verifyBody("grant_type=client_credentials"),
					ghttp.RespondWithJSONEncoded(http.StatusOK, responseBody),
				))

			fetcher := NewTokenFetcher(cfg)
			token, err := fetcher.FetchToken()
			Expect(err).NotTo(HaveOccurred())
			Expect(server.ReceivedRequests()).Should(HaveLen(1))
			Expect(token.AccessToken).To(Equal("the token"))
			Expect(token.ExpireTime).To(Equal(20))
		})

		It("logs requests and responses", func() {
			stdout := bytes.NewBuffer([]byte{})
			trace.SetStdout(stdout)
			trace.NewLogger("true")

			server.AppendHandlers(
				ghttp.RespondWith(http.StatusBadRequest, "you messed up"),
			)

			fetcher := NewTokenFetcher(cfg)
			_, err := fetcher.FetchToken()
			Expect(err).To(HaveOccurred())

			r, err := ioutil.ReadAll(stdout)
			log := string(r)
			Expect(err).NotTo(HaveOccurred())
			Expect(log).To(ContainSubstring("REQUEST: "))
			Expect(log).To(ContainSubstring("POST /oauth/token HTTP/1.1"))
Пример #3
0
		})

		Describe("Next", func() {
			Context("When the event source returns an error", func() {
				It("returns the error", func() {
					fakeRawEventSource.NextReturns(sse.Event{}, errors.New("boom"))
					_, err := eventSource.Next()
					Expect(err.Error()).To(Equal("boom"))
				})
			})

			Context("When the event source successfully returns an event", func() {
				It("logs the event", func() {
					stdout := bytes.NewBuffer([]byte{})
					trace.SetStdout(stdout)
					trace.Logger = trace.NewLogger("true")
					rawEvent := sse.Event{
						ID:    "1",
						Name:  "Test",
						Data:  []byte(`{"route":"jim.com","port":8080,"ip":"1.1.1.1","ttl":60,"log_guid":"logs"}`),
						Retry: 1,
					}
					expectedJSON, _ := json.Marshal(rawEvent)

					fakeRawEventSource.NextReturns(rawEvent, nil)
					eventSource.Next()

					log, err := ioutil.ReadAll(stdout)
					Expect(err).NotTo(HaveOccurred())
					Expect(log).To(ContainSubstring("EVENT: "))
					Expect(log).To(ContainSubstring(string(expectedJSON)))