def stamp(self, version=None):
"""Stamp the version table with the given version."""
return migration.stamp(version)
def test_stamp_none_version(self, stamp):
migration.stamp(None)
stamp.assert_called_once_with('fake_config', 'head')
def stamp(self, version=None):
"""Stamp the revision table with the given version."""
return migration.stamp(version)
def test_stamp(self, stamp):
migration.stamp('version_1')
stamp.assert_called_once_with('fake_config', 'version_1')
