def test_update_heads_file_two_heads(self):
with mock.patch("alembic.script.ScriptDirectory.from_config") as fc:
heads = ("b", "a")
fc.return_value.get_heads.return_value = heads
with mock.patch("six.moves.builtins.open") as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = mock.Mock()
cli.update_heads_file(mock.sentinel.config)
mock_open.return_value.write.assert_called_once_with("\n".join(sorted(heads)))
def test_update_heads_file_success(self):
with mock.patch('alembic.script.ScriptDirectory.from_config') as fc:
heads = ('a', 'b')
fc.return_value.get_heads.return_value = heads
with mock.patch('six.moves.builtins.open') as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = mock.Mock()
cli.update_heads_file(mock.sentinel.config)
mock_open.return_value.write.assert_called_once_with(
'\n'.join(heads))
def test_update_heads_file_two_heads(self):
with mock.patch('alembic.script.ScriptDirectory.from_config') as fc:
heads = ('b', 'a')
fc.return_value.get_heads.return_value = heads
with mock.patch('six.moves.builtins.open') as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = mock.Mock()
cli.update_heads_file(self.configs[0])
mock_open.return_value.write.assert_called_once_with(
'\n'.join(sorted(heads)))
def test_update_heads_file_success(self, *os_mocks):
with mock.patch('alembic.script.ScriptDirectory.from_config') as fc:
heads = ('a', 'b')
fc.return_value.get_heads.return_value = heads
with mock.patch('six.moves.builtins.open') as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = mock.Mock()
cli.update_heads_file(self.configs[0])
mock_open.return_value.write.assert_called_once_with(
'\n'.join(heads))
old_head_file = cli._get_head_file_path(self.configs[0])
for mock_ in os_mocks:
mock_.assert_called_with(old_head_file)
def test_update_heads_file_success(self, *os_mocks):
with mock.patch('alembic.script.ScriptDirectory.from_config') as fc:
heads = ('a', 'b')
fc.return_value.get_heads.return_value = heads
with mock.patch('six.moves.builtins.open') as mock_open:
mock_open.return_value.__enter__ = lambda s: s
mock_open.return_value.__exit__ = mock.Mock()
cli.update_heads_file(self.configs[0])
mock_open.return_value.write.assert_called_once_with(
'\n'.join(heads))
old_head_file = cli._get_head_file_path(self.configs[0])
for mock_ in os_mocks:
mock_.assert_called_with(old_head_file)
