示例#1
0
func TestStatusItemHandlerNoItemId(t *testing.T) {

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{"access_token":"test","token_type":"bearer","expires_in":3600}`)
	}))
	defer ts.Close()

	tokenStore = tokenstore.NewTokenStore()
	tokenStore.Refresher(ts.URL, "", "")
	defer close(tokenStore.Refresh)

	req, err := http.NewRequest("GET", "/status/item/", nil)
	if err != nil {
		t.Fatal(err)
	}

	w := httptest.NewRecorder()
	statusItemHandler(w, req)

	if w.Code != http.StatusBadRequest {
		t.Errorf("Status handler didn't error %v when no item id provided", http.StatusBadRequest)
	}

	if w.Body.String() != "Error, you need to provide an ItemID. /status/item/[ItemID]\n" {
		t.Error("Status handler didn't return the correct information when no item id provided")
	}

}
示例#2
0
func TestStatusItemHandlerGoodResponseFromSierra(t *testing.T) {

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{"access_token":"test","token_type":"bearer","expires_in":3600}`)
	}))
	defer ts.Close()

	tokenStore = tokenstore.NewTokenStore()
	tokenStore.Refresher(ts.URL, "", "")
	defer close(tokenStore.Refresh)

	ts2 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{"id":2536252,"updatedDate":"2014-09-19T03:09:16Z","createdDate":"2007-05-11T18:37:00Z","deleted":false,"bibIds":[2401597],"location":{"code":"flr4 ","name":"Floor 4 Books"},"status":{"code":"-","display":"IN LIBRARY"},"barcode":"12016135026","callNumber":"|aJC578.R383|bG67 2007"}`)
	}))
	defer ts2.Close()

	//Get statusBibIDHandler to look at our mocked server
	oldAPIURL := *apiURL
	*apiURL = ts2.URL
	defer func() { *apiURL = oldAPIURL }()

	req, err := http.NewRequest("GET", "/status/item/2401597", nil)
	if err != nil {
		t.Fatal(err)
	}

	w := httptest.NewRecorder()
	statusItemHandler(w, req)

	if w.Code != http.StatusOK {
		t.Errorf("Status handler didn't return %v when provided with a good response.", http.StatusBadRequest)
	}
}
示例#3
0
func TestRawHandlerTestRewrite(t *testing.T) {

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{"access_token":"test","token_type":"bearer","expires_in":3600}`)
	}))
	defer ts.Close()

	tokenStore = tokenstore.NewTokenStore()
	tokenStore.Refresher(ts.URL, "", "")
	defer close(tokenStore.Refresh)

	req, err := http.NewRequest("GET", "/raw/?bibIds=1234", nil)
	if err != nil {
		t.Fatal(err)
	}

	oldAPIURL := *apiURL
	*apiURL = "http://apiurl.com/test/"
	defer func() { *apiURL = oldAPIURL }()

	rawRewriter(req)

	if req.URL.String() != "http://apiurl.com/test?bibIds=1234" {
		t.Error("The raw handler is not correctly rewriting the url")
	}

}
示例#4
0
func TestStatusBibHandlerBadURLParse(t *testing.T) {

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{"access_token":"test","token_type":"bearer","expires_in":3600}`)
	}))
	defer ts.Close()

	tokenStore = tokenstore.NewTokenStore()
	tokenStore.Refresher(ts.URL, "", "")
	defer close(tokenStore.Refresh)

	r, err := http.NewRequest("GET", "/status/bib/2401597", nil)
	if err != nil {
		t.Fatal(err)
	}

	w := httptest.NewRecorder()

	oldAPIURL := *apiURL
	*apiURL = ":"
	defer func() { *apiURL = oldAPIURL }()

	statusBibHandler(w, r)

	if w.Code != http.StatusInternalServerError {
		t.Error("Should have returned InternalServerError")
	}

}
示例#5
0
func TestGetTokenOrErrorFailTokenStoreTimeout(t *testing.T) {

	tokenStore = tokenstore.NewTokenStore()

	r, err := http.NewRequest("GET", "", nil)
	if err != nil {
		t.Fatal(err)
	}
	w := httptest.NewRecorder()

	time.Sleep(time.Second * 31)

	if _, err := getTokenOrError(w, r); err == nil {
		t.Error("Should have failed with unparseable URL.")
	}

	if w.Code != http.StatusInternalServerError {
		t.Error("Should have returned a InternalServerError")
	}

}
示例#6
0
func TestGetTokenOrErrorFailTokenStoreUninitialized(t *testing.T) {

	tokenStore = tokenstore.NewTokenStore()
	tokenStore.Refresher(":", "", "")
	defer close(tokenStore.Refresh)

	r, err := http.NewRequest("GET", "", nil)
	if err != nil {
		t.Fatal(err)
	}
	w := httptest.NewRecorder()

	if _, err := getTokenOrError(w, r); err == nil {
		t.Error("Should have failed with unparseable URL.")
	}

	if w.Code != http.StatusInternalServerError {
		t.Error("Should have returned a InternalServerError")
	}

}
示例#7
0
文件: main.go 项目: cu-library/tyro
	apiURL       = flag.String("url", sierraapi.DefaultURL, "API url.")
	certFile     = flag.String("certfile", "", "Certificate file location.")
	keyFile      = flag.String("keyfile", "", "Private key file location.")
	clientKey    = flag.String("key", "", "Client Key")
	clientSecret = flag.String("secret", "", "Client Secret")
	headerACAO   = flag.String("acaoheader", DefaultACAOHeader, "Access-Control-Allow-Origin Header for CORS. Multiple origins separated by ;")
	raw          = flag.Bool("raw", DefaultRawAccess, "Allow access to the raw Sierra API under /raw/")
	newLimit     = flag.Int("newlimit", 16, "The number of items to serve from the /new endpoint.")

	logFileLocation = flag.String("logfile", l.DefaultLogFileLocation, "Log file. By default, log messages will be printed to stdout.")
	logMaxSize      = flag.Int("logmaxsize", l.DefaultLogMaxSize, "The maximum size of log files before they are rotated, in megabytes.")
	logMaxBackups   = flag.Int("logmaxbackups", l.DefaultLogMaxBackups, "The maximum number of old log files to keep.")
	logMaxAge       = flag.Int("logmaxage", l.DefaultLogMaxAge, "The maximum number of days to retain old log files, in days.")
	logLevel        = flag.String("loglevel", "warn", "The maximum log level which will be logged. error < warn < info < debug < trace. For example, trace will log everything, info will log info, warn, and error.")

	tokenStore = tokenstore.NewTokenStore()
)

func init() {

	flag.Usage = func() {
		fmt.Fprint(os.Stderr, "Tyro: A helper for Sierra APIs\nVersion 0.7.8\n\n")
		flag.PrintDefaults()
		fmt.Fprintln(os.Stderr, "  The possible environment variables:")

		flag.VisitAll(func(f *flag.Flag) {
			uppercaseName := strings.ToUpper(f.Name)
			fmt.Fprintf(os.Stderr, "  %v%v\n", EnvPrefix, uppercaseName)
		})

		fmt.Fprintln(os.Stderr, "If a certificate file is provided, Tyro will attempt to use HTTPS.")