def test_put(self): unit = Mock() storage = SharedStorage('git', '1234') storage.link = Mock() storage.put(unit) storage.link.assert_called_once_with(unit)