Fix index pattern data source reference not updated in sample data (#…

…6851)

* Fix index pattern data source reference not updated in sample data

Signed-off-by: Lin Wang <[email protected]>

* Changeset file for PR #6851 created/updated

---------

Signed-off-by: Lin Wang <[email protected]>
Co-authored-by: opensearch-changeset-bot[bot] <154024398+opensearch-changeset-bot[bot]@users.noreply.github.com>

changelogs/fragments/6851.yml

@@ -0,0 +1,2 @@
+fix:
+- Update index pattern references with data source when import sample data ([#6851](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/6851))

src/plugins/home/server/services/sample_data/data_sets/util.test.ts

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { getSavedObjectsWithDataSource, getFinalSavedObjects, appendDataSourceId } from './util';
+import { getSavedObjectsWithDataSource, getFinalSavedObjects } from './util';
 import { SavedObject, updateDataSourceNameInVegaSpec } from '../../../../../../core/server';
 import visualizationObjects from './test_utils/visualization_objects.json';
 
@@ -62,6 +62,39 @@ describe('getSavedObjectsWithDataSource()', () => {
 
     expect(updatedVegaVisualizationsFields).toEqual(expect.arrayContaining(expectedUpdatedFields));
   });
+
+  it('should update index-pattern id and references with given data source', () => {
+    const dataSourceId = 'some-datasource-id';
+    const dataSourceName = 'Data Source Name';
+
+    expect(
+      getSavedObjectsWithDataSource(
+        [
+          {
+            id: 'saved-object-1',
+            type: 'index-pattern',
+            attributes: {},
+            references: [],
+          },
+        ],
+        dataSourceId,
+        dataSourceName
+      )
+    ).toEqual([
+      {
+        id: 'some-datasource-id_saved-object-1',
+        type: 'index-pattern',
+        attributes: {},
+        references: [
+          {
+            id: `${dataSourceId}`,
+            type: 'data-source',
+            name: 'dataSource',
+          },
+        ],
+      },
+    ]);
+  });
 });
 
 describe('getFinalSavedObjects()', () => {

src/plugins/home/server/services/sample_data/data_sets/util.ts

@@ -71,6 +71,17 @@ export const getSavedObjectsWithDataSource = (
     return saveObjectList.map((saveObject) => {
       overrideSavedObjectId(saveObject, idGenerator);
 
+      // update reference
+      if (saveObject.type === 'index-pattern') {
+        saveObject.references = [
+          {
+            id: `${dataSourceId}`,
+            type: 'data-source',
+            name: 'dataSource',
+          },
+        ];
+      }
+
       if (dataSourceTitle) {
         if (
           saveObject.type === 'dashboard' ||