diff --git a/superset-frontend/spec/javascripts/sqllab/LimitControl_spec.jsx b/superset-frontend/spec/javascripts/sqllab/LimitControl_spec.jsx
index 53fb00d6add76..c05a13837bdac 100644
--- a/superset-frontend/spec/javascripts/sqllab/LimitControl_spec.jsx
+++ b/superset-frontend/spec/javascripts/sqllab/LimitControl_spec.jsx
@@ -99,7 +99,10 @@ describe('LimitControl', () => {
wrappingComponent: ThemeProvider,
wrappingComponentProps: { theme: supersetTheme },
});
- popoverContentWrapper.find('.ok').first().simulate('click');
+ popoverContentWrapper
+ .find('[data-test="limit-control-confirm"]')
+ .first()
+ .simulate('click');
expect(wrapper.state().showOverlay).toBe(false);
});
it('resets and closes', () => {
@@ -114,7 +117,10 @@ describe('LimitControl', () => {
wrappingComponent: ThemeProvider,
wrappingComponentProps: { theme: supersetTheme },
});
- popoverContentWrapper.find('.reset').first().simulate('click');
+ popoverContentWrapper
+ .find('[data-test="limit-control-cancel"]')
+ .first()
+ .simulate('click');
expect(wrapper.state().textValue).toEqual(
defaultProps.defaultQueryLimit.toString(),
);
diff --git a/superset-frontend/spec/javascripts/sqllab/SaveQuery_spec.jsx b/superset-frontend/spec/javascripts/sqllab/SaveQuery_spec.jsx
index 6baa96e25ed47..231427bde29ce 100644
--- a/superset-frontend/spec/javascripts/sqllab/SaveQuery_spec.jsx
+++ b/superset-frontend/spec/javascripts/sqllab/SaveQuery_spec.jsx
@@ -48,7 +48,7 @@ describe('SavedQuery', () => {
const wrapper = shallow();
const modal = wrapper.find(Modal);
- expect(modal.find('.cancelQuery')).toHaveLength(1);
+ expect(modal.find('[data-test="cancel-query"]')).toHaveLength(1);
});
it('has 2 FormControls', () => {
const wrapper = shallow();
diff --git a/superset-frontend/spec/javascripts/sqllab/SqlEditor_spec.jsx b/superset-frontend/spec/javascripts/sqllab/SqlEditor_spec.jsx
index 15559c7cf877c..58e7ae25e03c7 100644
--- a/superset-frontend/spec/javascripts/sqllab/SqlEditor_spec.jsx
+++ b/superset-frontend/spec/javascripts/sqllab/SqlEditor_spec.jsx
@@ -108,7 +108,7 @@ describe('SqlEditor', () => {
it('allows toggling autocomplete', () => {
const wrapper = shallow();
expect(wrapper.find(AceEditorWrapper).props().autocomplete).toBe(true);
- wrapper.find('.autocomplete').simulate('click');
+ wrapper.find('[data-test="autocomplete"]').simulate('click');
expect(wrapper.find(AceEditorWrapper).props().autocomplete).toBe(false);
});
});
diff --git a/superset-frontend/src/SqlLab/components/ExploreResultsButton.jsx b/superset-frontend/src/SqlLab/components/ExploreResultsButton.jsx
index 388e74f195fad..ea52f09bed94c 100644
--- a/superset-frontend/src/SqlLab/components/ExploreResultsButton.jsx
+++ b/superset-frontend/src/SqlLab/components/ExploreResultsButton.jsx
@@ -75,7 +75,7 @@ class ExploreResultsButton extends React.PureComponent {
this.dialog.show({
title: t('Explore'),
body: msg,
- actions: [Dialog.DefaultAction('Ok', () => {}, 'btn-primary')],
+ actions: [Dialog.DefaultAction('Ok', () => {})],
bsSize: 'large',
bsStyle: 'warning',
onHide: dialog => {
diff --git a/superset-frontend/src/SqlLab/components/LimitControl.tsx b/superset-frontend/src/SqlLab/components/LimitControl.tsx
index 95348e321a7e1..d771f37edaa53 100644
--- a/superset-frontend/src/SqlLab/components/LimitControl.tsx
+++ b/superset-frontend/src/SqlLab/components/LimitControl.tsx
@@ -118,7 +118,8 @@ export default class LimitControl extends React.PureComponent<
-