Fix tests

grafana · Oct 7, 2023 · 56fc36a · 56fc36a
1 parent 9e3d7bf
commit 56fc36a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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)