Add test for generateSuiteTopLevel()

in testing/codegen.

Author: tchssk

testing/codegen/suite_test.go

@@ -0,0 +1,43 @@
+package codegen
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	httpcodegen "goa.design/goa/v3/http/codegen"
+	"goa.design/plugins/v3/testing/codegen/testdata"
+)
+
+func TestGenerateSuiteTopLevel(t *testing.T) {
+	cases := map[string]struct {
+		DSL  func()
+		Code map[string][]string
+		Path string
+	}{
+		"with-result": {
+			DSL: testdata.WithResultDSL,
+			Code: map[string][]string{
+				"suite-test": {testdata.SuiteTestWithResultCode},
+			},
+			Path: "with_result_service_suite_test.go",
+		},
+		"without-result": {
+			DSL: testdata.WithoutResultDSL,
+			Code: map[string][]string{
+				"suite-test": {testdata.SuiteTestWithoutResultCode},
+			},
+			Path: "without_result_service_suite_test.go",
+		},
+	}
+	for name, c := range cases {
+		t.Run(name, func(t *testing.T) {
+			root := httpcodegen.RunHTTPDSL(t, c.DSL)
+			svc := root.Services[0]
+			f := generateSuiteTopLevel("", "", root, svc)
+			assert.Equal(t, c.Path, f.Path)
+			for sec, secCode := range c.Code {
+				testCode(t, f, sec, secCode)
+			}
+		})
+	}
+}

testing/codegen/testdata/code.go

@@ -912,3 +912,46 @@ func (r *ScenarioRunner) selectTransport(client *Client, transport string) *Clie
 	}
 }
 `
+
+var SuiteTestWithResultCode = `// RunWithResultServiceHarness exercises the generated harness against your
+// service implementation.// Call this helper from your test, passing your service implementation.
+func RunWithResultServiceHarness(t *testing.T, svc withresultservice.Service) {
+	t.Helper()
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	h := withResultServicetest.NewHarness(t, svc)
+	defer h.Close()
+
+	td := withResultServicetest.NewTestData()
+	t.Run("WithResultMethod", func(t *testing.T) {
+		result, err := h.Client.WithResultMethod(ctx)
+		if err != nil {
+			t.Errorf("WithResultMethod failed: %v", err)
+		}
+		if result == nil {
+			t.Error("WithResultMethod returned nil result")
+		}
+	})
+}
+`
+
+var SuiteTestWithoutResultCode = `// RunWithoutResultServiceHarness exercises the generated harness against your
+// service implementation.// Call this helper from your test, passing your service implementation.
+func RunWithoutResultServiceHarness(t *testing.T, svc withoutresultservice.Service) {
+	t.Helper()
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	h := withoutResultServicetest.NewHarness(t, svc)
+	defer h.Close()
+
+	td := withoutResultServicetest.NewTestData()
+	t.Run("WithoutResultMethod", func(t *testing.T) {
+		result, err := h.Client.WithoutResultMethod(ctx)
+		if err != nil {
+			t.Errorf("WithoutResultMethod failed: %v", err)
+		}
+	})
+}
+`