diff --git a/openfeature/testing/testprovider.go b/openfeature/testing/testprovider.go index 584b9932..551c411e 100644 --- a/openfeature/testing/testprovider.go +++ b/openfeature/testing/testprovider.go @@ -5,7 +5,6 @@ import ( "fmt" "runtime" "sync" - "testing" "github.com/open-feature/go-sdk/openfeature" "github.com/open-feature/go-sdk/openfeature/memprovider" @@ -29,7 +28,9 @@ type TestProvider struct { } // UsingFlags sets flags for the scope of a test -func (tp TestProvider) UsingFlags(test testing.TB, flags map[string]memprovider.InMemoryFlag) { +type TestFramework = interface{ Name() string } + +func (tp TestProvider) UsingFlags(test TestFramework, flags map[string]memprovider.InMemoryFlag) { storeGoroutineLocal(test.Name()) tp.providers.Store(test.Name(), memprovider.NewInMemoryProvider(flags)) }