diff --git a/superset-frontend/src/SqlLab/reducers/getInitialState.js b/superset-frontend/src/SqlLab/reducers/getInitialState.js index 14efdeef91..9f7a0a1087 100644 --- a/superset-frontend/src/SqlLab/reducers/getInitialState.js +++ b/superset-frontend/src/SqlLab/reducers/getInitialState.js @@ -27,6 +27,7 @@ export default function getInitialState({ databases, queries: queries_, requested_query: requestedQuery, + user, }) { /** * Before YYYY-MM-DD, the state for SQL Lab was stored exclusively in the @@ -180,6 +181,7 @@ export default function getInitialState({ tabHistory, tables, queriesLastUpdate: Date.now(), + user, }, requestedQuery, messageToasts: getToastsFromPyFlashMessages( diff --git a/superset-frontend/src/SqlLab/reducers/getInitialState.test.ts b/superset-frontend/src/SqlLab/reducers/getInitialState.test.ts new file mode 100644 index 0000000000..e08443b059 --- /dev/null +++ b/superset-frontend/src/SqlLab/reducers/getInitialState.test.ts @@ -0,0 +1,43 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +import getInitialState from './getInitialState'; + +const apiData = { + defaultDbId: 1, + common: { + conf: { + DEFAULT_SQLLAB_LIMIT: 1, + }, + }, + active_tab: null, + tab_state_ids: [], + databases: [], + queries: [], + requested_query: null, + user: { + userId: 1, + username: 'some name', + }, +}; +describe('getInitialState', () => { + it('should output the user that is passed in', () => { + expect(getInitialState(apiData).sqlLab.user.userId).toEqual(1); + }); +});