mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
50d8151995
* ci: Run tests using PostgreSQL database and mock This allows us to use the mock database for quick iterative testing, and have confidence from CI using a real PostgreSQL database. PostgreSQL tests are only ran on Linux. They are *really* slow on MacOS and Windows runners, and don't provide much additional confidence. * Only run PostgreSQL tests once for speed * Fix race condition of log after close Not all resources were cleaned up immediately after a peer connection was closed. DataChannels could have a goroutine exit after Close() prior to this. * Fix comment
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package coderdtest
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cdr.dev/slog/sloggers/slogtest"
|
|
"github.com/coder/coder/coderd"
|
|
"github.com/coder/coder/codersdk"
|
|
"github.com/coder/coder/database"
|
|
"github.com/coder/coder/database/databasefake"
|
|
"github.com/coder/coder/database/postgres"
|
|
)
|
|
|
|
// Server represents a test instance of coderd.
|
|
// The database is intentionally omitted from
|
|
// this struct to promote data being exposed via
|
|
// the API.
|
|
type Server struct {
|
|
Client *codersdk.Client
|
|
URL *url.URL
|
|
}
|
|
|
|
// New constructs a new coderd test instance.
|
|
func New(t *testing.T) Server {
|
|
// This can be hotswapped for a live database instance.
|
|
db := databasefake.New()
|
|
if os.Getenv("DB") != "" {
|
|
connectionURL, close, err := postgres.Open()
|
|
require.NoError(t, err)
|
|
t.Cleanup(close)
|
|
sqlDB, err := sql.Open("postgres", connectionURL)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
_ = sqlDB.Close()
|
|
})
|
|
err = database.Migrate(sqlDB)
|
|
require.NoError(t, err)
|
|
db = database.New(sqlDB)
|
|
}
|
|
|
|
handler := coderd.New(&coderd.Options{
|
|
Logger: slogtest.Make(t, nil),
|
|
Database: db,
|
|
})
|
|
srv := httptest.NewServer(handler)
|
|
serverURL, err := url.Parse(srv.URL)
|
|
require.NoError(t, err)
|
|
t.Cleanup(srv.Close)
|
|
|
|
client := codersdk.New(serverURL)
|
|
_, err = client.CreateInitialUser(context.Background(), coderd.CreateUserRequest{
|
|
Email: "testuser@coder.com",
|
|
Username: "testuser",
|
|
Password: "testpassword",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
login, err := client.LoginWithPassword(context.Background(), coderd.LoginWithPasswordRequest{
|
|
Email: "testuser@coder.com",
|
|
Password: "testpassword",
|
|
})
|
|
require.NoError(t, err)
|
|
err = client.SetSessionToken(login.SessionToken)
|
|
require.NoError(t, err)
|
|
|
|
return Server{
|
|
Client: client,
|
|
URL: serverURL,
|
|
}
|
|
}
|