def testYamlInlineChanges(self):
mockYaml = """1:
author: a1
comment: c1
tag: t1
sql: sql
rollback: rollback
2:
author: a2
comment: c2
sql: sql 2
rollback: rollback 2
"""
db = Database()
with patch('liquipy.db.open',
mock_open(read_data=mockYaml),
create=True):
changes = db.inputYamlToChangeSets('yamlPath')
changeKeys = changes.keys()
self.assertNotIn(0, changeKeys)
self.assertIn(1, changeKeys)
self.assertIn(2, changeKeys)
self.assertNotIn(3, changeKeys)
# Just test the structure of one. Other tests around how change sets are
# created from YAML input are in xmlwriter_test.py.
changeOne = changes[1]
self.assertIn('author', changeOne)
self.assertIn('comment', changeOne)
self.assertIn('tag', changeOne)
self.assertIn('sql', changeOne)
self.assertIn('rollback', changeOne)
def testYamlAbsoluteFileInclude(self):
mockMasterYaml = """include:
directory: migrations
"""
mockIncludedYaml = """1:
author: a1
comment: c1
sql: sql
rollback: rollback
"""
masterChangeSetFilePath = '/foo/bar/baz/master_changeset.yml'
db = Database()
def sideEffect(filePath, _mode):
mockFile = Mock()
if filePath == masterChangeSetFilePath:
mockFile.read = Mock(return_value=mockMasterYaml)
else:
self.assertEqual('/foo/bar/baz/migrations/include1.yml',
filePath)
mockFile.read = Mock(return_value=mockIncludedYaml)
return mockFile
mockOpen = mock_open()
mockOpen.side_effect = sideEffect
mockListdir = Mock(return_value=['include1.yml', 'other.thing'])
with patch('liquipy.db.open', mockOpen, create=True):
with patch('liquipy.db.os.listdir', mockListdir, create=True):
changes = db.inputYamlToChangeSets(masterChangeSetFilePath)
mockListdir.assert_called_once_with('/foo/bar/baz/migrations')
changeKeys = changes.keys()
self.assertNotIn(0, changeKeys)
self.assertIn(1, changeKeys)
self.assertNotIn(2, changeKeys)
# Just test the structure of one. Other tests around how change sets are
# created from YAML input are in xmlwriter_test.py.
changeOne = changes[1]
self.assertIn('author', changeOne)
self.assertIn('comment', changeOne)
self.assertIn('sql', changeOne)
self.assertIn('rollback', changeOne)
def testIncludeWithMissingDirectoryRaisesException(self):
mockMasterYaml = """include:
directory: migrations
"""
masterChangeSetFilePath = '/foo/bar/baz/master_changeset.yml'
db = Database()
def sideEffect(_filePath, _mode):
mockFile = Mock()
mockFile.read = Mock(return_value=mockMasterYaml)
return mockFile
mockOpen = mock_open()
mockOpen.side_effect = sideEffect
with patch('liquipy.db.open', mockOpen, create=True):
with self.assertRaisesRegexp(
Exception,
"Included directory '/foo/bar/baz/migrations' does not exist"
):
db.inputYamlToChangeSets(masterChangeSetFilePath)
