added basic test

main_test.go

@@ -0,0 +1,111 @@
+package main
+
+import (
+	"database/sql"
+	"flag"
+	"fmt"
+	"github.com/gchaincl/dotsql"
+	"os"
+	"testing"
+)
+
+func TestMain(m *testing.M) {
+	var err error
+	flag.Parse()
+
+	cfg, err = initConfig(os.Getenv("GOLANG_ENV"))
+	if err != nil {
+		// panic if the server is missing a vital configuration detail
+		panic(fmt.Errorf("server configuration error: %s", err.Error()))
+	}
+
+	teardown := setupTestDatabase()
+
+	retCode := m.Run()
+	teardown()
+	os.Exit(retCode)
+}
+
+func setupTestDatabase() func() {
+	var err error
+	appDB, err = SetupConnection(os.Getenv("POSTGRES_URL"))
+	if err != nil {
+		appDB.Close()
+		panic(err.Error())
+	}
+
+	teardown, err := initializeAppSchema(appDB)
+	if err != nil {
+		panic(err.Error())
+	}
+
+	if err := resetTestData(appDB,
+		"primers",
+		"sources",
+		"urls",
+		"links",
+		"metadata",
+		"snapshots",
+		"collections",
+		"archive_requests",
+		"uncrawlables"); err != nil {
+		panic(err.Error())
+	}
+
+	return teardown
+}
+
+// WARNING - THIS ZAPS WHATEVER DB IT'S GIVEN. DO NOT CALL THIS SHIT.
+// used for testing only, returns a teardown func
+func initializeAppSchema(db *sql.DB) (func(), error) {
+	schema, err := dotsql.LoadFromFile("sql/schema.sql")
+	if err != nil {
+		return nil, err
+	}
+
+	for _, cmd := range []string{
+		"drop-all",
+		"create-primers",
+		"create-sources",
+		"create-urls",
+		"create-links",
+		"create-metadata",
+		"create-snapshots",
+		"create-collections",
+		"create-archive_requests",
+		"create-uncrawlables",
+	} {
+		if _, err := schema.Exec(db, cmd); err != nil {
+			fmt.Println(cmd, "error:", err)
+			return nil, err
+		}
+	}
+
+	teardown := func() {
+		if _, err := schema.Exec(db, "drop-all"); err != nil {
+			panic(err.Error())
+		}
+	}
+
+	return teardown, nil
+}
+
+// drops test data tables & re-inserts base data from sql/test_data.sql, based on
+// passed in table names
+func resetTestData(db *sql.DB, tables ...string) error {
+	schema, err := dotsql.LoadFromFile("sql/test_data.sql")
+	if err != nil {
+		return err
+	}
+	for _, t := range tables {
+		if _, err := schema.Exec(db, fmt.Sprintf("delete-%s", t)); err != nil {
+			err = fmt.Errorf("error delete-%s: %s", t, err.Error())
+			return err
+		}
+		if _, err := schema.Exec(db, fmt.Sprintf("insert-%s", t)); err != nil {
+			err = fmt.Errorf("error insert-%s: %s", t, err.Error())
+			return err
+		}
+	}
+	return nil
+}

server.go

@@ -18,17 +18,7 @@ var (
 	appDB *sql.DB
 )
 
-func main() {
-	var err error
-	cfg, err = initConfig(os.Getenv("GOLANG_ENV"))
-	if err != nil {
-		// panic if the server is missing a vital configuration detail
-		panic(fmt.Errorf("server configuration error: %s", err.Error()))
-	}
-
-	connectToAppDb()
-
-	s := &http.Server{}
+func NewServerRoutes() *http.ServeMux {
 	m := http.NewServeMux()
 
 	m.HandleFunc("/", NotFoundHandler)
@@ -59,8 +49,22 @@ func main() {
 	// m.Handle("/v0/collections", middleware())
 	// m.Handle("/v0/collections/", middleware())
 
+	return m
+}
+
+func main() {
+	var err error
+	cfg, err = initConfig(os.Getenv("GOLANG_ENV"))
+	if err != nil {
+		// panic if the server is missing a vital configuration detail
+		panic(fmt.Errorf("server configuration error: %s", err.Error()))
+	}
+
+	connectToAppDb()
+
+	s := &http.Server{}
 	// connect mux to server
-	s.Handler = m
+	s.Handler = NewServerRoutes()
 
 	// print notable config settings
 	printConfigInfo()

server_test.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestServer(t *testing.T) {
+	s := httptest.NewServer(NewServerRoutes())
+	defer s.Close()
+
+	res, err := http.Get(s.URL + "/status")
+	if err != nil {
+		t.Error(err.Error())
+		return
+	}
+	defer res.Body.Close()
+	if res.StatusCode != http.StatusOK {
+		t.Errorf("status code mismatch. expected: %d, got: %d", http.StatusOK, res.StatusCode)
+	}
+}