Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Commit

Permalink
fix: Preserve unknown URL params (apache#21785)
Browse files Browse the repository at this point in the history
  • Loading branch information
michael-s-molina authored Oct 12, 2022
1 parent 75e6a04 commit 11d7d6e
Show file tree
Hide file tree
Showing 5 changed files with 31 additions and 15 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,8 @@ const reduxState = {
},
};

const key = 'aWrs7w29sd';
const KEY = 'aWrs7w29sd';
const SEARCH = `?form_data_key=${KEY}&dataset_id=1`;

jest.mock('react-resize-detector', () => ({
__esModule: true,
Expand All @@ -87,21 +88,20 @@ jest.mock('lodash/debounce', () => ({
default: (fuc: Function) => fuc,
}));

fetchMock.post('glob:*/api/v1/explore/form_data*', { key });
fetchMock.put('glob:*/api/v1/explore/form_data*', { key });
fetchMock.post('glob:*/api/v1/explore/form_data*', { key: KEY });
fetchMock.put('glob:*/api/v1/explore/form_data*', { key: KEY });
fetchMock.get('glob:*/api/v1/explore/form_data*', {});
fetchMock.get('glob:*/favstar/slice*', { count: 0 });

const defaultPath = '/explore/';
const renderWithRouter = ({
withKey,
search = '',
overridePathname,
}: {
withKey?: boolean;
search?: string;
overridePathname?: string;
} = {}) => {
const path = overridePathname ?? defaultPath;
const search = withKey ? `?form_data_key=${key}&dataset_id=1` : '';
Object.defineProperty(window, 'location', {
get() {
return { pathname: path, search };
Expand Down Expand Up @@ -143,12 +143,12 @@ test('generates a new form_data param when none is available', async () => {

test('generates a different form_data param when one is provided and is mounting', async () => {
const replaceState = jest.spyOn(window.history, 'replaceState');
await waitFor(() => renderWithRouter({ withKey: true }));
await waitFor(() => renderWithRouter({ search: SEARCH }));
expect(replaceState).not.toHaveBeenLastCalledWith(
0,
expect.anything(),
undefined,
expect.stringMatching(key),
expect.stringMatching(KEY),
);
expect(replaceState).toHaveBeenCalledWith(
expect.anything(),
Expand All @@ -164,7 +164,7 @@ test('reuses the same form_data param when updating', async () => {
});
const replaceState = jest.spyOn(window.history, 'replaceState');
const pushState = jest.spyOn(window.history, 'pushState');
await waitFor(() => renderWithRouter({ withKey: true }));
await waitFor(() => renderWithRouter({ search: SEARCH }));
expect(replaceState.mock.calls.length).toBe(1);
userEvent.click(screen.getByText('Update chart'));
await waitFor(() => expect(pushState.mock.calls.length).toBe(1));
Expand All @@ -188,3 +188,17 @@ test('doesnt call replaceState when pathname is not /explore', async () => {
expect(replaceState).not.toHaveBeenCalled();
replaceState.mockRestore();
});

test('preserves unknown parameters', async () => {
const replaceState = jest.spyOn(window.history, 'replaceState');
const unknownParam = 'test=123';
await waitFor(() =>
renderWithRouter({ search: `${SEARCH}&${unknownParam}` }),
);
expect(replaceState).toHaveBeenCalledWith(
expect.anything(),
undefined,
expect.stringMatching(unknownParam),
);
replaceState.mockRestore();
});
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,9 @@ const updateHistory = debounce(
) => {
const payload = { ...formData };
const chartId = formData.slice_id;
const additionalParam = {};
const params = new URLSearchParams(window.location.search);
const additionalParam = Object.fromEntries(params);

if (chartId) {
additionalParam[URL_PARAMS.sliceId.name] = chartId;
} else {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ test('get form_data_key and slice_id from search params - url when moving from d
`${EXPLORE_BASE_URL}?form_data_key=yrLXmyE9fmhQ11lM1KgaD1PoPSBpuLZIJfqdyIdw9GoBwhPFRZHeIgeFiNZljbpd&slice_id=56`,
);
expect(getParsedExploreURLParams().toString()).toEqual(
'slice_id=56&form_data_key=yrLXmyE9fmhQ11lM1KgaD1PoPSBpuLZIJfqdyIdw9GoBwhPFRZHeIgeFiNZljbpd',
'form_data_key=yrLXmyE9fmhQ11lM1KgaD1PoPSBpuLZIJfqdyIdw9GoBwhPFRZHeIgeFiNZljbpd&slice_id=56',
);
});

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ const EXPLORE_URL_PATH_PARAMS = {
// we need to "flatten" the search params to use them with /v1/explore endpoint
const getParsedExploreURLSearchParams = (search: string) => {
const urlSearchParams = new URLSearchParams(search);
return Object.keys(EXPLORE_URL_SEARCH_PARAMS).reduce((acc, currentParam) => {
return Array.from(urlSearchParams.keys()).reduce((acc, currentParam) => {
const paramValue = urlSearchParams.get(currentParam);
if (paramValue === null) {
return acc;
Expand All @@ -93,9 +93,10 @@ const getParsedExploreURLSearchParams = (search: string) => {
if (typeof parsedParamValue === 'object') {
return { ...acc, ...parsedParamValue };
}
const key = EXPLORE_URL_SEARCH_PARAMS[currentParam]?.name || currentParam;
return {
...acc,
[EXPLORE_URL_SEARCH_PARAMS[currentParam].name]: parsedParamValue,
[key]: parsedParamValue,
};
}, {});
};
Expand Down
3 changes: 1 addition & 2 deletions superset/models/slice.py
Original file line number Diff line number Diff line change
Expand Up @@ -334,8 +334,7 @@ def icons(self) -> str:

@property
def url(self) -> str:
form_data = f"%7B%22slice_id%22%3A%20{self.id}%7D"
return f"/explore/?slice_id={self.id}&form_data={form_data}"
return f"/explore/?slice_id={self.id}"

def get_query_context_factory(self) -> QueryContextFactory:
if self.query_context_factory is None:
Expand Down

0 comments on commit 11d7d6e

Please sign in to comment.