diff --git a/superset-frontend/src/views/CRUD/data/database/DatabaseList.test.jsx b/superset-frontend/src/views/CRUD/data/database/DatabaseList.test.jsx index 12580d8ee73..964adc64d51 100644 --- a/superset-frontend/src/views/CRUD/data/database/DatabaseList.test.jsx +++ b/superset-frontend/src/views/CRUD/data/database/DatabaseList.test.jsx @@ -108,7 +108,7 @@ fetchMock.get(databaseRelatedEndpoint, { const useSelectorMock = jest.spyOn(redux, 'useSelector'); -describe('DatabaseList', () => { +describe('Admin DatabaseList', () => { useSelectorMock.mockReturnValue({ CSV_EXTENSIONS: ['csv'], EXCEL_EXTENSIONS: ['xls', 'xlsx'], @@ -207,6 +207,32 @@ describe('DatabaseList', () => { `"http://localhost/api/v1/database/?q=(filters:!((col:expose_in_sqllab,opr:eq,value:!t),(col:allow_run_async,opr:eq,value:!f),(col:database_name,opr:ct,value:fooo)),order_column:changed_on_delta_humanized,order_direction:desc,page:0,page_size:25)"`, ); }); + + it('should not render dropdown menu button if user is not admin', () => { + userSelectorMock.mockReturnValue({ + createdOn: '2021-05-27T18:12:38.952304', + email: 'alpha@gmail.com', + firstName: 'alpha', + isActive: true, + lastName: 'alpha', + permissions: {}, + roles: { + Alpha: [ + ['can_sqllab', 'Superset'], + ['can_write', 'Dashboard'], + ['can_write', 'Chart'], + ], + }, + userId: 2, + username: 'alpha', + }); + const newWrapper = mount( + + + , + ); + expect(newWrapper.find('.dropdown-menu-links')).not.toExist(); + }); }); describe('RTL', () => { diff --git a/superset-frontend/src/views/CRUD/data/database/DatabaseList.tsx b/superset-frontend/src/views/CRUD/data/database/DatabaseList.tsx index 10149bc9e8a..0080cc4e919 100644 --- a/superset-frontend/src/views/CRUD/data/database/DatabaseList.tsx +++ b/superset-frontend/src/views/CRUD/data/database/DatabaseList.tsx @@ -34,6 +34,7 @@ import ImportModelsModal from 'src/components/ImportModal/index'; import handleResourceExport from 'src/utils/export'; import { ExtentionConfigs } from 'src/views/components/types'; import { UserWithPermissionsAndRoles } from 'src/types/bootstrapTypes'; +import type { MenuObjectProps } from 'src/views/components/Menu'; import DatabaseModal from './DatabaseModal'; import { DatabaseObject } from './types'; @@ -217,11 +218,13 @@ function DatabaseList({ addDangerToast, addSuccessToast }: DatabaseListProps) { }, ]; - const filteredDropDown = uploadDropdownMenu.map(link => { + const filteredDropDown = uploadDropdownMenu.reduce((prev, cur) => { // eslint-disable-next-line no-param-reassign - link.childs = link.childs.filter(item => item.perm); - return link; - }); + cur.childs = cur.childs.filter(item => item.perm); + if (!cur.childs.length) return prev; + prev.push(cur); + return prev; + }, [] as MenuObjectProps[]); const menuData: SubMenuProps = { activeChild: 'Databases',