package handlers_test import ( "bytes" "database/sql" "encoding/json" "github.com/aclindsa/moneygo/internal/config" "github.com/aclindsa/moneygo/internal/db" "github.com/aclindsa/moneygo/internal/handlers" "io" "io/ioutil" "log" "net/http" "net/http/httptest" "os" "testing" ) var server *httptest.Server func Delete(client *http.Client, url string) (*http.Response, error) { request, err := http.NewRequest(http.MethodDelete, url, nil) if err != nil { return nil, err } return client.Do(request) } func Put(client *http.Client, url string, contentType string, body io.Reader) (*http.Response, error) { request, err := http.NewRequest(http.MethodPut, url, body) if err != nil { return nil, err } request.Header.Set("Content-Type", contentType) return client.Do(request) } type TransactType interface { Read(string) error } func create(client *http.Client, input, output TransactType, urlsuffix string) error { obj, err := json.MarshalIndent(input, "", " ") if err != nil { return err } response, err := client.Post(server.URL+urlsuffix, "application/json", bytes.NewReader(obj)) if err != nil { return err } body, err := ioutil.ReadAll(response.Body) response.Body.Close() if err != nil { return err } var e handlers.Error err = (&e).Read(string(body)) if err != nil { return err } if e.ErrorId != 0 || len(e.ErrorString) != 0 { return &e } err = output.Read(string(body)) if err != nil { return err } return nil } func read(client *http.Client, output TransactType, urlsuffix string) error { response, err := client.Get(server.URL + urlsuffix) if err != nil { return err } body, err := ioutil.ReadAll(response.Body) response.Body.Close() if err != nil { return err } var e handlers.Error err = (&e).Read(string(body)) if err != nil { return err } if e.ErrorId != 0 || len(e.ErrorString) != 0 { return &e } err = output.Read(string(body)) if err != nil { return err } return nil } func update(client *http.Client, input, output TransactType, urlsuffix string) error { obj, err := json.MarshalIndent(input, "", " ") if err != nil { return err } response, err := Put(client, server.URL+urlsuffix, "application/json", bytes.NewReader(obj)) if err != nil { return err } body, err := ioutil.ReadAll(response.Body) response.Body.Close() if err != nil { return err } var e handlers.Error err = (&e).Read(string(body)) if err != nil { return err } if e.ErrorId != 0 || len(e.ErrorString) != 0 { return &e } err = output.Read(string(body)) if err != nil { return err } return nil } func remove(client *http.Client, urlsuffix string) error { response, err := Delete(client, server.URL+urlsuffix) if err != nil { return err } body, err := ioutil.ReadAll(response.Body) response.Body.Close() if err != nil { return err } var e handlers.Error err = (&e).Read(string(body)) if err != nil { return err } if e.ErrorId != 0 || len(e.ErrorString) != 0 { return &e } return nil } func RunWith(t *testing.T, d *TestData, fn TestDataFunc) { testdata, err := d.Initialize() if err != nil { t.Fatalf("Failed to initialize test data: %s", err) } defer func() { err := testdata.Teardown() if err != nil { t.Fatal(err) } }() fn(t, testdata) } func RunTests(m *testing.M) int { dsn := db.GetDSN(config.SQLite, ":memory:") database, err := sql.Open("sqlite3", dsn) if err != nil { log.Fatal(err) } defer database.Close() dbmap, err := db.GetDbMap(database, config.SQLite) if err != nil { log.Fatal(err) } server = httptest.NewTLSServer(&handlers.APIHandler{DB: dbmap}) defer server.Close() return m.Run() } func TestMain(m *testing.M) { log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile) os.Exit(RunTests(m)) }