From 5a9847b03d05847c7c96668a8f180fc96521f542 Mon Sep 17 00:00:00 2001
From: Pieter Noordhuis <pieter.noordhuis@databricks.com>
Date: Wed, 4 Sep 2024 16:45:47 +0200
Subject: [PATCH] Update test

---
 libs/dyn/visit_test.go | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/libs/dyn/visit_test.go b/libs/dyn/visit_test.go
index 86b8875fdb..5b61399be0 100644
--- a/libs/dyn/visit_test.go
+++ b/libs/dyn/visit_test.go
@@ -1,26 +1,27 @@
-package dyn
+package dyn_test
 
 import (
 	"testing"
 
-	"github.com/stretchr/testify/assert"
+	"github.com/databricks/cli/libs/dyn"
+	assert "github.com/databricks/cli/libs/dyn/dynassert"
 )
 
 func TestVisitCallbackPathCopy(t *testing.T) {
-	vin := V(map[string]Value{
-		"foo": V(42),
-		"bar": V(43),
+	vin := dyn.V(map[string]dyn.Value{
+		"foo": dyn.V(42),
+		"bar": dyn.V(43),
 	})
 
-	var paths []Path
+	var paths []dyn.Path
 
 	// The callback should receive a copy of the path.
 	// If the same underlying value is used, all collected paths will be the same.
-	_, _ = visit(vin, EmptyPath, NewPattern(AnyKey()), visitOptions{
-		fn: func(p Path, v Value) (Value, error) {
-			paths = append(paths, p)
-			return v, nil
-		},
+	// This test uses `MapByPattern` to collect all paths in the map.
+	// Visit itself doesn't have public functions and we exclusively use black-box testing for this package.
+	_, _ = dyn.MapByPattern(vin, dyn.NewPattern(dyn.AnyKey()), func(p dyn.Path, v dyn.Value) (dyn.Value, error) {
+		paths = append(paths, p)
+		return v, nil
 	})
 
 	// Verify that the paths retained their original values.