From 4401ef16885ae472a6048e0a285ed5e196aa4527 Mon Sep 17 00:00:00 2001
From: Vera Liu <vera.liu@airbnb.com>
Date: Mon, 27 Feb 2017 15:32:33 -0800
Subject: [PATCH] Add more tests to Save Modal specs

---
 .../explorev2/components/SaveModal.js         |  2 +
 .../explorev2/components/SaveModal_spec.js    | 43 ++++++++++++++++++-
 2 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/superset/assets/javascripts/explorev2/components/SaveModal.js b/superset/assets/javascripts/explorev2/components/SaveModal.js
index 2450ca47e3224..6be46505492a1 100644
--- a/superset/assets/javascripts/explorev2/components/SaveModal.js
+++ b/superset/assets/javascripts/explorev2/components/SaveModal.js
@@ -141,6 +141,7 @@ class SaveModal extends React.Component {
           }
           {this.props.slice &&
             <Radio
+              id="overwrite-radio"
               disabled={!this.props.can_overwrite}
               checked={this.state.action === 'overwrite'}
               onChange={this.changeAction.bind(this, 'overwrite')}
@@ -150,6 +151,7 @@ class SaveModal extends React.Component {
           }
 
           <Radio
+            id="saveas-radio"
             inline
             checked={this.state.action === 'saveas'}
             onChange={this.changeAction.bind(this, 'saveas')}
diff --git a/superset/assets/spec/javascripts/explorev2/components/SaveModal_spec.js b/superset/assets/spec/javascripts/explorev2/components/SaveModal_spec.js
index fa21b2f7bc2b7..157d5ffcbb410 100644
--- a/superset/assets/spec/javascripts/explorev2/components/SaveModal_spec.js
+++ b/superset/assets/spec/javascripts/explorev2/components/SaveModal_spec.js
@@ -14,7 +14,8 @@ const defaultProps = {
     saveSlice: sinon.spy(),
   },
   form_data: defaultFormData,
-  user_id: 1,
+  user_id: '1',
+  dashboards: [],
   slice: {},
 };
 
@@ -31,4 +32,44 @@ describe('SaveModal', () => {
     expect(wrapper.find(Button)).to.have.lengthOf(2);
     expect(wrapper.find(Radio)).to.have.lengthOf(5);
   });
+
+  it('does not show overwrite option for new slice', () => {
+    defaultProps.slice = null;
+    const wrapperNewSlice = shallow(<SaveModal {...defaultProps} />);
+    expect(wrapperNewSlice.find('#overwrite-radio')).to.have.lengthOf(0);
+    expect(wrapperNewSlice.find('#saveas-radio')).to.have.lengthOf(1);
+  });
+
+  it('disable overwrite option for non-owner', () => {
+    defaultProps.slice = {};
+    defaultProps.can_overwrite = false;
+    const wrapperForNonOwner = shallow(<SaveModal {...defaultProps} />);
+    const overwriteRadio = wrapperForNonOwner.find('#overwrite-radio');
+    expect(overwriteRadio).to.have.lengthOf(1);
+    expect(overwriteRadio.prop('disabled')).to.equal(true);
+  });
+
+  it('saves a new slice', () => {
+    defaultProps.slice = {
+      slice_id: 1,
+      slice_name: 'title',
+    };
+    defaultProps.can_overwrite = false;
+    const wrapperForNewSlice = shallow(<SaveModal {...defaultProps} />);
+    const saveasRadio = wrapperForNewSlice.find('#saveas-radio');
+    saveasRadio.simulate('click');
+    expect(wrapperForNewSlice.state().action).to.equal('saveas');
+  });
+
+  it('overwrite a slice', () => {
+    defaultProps.slice = {
+      slice_id: 1,
+      slice_name: 'title',
+    };
+    defaultProps.can_overwrite = true;
+    const wrapperForOverwrite = shallow(<SaveModal {...defaultProps} />);
+    const overwriteRadio = wrapperForOverwrite.find('#overwrite-radio');
+    overwriteRadio.simulate('click');
+    expect(wrapperForOverwrite.state().action).to.equal('overwrite');
+  });
 });