diff --git a/superset/assets/spec/javascripts/sqllab/SaveQuery_spec.jsx b/superset/assets/spec/javascripts/sqllab/SaveQuery_spec.jsx
index 98c8758a8db2..146a0746b833 100644
--- a/superset/assets/spec/javascripts/sqllab/SaveQuery_spec.jsx
+++ b/superset/assets/spec/javascripts/sqllab/SaveQuery_spec.jsx
@@ -1,9 +1,10 @@
import React from 'react';
-import { Overlay, Popover, FormControl } from 'react-bootstrap';
+import { FormControl } from 'react-bootstrap';
import { shallow } from 'enzyme';
import { describe, it } from 'mocha';
import { expect } from 'chai';
import SaveQuery from '../../../javascripts/SqlLab/components/SaveQuery';
+import ModalTrigger from '../../../javascripts/components/ModalTrigger';
describe('SavedQuery', () => {
const mockedProps = {
@@ -23,25 +24,18 @@ describe('SavedQuery', () => {
React.isValidElement(),
).to.equal(true);
});
- it('has an Overlay and a Popover', () => {
+ it('has a ModalTrigger', () => {
const wrapper = shallow();
- expect(wrapper.find(Overlay)).to.have.length(1);
- expect(wrapper.find(Popover)).to.have.length(1);
- });
- it('pops and hides', () => {
- const wrapper = shallow();
- expect(wrapper.state().showSave).to.equal(false);
- wrapper.find('.toggleSave').simulate('click', { target: { value: 'test' } });
- expect(wrapper.state().showSave).to.equal(true);
- wrapper.find('.toggleSave').simulate('click', { target: { value: 'test' } });
- expect(wrapper.state().showSave).to.equal(false);
+ expect(wrapper.find(ModalTrigger)).to.have.length(1);
});
it('has a cancel button', () => {
const wrapper = shallow();
- expect(wrapper.find('.cancelQuery')).to.have.length(1);
+ const modal = shallow(wrapper.instance().renderModalBody());
+ expect(modal.find('.cancelQuery')).to.have.length(1);
});
it('has 2 FormControls', () => {
const wrapper = shallow();
- expect(wrapper.find(FormControl)).to.have.length(2);
+ const modal = shallow(wrapper.instance().renderModalBody());
+ expect(modal.find(FormControl)).to.have.length(2);
});
});