Move fn tests off deprecated GRPC APIs

namespacelabs · Jul 26, 2024 · a586383 · a586383
1 parent 471b43b
commit a586383
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 9 deletions.
diff --git a/framework/testing/do.go b/framework/testing/do.go
@@ -27,9 +27,8 @@ type Test struct {
 	testboot.TestData
 }
 
-func (t Test) Connect(ctx context.Context, endpoint *schema.Endpoint) (*grpc.ClientConn, error) {
-	return grpc.DialContext(ctx, endpoint.Address(),
-		grpc.WithBlock(),
+func (t Test) NewClient(endpoint *schema.Endpoint) (*grpc.ClientConn, error) {
+	return grpc.NewClient(endpoint.Address(),
 		grpc.WithTransportCredentials(insecure.NewCredentials())) ///  XXX mTLS etc.
 }
 

diff --git a/internal/frontend/scaffold/gengotest.go b/internal/frontend/scaffold/gengotest.go
@@ -52,7 +52,7 @@ import (
 
 func main() {
 	testing.Do(func(ctx context.Context, t testing.Test) error {
-		conn, err := t.Connect(ctx, t.MustEndpoint("{{.ServicePkg}}", "{{.ServiceImportAlias}}"))
+		conn, err := t.NewClient(t.MustEndpoint("{{.ServicePkg}}", "{{.ServiceImportAlias}}"))
 		if err != nil {
 			return err
 		}

diff --git a/internal/testdata/tests/localstacks3/tests3demo.go b/internal/testdata/tests/localstacks3/tests3demo.go
@@ -16,7 +16,7 @@ func main() {
 	testing.Do(func(ctx context.Context, t testing.Test) error {
 		endpoint := t.MustEndpoint("namespacelabs.dev/foundation/internal/testdata/service/localstacks3", "localstacks3")
 
-		conn, err := t.Connect(ctx, endpoint)
+		conn, err := t.NewClient(endpoint)
 		if err != nil {
 			return err
 		}

diff --git a/internal/testdata/tests/modeling/scopes/testscoping.go b/internal/testdata/tests/modeling/scopes/testscoping.go
@@ -16,7 +16,7 @@ func main() {
 	testing.Do(func(ctx context.Context, t testing.Test) error {
 		endpoint := t.MustEndpoint("namespacelabs.dev/foundation/internal/testdata/service/count", "count")
 
-		conn, err := t.Connect(ctx, endpoint)
+		conn, err := t.NewClient(endpoint)
 		if err != nil {
 			return err
 		}

diff --git a/internal/testdata/tests/postgres/testlist.go b/internal/testdata/tests/postgres/testlist.go
@@ -18,7 +18,7 @@ func main() {
 	testing.Do(func(ctx context.Context, t testing.Test) error {
 		endpoint := t.MustEndpoint("namespacelabs.dev/foundation/internal/testdata/service/list", "list")
 
-		conn, err := t.Connect(ctx, endpoint)
+		conn, err := t.NewClient(endpoint)
 		if err != nil {
 			return err
 		}

diff --git a/internal/testdata/tests/prometheus/canscrape/testscrape.go b/internal/testdata/tests/prometheus/canscrape/testscrape.go
@@ -16,9 +16,9 @@ import (
 	dto "github.com/prometheus/client_model/go"
 	"github.com/prometheus/common/expfmt"
 	"namespacelabs.dev/foundation/framework/testing"
+	"namespacelabs.dev/foundation/internal/testdata/service/proto"
 	"namespacelabs.dev/foundation/schema"
 	"namespacelabs.dev/foundation/schema/schemahelper"
-	"namespacelabs.dev/foundation/internal/testdata/service/proto"
 )
 
 func main() {
@@ -36,7 +36,7 @@ func main() {
 			return errors.New("prometheus metrics endpoint missing")
 		}
 
-		conn, err := t.Connect(ctx, endpoint)
+		conn, err := t.NewClient(endpoint)
 		if err != nil {
 			return err
 		}