def test_replace_postgres_local_file_source(self, mock_check_call):
command = Command(self.parser.parse_args(['-f', 'db.sql', '-n', 'destdb']))
command.replace_postgres_db(None)
expected_calls = [call(['dropdb', '--if-exists', 'destdb']),
call(['createdb', 'destdb']),
call(['pg_restore', '--no-acl', '--no-owner', '--dbname=destdb',
'db.sql'])]
self.assertEqual(expected_calls, mock_check_call.call_args_list)
def test_replace_postgres_db_url_file_source(self, mock_check_call, mock_urlopen):
mock_urlopen.return_value = tempfile.NamedTemporaryFile()
command = Command(self.parser.parse_args(['-u', 'http://www.example.com/', '-n', 'destdb']))
command.replace_postgres_db('http://www.example.com/')
expected_calls = [call(['dropdb', '--if-exists', 'destdb']),
call(['createdb', 'destdb']),
call(['pg_restore', '--no-acl', '--no-owner', '--dbname=destdb',
StringStartsWith('www_example_com-backup-')])]
self.assertEqual(expected_calls, mock_check_call.call_args_list)
def test_replace_postgres_db_source(self, mock_check_call, mock_urlopen):
mock_urlopen.return_value = tempfile.NamedTemporaryFile()
command = Command(self.parser.parse_args(['-b', 'sourcedb', '-n', 'destdb']))
command.replace_postgres_db(None)
expected_calls = [call(['pg_dump', '-Fc', '--no-acl', '--no-owner', '--dbname=sourcedb',
StringStartsWith('--file=sourcedb-backup-')]),
call(['dropdb', '--if-exists', 'destdb']),
call(['createdb', 'destdb']),
call(['pg_restore', '--no-acl', '--no-owner', '--dbname=destdb',
StringStartsWith('sourcedb-backup-')])]
self.assertEqual(expected_calls, mock_check_call.call_args_list)