def test_replace_heroku_db_no_file_url(self, mock_check_call):
command = Command(self.parser.parse_args(['-d', 'app2']))
command.databases['source']['name'] = 'srcdb'
command.replace_heroku_db(None)
expected_calls = [call(['heroku', 'pg:reset', '--app=app2', 'DATABASE_URL']),
call(['heroku', 'pg:push', 'srcdb', 'DATABASE_URL', '--app=app2'])]
self.assertEqual(expected_calls, mock_check_call.call_args_list)
def test_replace_heroku_db_with_file_url(self, mock_check_call):
command = Command(self.parser.parse_args(['-u', 'www.example.com', '-d', 'app2']))
command.databases['source']['name'] = 'srcdb'
command.replace_heroku_db('www.example.com')
expected_calls = [call(['heroku', 'pg:reset', '--app=app2', 'DATABASE_URL']),
call(['heroku', 'pg:backups:restore', 'www.example.com', '--app=app2',
'DATABASE', '--confirm', 'app2'])]
self.assertEqual(expected_calls, mock_check_call.call_args_list)