-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
tests: unify most static helpers (#8)
This gets rid of a lot repeated code for loading static data in unit testing.
- Loading branch information
Showing
6 changed files
with
120 additions
and
243 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,9 +1,111 @@ | ||
| package gtfs_test | ||
|
|
||
| // Most tests in this package run against the in-memory and storage | ||
| // Helpers and configuration for tests. | ||
| // | ||
| // Many tests in this package run against the in-memory and sqlite | ||
| // backends by default. If PostgresConnStr is set, they'll also run | ||
| // against postgres. | ||
|
|
||
| import ( | ||
| "archive/zip" | ||
| "bytes" | ||
| "io/ioutil" | ||
| "strings" | ||
| "testing" | ||
|
|
||
| "github.com/stretchr/testify/require" | ||
|
|
||
| "tidbyt.dev/gtfs" | ||
| "tidbyt.dev/gtfs/parse" | ||
| "tidbyt.dev/gtfs/storage" | ||
| ) | ||
|
|
||
| const ( | ||
| PostgresConnStr = "" // "postgres://postgres:mysecretpassword@localhost:5432/gtfs?sslmode=disable" | ||
| ) | ||
|
|
||
| func GFSTest_BuildStorage(t testing.TB, backend string) storage.Storage { | ||
| var s storage.Storage | ||
| var err error | ||
| if backend == "memory" { | ||
| s = storage.NewMemoryStorage() | ||
| } else if backend == "sqlite" { | ||
| s, err = storage.NewSQLiteStorage() | ||
| require.NoError(t, err) | ||
| } else if backend == "postgres" { | ||
| s, err = storage.NewPSQLStorage(PostgresConnStr, true) | ||
| require.NoError(t, err) | ||
| } | ||
| require.NotEqual(t, nil, s, "unknown backend %q", backend) | ||
|
|
||
| return s | ||
| } | ||
|
|
||
| func GTFSTest_LoadStatic(t testing.TB, backend string, buf *bytes.Buffer) (*gtfs.Static, storage.FeedReader) { | ||
| s := GFSTest_BuildStorage(t, backend) | ||
|
|
||
| // Parse buf into storage | ||
| feedWriter, err := s.GetWriter("test") | ||
| require.NoError(t, err) | ||
|
|
||
| metadata, err := parse.ParseStatic(feedWriter, buf.Bytes()) | ||
| require.NoError(t, err) | ||
|
|
||
| require.NoError(t, feedWriter.Close()) | ||
|
|
||
| // Create Static | ||
| reader, err := s.GetReader("test") | ||
| require.NoError(t, err) | ||
|
|
||
| static, err := gtfs.NewStatic(reader, metadata) | ||
| require.NoError(t, err) | ||
|
|
||
| return static, reader | ||
| } | ||
|
|
||
| func GTFSTest_LoadStaticFile(t testing.TB, backend string, filename string) (*gtfs.Static, storage.FeedReader) { | ||
| buf, err := ioutil.ReadFile(filename) | ||
| require.NoError(t, err) | ||
|
|
||
| return GTFSTest_LoadStatic(t, backend, bytes.NewBuffer(buf)) | ||
| } | ||
|
|
||
| func GTFSTest_BuildStatic( | ||
| t testing.TB, | ||
| backend string, | ||
| files map[string][]string, | ||
| ) (*gtfs.Static, storage.FeedReader) { | ||
|
|
||
| // Fill in missing files with (mostly blank) dummy data. | ||
| if files["agency.txt"] == nil { | ||
| files["agency.txt"] = []string{"agency_timezone,agency_name,agency_url", "UTC,FooAgency,http://example.com"} | ||
| } | ||
| if files["calendar.txt"] == nil && files["calendar_dates.txt"] == nil { | ||
| files["calendar.txt"] = []string{"service_id"} | ||
| } | ||
| if files["routes.txt"] == nil { | ||
| files["routes.txt"] = []string{"route_id"} | ||
| } | ||
| if files["trips.txt"] == nil { | ||
| files["trips.txt"] = []string{"trip_id"} | ||
| } | ||
| if files["stops.txt"] == nil { | ||
| files["stops.txt"] = []string{"stop_id"} | ||
| } | ||
| if files["stop_times.txt"] == nil { | ||
| files["stop_times.txt"] = []string{"stop_id"} | ||
| } | ||
|
|
||
| // Create zip | ||
| buf := &bytes.Buffer{} | ||
| w := zip.NewWriter(buf) | ||
| for filename, content := range files { | ||
| f, err := w.Create(filename) | ||
| require.NoError(t, err) | ||
| _, err = f.Write([]byte(strings.Join(content, "\n"))) | ||
| require.NoError(t, err) | ||
| } | ||
| require.NoError(t, w.Close()) | ||
|
|
||
| return GTFSTest_LoadStatic(t, backend, buf) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.