// AllVehicles returns all of the vehicles in the system func AllVehicles(w http.ResponseWriter, req *http.Request) { w.Header().Set("Content-Type", "application/json") db, err := storage.CreateConnection() if err != nil { log.Println("Error connecting to DB in AllVehicles") return } defer db.Close() vd, err := storage.GetVehiclesAfterTime(db, time.Now().Add(time.Minute*-4)) compacted := utils.CompactVehicles(vd) out, err := json.Marshal(compacted) if err != nil { log.Println(err) } fmt.Fprint(w, string(out)) }
func TestFetchTransit(t *testing.T) { fakeServer := makeFakeServer() muni.SetConfig(muni.TransitConfig{DefaultURL: fakeServer.URL + "/"}) storage.RunStorageTest(t, func(db *sqlx.DB, t *testing.T) { err := fetchTransit([]string{"N"}) if err != nil { t.Error(err) } v, _ := storage.GetVehiclesAfterTime(db, time.Unix(1460498787925/1000, 0).Add(time.Minute*-5)) if len(v) == 0 { t.Error("Did not add any vehicles to the DB") } if len(v) == 37 { t.Error("Did not add all of the vehicles to the DB") } }) }