def test_from_version_and_to_version(self, upgrade):
persister = Mock()
upgrade(persister, from_version='0.1', to_version='0.2')
persister.upgrade.assert_called_with(from_version='0.1',
to_version='0.2')
def test_from_version_and_to_version(self, upgrade):
persister = Mock()
upgrade(persister, from_version='0.1', to_version='0.2')
persister.upgrade.assert_called_with(from_version='0.1',
to_version='0.2')
def test_no_args(self, upgrade):
persister = Mock()
upgrade(persister)
persister.upgrade.assert_called_with(from_version=None)
def test_no_args(self, upgrade):
persister = Mock()
upgrade(persister)
persister.upgrade.assert_called_with(from_version=None)