From 56fc36a1dd964da14ac14216be8ed622174e3779 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?K=C3=A9vin=20Gomez?= Date: Fri, 6 Oct 2023 00:28:00 +0200 Subject: [PATCH] Fix tests --- internal/openapi/generator_test.go | 4 ++-- internal/veneers/rewrite_test.go | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/internal/openapi/generator_test.go b/internal/openapi/generator_test.go index 904965708..b5c8a1821 100644 --- a/internal/openapi/generator_test.go +++ b/internal/openapi/generator_test.go @@ -146,7 +146,7 @@ func TestArrays(t *testing.T) { require.NoError(t, err) require.Len(t, f.Objects, 2) - def := f.LocateDefinition("Arrays") + def := f.LocateObject("Arrays") assert.Equal(t, def.Type.Kind, ast.KindStruct) structType := def.Type.AsStruct() @@ -184,7 +184,7 @@ func TestRefs(t *testing.T) { require.NoError(t, err) require.Len(t, f.Objects, 2) - def := f.LocateDefinition("Refs") + def := f.LocateObject("Refs") assert.Equal(t, def.Type.Kind, ast.KindStruct) structType := def.Type.AsStruct() diff --git a/internal/veneers/rewrite_test.go b/internal/veneers/rewrite_test.go index 7276755c4..d0aa94072 100644 --- a/internal/veneers/rewrite_test.go +++ b/internal/veneers/rewrite_test.go @@ -207,7 +207,8 @@ func TestRewriter_ApplyTo(t *testing.T) { expectedBuildersJSON := mustMarshalJSON(t, tc.outputBuilders) // apply the rewrite rules - rewrittenBuilders := rewriter.ApplyTo(tc.inputBuilders) + rewrittenBuilders, err := rewriter.ApplyTo(tc.inputBuilders) + req.NoError(err) // save the output states originalBuildersJSONAfterApply := mustMarshalJSON(t, tc.inputBuilders)