def test_db_manage_version(self, mock_version):
with mock.patch('sys.stdout', new=six.StringIO()) as fakeOutput:
mock_version.return_value = '123456'
db_manage.main()
self.assertEqual('Current DB revision is 123456\n',
fakeOutput.getvalue())
mock_version.assert_called_once_with()
def test_db_manage_version(self, mock_version):
with mock.patch('sys.stdout', new=six.StringIO()) as fakeOutput:
mock_version.return_value = '123456'
db_manage.main()
self.assertEqual('Current DB revision is 123456\n',
fakeOutput.getvalue())
mock_version.assert_called_once_with()
def test_db_manage_revision(self, mock_revision):
db_manage.main()
mock_revision.assert_called_once_with(
message='foo bar',
autogenerate=base.CONF.command.autogenerate)
def test_db_manage_stamp(self, mock_stamp):
db_manage.main()
mock_stamp.assert_called_once_with('foo bar')
def test_db_manage_upgrade(self, mock_upgrade):
db_manage.main()
mock_upgrade.assert_called_once_with(base.CONF.command.revision)
def test_db_manage_revision(self, mock_revision):
db_manage.main()
mock_revision.assert_called_once_with(
message='foo bar',
autogenerate=base.CONF.command.autogenerate)
def test_db_manage_stamp(self, mock_stamp):
db_manage.main()
mock_stamp.assert_called_once_with('foo bar')
def test_db_manage_upgrade(self, mock_upgrade):
db_manage.main()
mock_upgrade.assert_called_once_with(base.CONF.command.revision)
