From b5bc18e2607170e4884082b24e4cb539d31e7771 Mon Sep 17 00:00:00 2001 From: Matthieu MOREL Date: Mon, 28 Aug 2023 18:29:54 +0200 Subject: [PATCH] Fix module generator for examples Closes #1544 --- modulegen/_template/example_test.go.tmpl | 4 ++-- modulegen/main_test.go | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/modulegen/_template/example_test.go.tmpl b/modulegen/_template/example_test.go.tmpl index db824245d3..d4ade7fa56 100644 --- a/modulegen/_template/example_test.go.tmpl +++ b/modulegen/_template/example_test.go.tmpl @@ -1,4 +1,4 @@ -{{ $lower := ToLower }}{{ $title := Title }}package {{ $lower }} +{{ $entrypoint := Entrypoint }}{{ $lower := ToLower }}{{ $title := Title }}package {{ $lower }} import ( "context" @@ -8,7 +8,7 @@ import ( func Test{{ $title }}(t *testing.T) { ctx := context.Background() - container, err := RunContainer(ctx) + container, err := {{ $entrypoint }}(ctx) if err != nil { t.Fatal(err) } diff --git a/modulegen/main_test.go b/modulegen/main_test.go index 123dab85de..e8ced0fdf1 100644 --- a/modulegen/main_test.go +++ b/modulegen/main_test.go @@ -446,7 +446,7 @@ func assertExampleTestContent(t *testing.T, example Example, exampleTestFile str data := sanitiseContent(content) assert.Equal(t, data[0], "package "+example.Lower()) assert.Equal(t, data[7], "func Test"+example.Title()+"(t *testing.T) {") - assert.Equal(t, data[10], "\tcontainer, err := RunContainer(ctx)") + assert.Equal(t, data[10], "\tcontainer, err := "+example.Entrypoint()+"(ctx)") } // assert content example