fixup! cloud/expv2: Retry and concurrent flush

grafana · Jun 14, 2023 · f19e846 · f19e846
1 parent 71c4883
commit f19e846
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/output/cloud/output.go b/output/cloud/output.go
@@ -336,7 +336,7 @@ func (out *Output) startVersionedOutput() error {
 	case int64(apiVersion1):
 		out.versionedOutput, err = cloudv1.New(out.logger, out.config, out.client)
 	case int64(apiVersion2):
-		out.versionedOutput, err = cloudv2.New(out.logger, out.config)
+		out.versionedOutput, err = cloudv2.New(out.logger, out.config, out.client)
 	default:
 		err = fmt.Errorf("v%d is an unexpected version", out.config.APIVersion.Int64)
 	}

diff --git a/output/cloud/output_test.go b/output/cloud/output_test.go
@@ -176,6 +176,9 @@ func TestOutputStartVersionedOutputV2(t *testing.T) {
 		},
 	}
 
+	o.client = cloudapi.NewClient(
+		nil, o.config.Token.String, o.config.Host.String, "v/tests", o.config.Timeout.TimeDuration())
+
 	err := o.startVersionedOutput()
 	require.NoError(t, err)