Skip to content

Commit

Permalink
fix integration tests
Browse files Browse the repository at this point in the history
  • Loading branch information
matt2e committed Oct 10, 2024
1 parent 50c033e commit d50685e
Showing 1 changed file with 17 additions and 6 deletions.
23 changes: 17 additions & 6 deletions backend/controller/admin/local_client_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ package admin

import (
"context"
"net/url"
"testing"

"github.com/alecthomas/assert/v2"
Expand All @@ -16,18 +17,30 @@ import (
"github.com/TBD54566975/ftl/internal/configuration/routers"
in "github.com/TBD54566975/ftl/internal/integration"
"github.com/TBD54566975/ftl/internal/log"
"github.com/TBD54566975/ftl/internal/schema"
)

func getDiskSchema(t testing.TB, ctx context.Context) *schema.Schema {
t.Helper()

bindURL, err := url.Parse("http://127.0.0.1:8893")
assert.NoError(t, err)
bindAllocator, err := bind.NewBindAllocator(bindURL)
assert.NoError(t, err)
dsr := &diskSchemaRetriever{}
sch, err := dsr.GetActiveSchema(ctx, optional.Some(bindAllocator))
assert.Error(t, err)
return sch
}

func TestDiskSchemaRetrieverWithBuildArtefact(t *testing.T) {
in.Run(t,
in.WithFTLConfig("ftl-project-dr.toml"),
in.WithoutController(),
in.CopyModule("dischema"),
in.Build("dischema"),
func(t testing.TB, ic in.TestContext) {
dsr := &diskSchemaRetriever{deployRoot: optional.Some[string](ic.WorkingDir())}
sch, err := dsr.GetActiveSchema(ic.Context, optional.None[*bind.BindAllocator]())
assert.NoError(t, err)
sch := getDiskSchema(t, ic.Context)

module, ok := sch.Module("dischema").Get()
assert.Equal(t, ok, true)
Expand All @@ -42,9 +55,7 @@ func TestDiskSchemaRetrieverWithNoSchema(t *testing.T) {
in.WithoutController(),
in.CopyModule("dischema"),
func(t testing.TB, ic in.TestContext) {
dsr := &diskSchemaRetriever{}
_, err := dsr.GetActiveSchema(ic.Context, optional.None[*bind.BindAllocator]())
assert.Error(t, err)
_ = getDiskSchema(t, ic.Context)
},
)
}
Expand Down

0 comments on commit d50685e

Please sign in to comment.