def test_dispatch_calls_only_bbpgsql_error(self, mock_restorewal_main, mock_archivepgsql_main, mock_archivewal_main, mock_bbpgsql_error): bbpgsql_main(self.get_argv_for_cmd(self.bbpgsql_exe)) mock_bbpgsql_error.assert_called_once_with() self.assertFalse(mock_archivepgsql_main.called) self.assertFalse(mock_archivewal_main.called) self.assertFalse(mock_restorewal_main.called)
def test_invocation_using_unknown_fails(self, mock_exit, mock_stdout_write): bbpgsql_main(['unknown']) mock_stdout_write.assert_called_once_with(self.unknownMsg) mock_exit.assert_called_once_with(1)
def test_invocation_using_main_script_fails(self, mock_exit): bbpgsql_main(['bbpgsql', '-c', self.config_path]) mock_exit.assert_called_with(1)