diff --git a/services/container_test.go b/services/container_test.go index 63a5fa1..fd40048 100644 --- a/services/container_test.go +++ b/services/container_test.go @@ -1,58 +1,11 @@ package services import ( - "context" - "net/http" - "net/http/httptest" - "os" - "strings" "testing" - "goweb/config" - "goweb/ent" - - "github.com/labstack/echo/v4" - "github.com/stretchr/testify/assert" ) -var ( - c *Container - ctx echo.Context - usr *ent.User - rec *httptest.ResponseRecorder -) - -func TestMain(m *testing.M) { - // Set the environment to test - config.SwitchEnvironment(config.EnvTest) - - // Create a new container - c = NewContainer() - - // Create a web context - req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("")) - rec = httptest.NewRecorder() - ctx = c.Web.NewContext(req, rec) - - // Create a test user - var err error - usr, err = c.ORM.User. - Create(). - SetEmail("test@test.dev"). - SetPassword("abc"). - SetName("Test User"). - Save(context.Background()) - - if err != nil { - panic(err) - } - - // Run tests - exitVal := m.Run() - os.Exit(exitVal) -} - func TestNewContainer(t *testing.T) { assert.NotNil(t, c.Web) assert.NotNil(t, c.Config) diff --git a/services/services_test.go b/services/services_test.go new file mode 100644 index 0000000..b048d06 --- /dev/null +++ b/services/services_test.go @@ -0,0 +1,52 @@ +package services + +import ( + "context" + "net/http" + "net/http/httptest" + "os" + "strings" + "testing" + + "goweb/config" + "goweb/ent" + + "github.com/labstack/echo/v4" +) + +var ( + c *Container + ctx echo.Context + usr *ent.User + rec *httptest.ResponseRecorder +) + +func TestMain(m *testing.M) { + // Set the environment to test + config.SwitchEnvironment(config.EnvTest) + + // Create a new container + c = NewContainer() + + // Create a web context + req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("")) + rec = httptest.NewRecorder() + ctx = c.Web.NewContext(req, rec) + + // Create a test user + var err error + usr, err = c.ORM.User. + Create(). + SetEmail("test@test.dev"). + SetPassword("abc"). + SetName("Test User"). + Save(context.Background()) + + if err != nil { + panic(err) + } + + // Run tests + exitVal := m.Run() + os.Exit(exitVal) +}