def testWriterRaisesExceptionWhenRollbackMissing(self):
changes = {'1': {
'author': 'Author',
'comment': 'Comment',
'sql': 'SQL',
'tag': 1.0
}}
writer = XMLWriter('')
with self.assertRaisesRegexp(
Exception,
"ChangeSet '1' missing required attribute 'rollback'"):
writer.write(changes)
def testOneSimpleChangeSetWithTagSuccess(self):
changes = {'1': {
'author': 'Author',
'comment': 'Comment',
'sql': 'SQL',
'rollback': 'Rollback',
'tag': 1.0
}}
expected = """<?xml version="1.0" encoding="UTF-8"?>
<databaseChangeLog
logicalFilePath="liquipy_logical_changelog_path.xml"
xmlns="http://www.liquibase.org/xml/ns/dbchangelog"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xmlns:ext="http://www.liquibase.org/xml/ns/dbchangelog-ext"
xsi:schemaLocation="http://www.liquibase.org/xml/ns/dbchangelog http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-3.0.xsd
http://www.liquibase.org/xml/ns/dbchangelog-ext http://www.liquibase.org/xml/ns/dbchangelog/dbchangelog-ext.xsd">
<changeSet id="1" author="Author">
<comment>Comment</comment>
<sql><![CDATA[
SQL
]]></sql>
<rollback><![CDATA[
Rollback
]]></rollback>
</changeSet>
<changeSet id="1-tag" author="liquipy">
<tagDatabase tag="1.0"/>
</changeSet>
</databaseChangeLog>
"""
writer = XMLWriter('outputFilePath')
mockOpen = mock_open()
mockOutputFile = MagicMock()
mockOpen.return_value = mockOutputFile
with patch('liquipy.changeset.open', mockOpen, create=True):
writer.write(changes)
_, fileWriteArgs, _ = mockOutputFile.mock_calls[1]
mockOpen.assert_called_once_with('outputFilePath', 'w')
self.assertEqual(expected, fileWriteArgs[0])
