def test_tethys_app_uninstall_handle_module_and_db_not_found(
self, mock_app, mock_extension, mock_installed_apps,
mock_installed_extensions, mock_stdout, mock_exit, mock_warn):
# Raise DoesNotExist on db query
mock_app.objects.get.return_value = mock.MagicMock()
mock_app.DoesNotExist = TethysApp.DoesNotExist
mock_extension.DoesNotExist = TethysExtension.DoesNotExist
mock_app.objects.get.side_effect = TethysApp.DoesNotExist
mock_extension.objects.get.return_value = mock.MagicMock()
mock_extension.objects.get.side_effect = TethysExtension.DoesNotExist
# No installed apps or extensions returned
mock_installed_apps.return_value = {}
mock_installed_extensions.return_value = {}
mock_exit.side_effect = SystemExit
cmd = tethys_app_uninstall.Command()
self.assertRaises(SystemExit,
cmd.handle,
app_or_extension=['tethysext.foo_extension'],
is_extension=True)
mock_installed_apps.assert_not_called()
mock_installed_extensions.assert_called()
mock_warn.assert_called_once()
def test_tethys_app_uninstall_handle_apps_delete_rmtree_Popen_remove_exceptions(
self, mock_app, mock_extension, mock_installed_apps,
mock_installed_extensions, mock_input, mock_stdout, mock_warnings,
mock_popen, mock_os_remove, mock_join, _):
mock_app.objects.get.return_value = mock.MagicMock()
mock_app.objects.get().delete.return_value = True
mock_extension.objects.get.return_value = mock.MagicMock()
mock_extension.objects.get().delete.return_value = True
mock_installed_apps.return_value = {'foo_app': '/foo/foo_app'}
mock_installed_extensions.return_value = {}
mock_input.side_effect = ['yes']
mock_popen.side_effect = KeyboardInterrupt
mock_os_remove.side_effect = [True, Exception]
mock_join.return_value = '/foo/tethysapp-foo-app-nspkg.pth'
cmd = tethys_app_uninstall.Command()
cmd.handle(app_or_extension=['tethysapp.foo_app'],
is_extension=False,
is_forced=False)
mock_installed_apps.assert_called_once()
mock_installed_extensions.assert_not_called()
self.assertIn('successfully uninstalled', mock_stdout.getvalue())
mock_warnings.assert_not_called(
) # Don't do the TethysModel.DoesNotExist exception this test
mock_app.objects.get.assert_called()
mock_app.objects.get().delete.assert_called_once()
mock_extension.objects.get.assert_called()
mock_extension.objects.get().delete.assert_not_called()
mock_popen.assert_called_once_with(
['pip', 'uninstall', '-y', 'tethysapp-foo_app'],
stderr=-2,
stdout=-1)
mock_join.assert_called()
def test_tethys_app_uninstall_add_arguments(self):
parser = ArgumentParser('foo_parser')
cmd = tethys_app_uninstall.Command()
cmd.add_arguments(parser)
self.assertIn('foo_parser', parser.format_usage())
self.assertIn('app_or_extension', parser.format_usage())
self.assertIn('[-e]', parser.format_usage())
self.assertIn('--extension', parser.format_help())
def test_tethys_app_uninstall_handle_apps_cancel(self, mock_installed_apps,
mock_installed_extensions,
mock_input, mock_stdout,
mock_exit):
mock_installed_apps.return_value = ['foo_app']
mock_installed_extensions.return_value = {}
mock_input.side_effect = ['foo', 'no']
mock_exit.side_effect = SystemExit
cmd = tethys_app_uninstall.Command()
self.assertRaises(SystemExit,
cmd.handle,
app_or_extension=['tethysapp.foo_app'],
is_extension=False)
mock_installed_apps.assert_called_once()
mock_installed_extensions.assert_not_called()
self.assertIn('Uninstall cancelled by user.', mock_stdout.getvalue())