From bda840e6bd6a01b1e56fbc07fa596415ee0249b7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?LeoYuan=20=E8=A2=81=E5=8A=9B=E7=9A=93?=
 <yuanlihao1988@gmail.com>
Date: Tue, 20 Dec 2022 11:07:32 +0800
Subject: [PATCH] chore: fix CodeGen UT failure

---
 .../demo-project/src/pages/$/index.jsx        | 46 ++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/modules/code-generator/tests/fixtures/test-cases/react-app/demo9-datasource-engine/expected/demo-project/src/pages/$/index.jsx b/modules/code-generator/tests/fixtures/test-cases/react-app/demo9-datasource-engine/expected/demo-project/src/pages/$/index.jsx
index ac1591add..9226280c7 100644
--- a/modules/code-generator/tests/fixtures/test-cases/react-app/demo9-datasource-engine/expected/demo-project/src/pages/$/index.jsx
+++ b/modules/code-generator/tests/fixtures/test-cases/react-app/demo9-datasource-engine/expected/demo-project/src/pages/$/index.jsx
@@ -4,6 +4,10 @@ import React from 'react';
 
 import { Switch } from '@alifd/next';
 
+import { createJsonpHandler as __$$createJsonpRequestHandler } from '@alilc/lowcode-datasource-jsonp-handler';
+
+import { create as __$$createDataSourceEngine } from '@alilc/lowcode-datasource-engine/runtime';
+
 import utils from '../../utils';
 
 import * as __$$i18n from '../../i18n';
@@ -15,6 +19,20 @@ import './index.css';
 class $$Page extends React.Component {
   _context = this;
 
+  _dataSourceConfig = this._defineDataSourceConfig();
+  _dataSourceEngine = __$$createDataSourceEngine(this._dataSourceConfig, this, {
+    runtimeConfig: true,
+    requestHandlersMap: { jsonp: __$$createJsonpRequestHandler() },
+  });
+
+  get dataSourceMap() {
+    return this._dataSourceEngine.dataSourceMap || {};
+  }
+
+  reloadDataSource = async () => {
+    await this._dataSourceEngine.reloadDataSource();
+  };
+
   get constants() {
     return __$$constants || {};
   }
@@ -33,7 +51,33 @@ class $$Page extends React.Component {
 
   $$ = () => [];
 
-  componentDidMount() {}
+  _defineDataSourceConfig() {
+    const _this = this;
+    return {
+      list: [
+        {
+          id: 'todos',
+          isInit: function () {
+            return true;
+          },
+          type: 'jsonp',
+          options: function () {
+            return {
+              method: 'GET',
+              uri: 'https://a0ee9135-6a7f-4c0f-a215-f0f247ad907d.mock.pstmn.io',
+            };
+          },
+          dataHandler: function dataHandler(data) {
+            return data.data;
+          },
+        },
+      ],
+    };
+  }
+
+  componentDidMount() {
+    this._dataSourceEngine.reloadDataSource();
+  }
 
   render() {
     const __$$context = this._context || this;