From dc4cedd2989ff3f964c01df889286668ff49af64 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20Jamr=C3=B3z?= <pm.jamroz@gmail.com>
Date: Fri, 14 Jul 2023 15:15:50 +0200
Subject: [PATCH] TestDatasource: Generate refId for the flame graph (#71547)

* Remove hardcoded refId

* Generate correct refId for flameGraph
---
 public/app/plugins/datasource/testdata/datasource.ts        | 6 +++---
 .../datasource/testdata/testData/flameGraphResponse.ts      | 1 -
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/public/app/plugins/datasource/testdata/datasource.ts b/public/app/plugins/datasource/testdata/datasource.ts
index 739e7bdade6..f364479d35b 100644
--- a/public/app/plugins/datasource/testdata/datasource.ts
+++ b/public/app/plugins/datasource/testdata/datasource.ts
@@ -95,7 +95,7 @@ export class TestDataDataSource extends DataSourceWithBackend<TestData> {
           streams.push(this.nodesQuery(target, options));
           break;
         case 'flame_graph':
-          streams.push(this.flameGraphQuery());
+          streams.push(this.flameGraphQuery(target));
           break;
         case 'trace':
           streams.push(this.trace(target, options));
@@ -242,8 +242,8 @@ export class TestDataDataSource extends DataSourceWithBackend<TestData> {
     return of({ data: frames }).pipe(delay(100));
   }
 
-  flameGraphQuery(): Observable<DataQueryResponse> {
-    return of({ data: [flameGraphData] }).pipe(delay(100));
+  flameGraphQuery(target: TestData): Observable<DataQueryResponse> {
+    return of({ data: [{ ...flameGraphData, refId: target.refId }] }).pipe(delay(100));
   }
 
   trace(target: TestData, options: DataQueryRequest<TestData>): Observable<DataQueryResponse> {
diff --git a/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts b/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
index 1e4045dc636..370cf3ea5ff 100644
--- a/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
+++ b/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
@@ -2,7 +2,6 @@ import { DataFrameDTO } from '@grafana/data';
 
 export const flameGraphData: DataFrameDTO = {
   name: 'response',
-  refId: 'A',
   // @ts-ignore
   meta: { preferredVisualisationType: 'flamegraph' },
   fields: [