From 9d7eaedb2b2d3d4c94226161d3c7131183c7e8b1 Mon Sep 17 00:00:00 2001
From: "Grot (@grafanabot)" <43478413+grafanabot@users.noreply.github.com>
Date: Wed, 21 Oct 2020 18:16:06 +0200
Subject: [PATCH] TestData: multiple arrow requests should return multiple
 frames (#28417) (#28441)

(cherry picked from commit f32d47a53515ad7ce5b30fc37c3d7e2e63e49f29)

Co-authored-by: Ryan McKinley <ryantxu@gmail.com>
---
 public/app/plugins/datasource/testdata/datasource.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/public/app/plugins/datasource/testdata/datasource.ts b/public/app/plugins/datasource/testdata/datasource.ts
index 7f21d261a36..5d5a39063de 100644
--- a/public/app/plugins/datasource/testdata/datasource.ts
+++ b/public/app/plugins/datasource/testdata/datasource.ts
@@ -192,7 +192,9 @@ function runArrowFile(target: TestDataQuery, req: DataQueryRequest<TestDataQuery
   if (target.stringInput && target.stringInput.length > 10) {
     try {
       const table = base64StringToArrowTable(target.stringInput);
-      data = [arrowTableToDataFrame(table)];
+      const frame = arrowTableToDataFrame(table);
+      frame.refId = target.refId;
+      data = [frame];
     } catch (e) {
       console.warn('Error reading saved arrow', e);
       const error = toDataQueryError(e);
@@ -200,7 +202,7 @@ function runArrowFile(target: TestDataQuery, req: DataQueryRequest<TestDataQuery
       return of({ state: LoadingState.Error, error, data });
     }
   }
-  return of({ state: LoadingState.Done, data });
+  return of({ state: LoadingState.Done, data, key: req.requestId + target.refId });
 }
 
 function runGrafanaAPI(target: TestDataQuery, req: DataQueryRequest<TestDataQuery>): Observable<DataQueryResponse> {