def test_successful_create(self, create_snapshot):
     ''' creating snapshot and testing idempotency '''
     set_module_args(self.set_default_args())
     my_obj = my_module()
     if not self.onbox:
         my_obj.server = self.server
     with pytest.raises(AnsibleExitJson) as exc:
         my_obj.apply()
     assert exc.value.args[0]['changed']
     create_snapshot.assert_called_with()
     # to reset na_helper from remembering the previous 'changed' value
     my_obj = my_module()
     if not self.onbox:
         my_obj.server = MockONTAPConnection('snapshot')
     with pytest.raises(AnsibleExitJson) as exc:
         my_obj.apply()
     assert not exc.value.args[0]['changed']
 def test_successful_modify(self, modify_snapshot):
     ''' modifying snapshot and testing idempotency '''
     data = self.set_default_args()
     data['comment'] = 'adding comment'
     data['snapmirror_label'] = 'label22'
     set_module_args(data)
     my_obj = my_module()
     if not self.onbox:
         my_obj.server = MockONTAPConnection('snapshot')
     with pytest.raises(AnsibleExitJson) as exc:
         my_obj.apply()
     assert exc.value.args[0]['changed']
     modify_snapshot.assert_called_with()
     # to reset na_helper from remembering the previous 'changed' value
     data['comment'] = 'new comment'
     data['snapmirror_label'] = 'label12'
     set_module_args(data)
     my_obj = my_module()
     if not self.onbox:
         my_obj.server = MockONTAPConnection('snapshot')
     with pytest.raises(AnsibleExitJson) as exc:
         my_obj.apply()
     assert not exc.value.args[0]['changed']
 def test_if_all_methods_catch_exception(self):
     module_args = {}
     module_args.update(self.set_default_args())
     set_module_args(module_args)
     my_obj = my_module()
     if not self.onbox:
         my_obj.server = MockONTAPConnection('snapshot_fail')
     with pytest.raises(AnsibleFailJson) as exc:
         my_obj.create_snapshot()
     assert 'Error creating snapshot ansible:' in exc.value.args[0]['msg']
     with pytest.raises(AnsibleFailJson) as exc:
         my_obj.delete_snapshot()
     assert 'Error deleting snapshot ansible:' in exc.value.args[0]['msg']
     with pytest.raises(AnsibleFailJson) as exc:
         my_obj.modify_snapshot()
     assert 'Error modifying snapshot ansible:' in exc.value.args[0]['msg']
 def test_ensure_get_called_existing(self):
     ''' test get_snapshot()  for existing snapshot'''
     set_module_args(self.set_default_args())
     my_obj = my_module()
     my_obj.server = MockONTAPConnection(kind='snapshot')
     assert my_obj.get_snapshot()
 def test_ensure_get_called(self):
     ''' test get_snapshot()  for non-existent snapshot'''
     set_module_args(self.set_default_args())
     my_obj = my_module()
     my_obj.server = self.server
     assert my_obj.get_snapshot() is None
 def test_module_fail_when_required_args_missing(self):
     ''' required arguments are reported as errors '''
     with pytest.raises(AnsibleFailJson) as exc:
         set_module_args({})
         my_module()
     print('Info: %s' % exc.value.args[0]['msg'])