fix unit tests

DevCycleHQ · Apr 23, 2024 · c65f967 · c65f967
1 parent 28f1127
commit c65f967
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/client_test.go b/client_test.go
@@ -554,7 +554,7 @@ func BenchmarkClient_VariableParallel(b *testing.B) {
 			}
 			if benchmarkEnableConfigUpdates && configCounter.Add(1)%10000 == 0 {
 				go func() {
-					err = client.configManager.setConfig([]byte(test_large_config), "")
+					err = client.configManager.setConfig([]byte(test_large_config), "", "")
 					setConfigCount.Add(1)
 				}()
 			}

diff --git a/configmanager_test.go b/configmanager_test.go
@@ -11,11 +11,13 @@ import (
 type recordingConfigReceiver struct {
 	configureCount int
 	etag           string
+	rayId          string
 }
 
-func (r *recordingConfigReceiver) StoreConfig(_ []byte, etag string) error {
+func (r *recordingConfigReceiver) StoreConfig(_ []byte, etag string, rayId string) error {
 	r.configureCount++
 	r.etag = etag
+	r.rayId = rayId
 	return nil
 }
 
@@ -27,6 +29,10 @@ func (r *recordingConfigReceiver) GetETag() string {
 	return r.etag
 }
 
+func (r *recordingConfigReceiver) GetRayId() string {
+	return r.rayId
+}
+
 func (r *recordingConfigReceiver) GetRawConfig() []byte {
 	return nil
 }