From e1258edbce39e3f2306b6fe1ac7cac311da01eba Mon Sep 17 00:00:00 2001
From: Marta Bondyra <marta.bondyra@elastic.co>
Date: Wed, 2 Dec 2020 19:53:31 +0100
Subject: [PATCH] tests

---
 .../xy_visualization/visualization.test.ts    | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/x-pack/plugins/lens/public/xy_visualization/visualization.test.ts b/x-pack/plugins/lens/public/xy_visualization/visualization.test.ts
index 546cf06d4014e..1eda269a66de3 100644
--- a/x-pack/plugins/lens/public/xy_visualization/visualization.test.ts
+++ b/x-pack/plugins/lens/public/xy_visualization/visualization.test.ts
@@ -558,6 +558,27 @@ describe('xy_visualization', () => {
         const accessorConfig = breakdownConfig!.accessors[0];
         expect(typeof accessorConfig !== 'string' && accessorConfig.palette).toEqual(customColors);
       });
+
+      it('should respect the order of accessors coming from datasource', () => {
+        const colorAssignment = require('./color_assignment');
+        const getAccessorColorConfigSpy = jest.spyOn(colorAssignment, 'getAccessorColorConfig');
+        mockDatasource.publicAPIMock.getTableSpec.mockReturnValue([
+          { columnId: 'c' },
+          { columnId: 'b' },
+        ]);
+        callConfigForYConfigs({});
+        expect(getAccessorColorConfigSpy).toHaveBeenCalledWith(
+          expect.anything(),
+          expect.anything(),
+          {
+            accessors: ['c', 'b'],
+            layerId: 'first',
+            seriesType: 'area',
+            xAccessor: 'a',
+          },
+          expect.anything()
+        );
+      });
     });
   });