def test_membership_already_exists(self, mock_get):
mock_get.return_value = ('instance', False)
hub = Mock(spec=Hub)()
hub.record_activity.return_value = True
app = Mock(spec=Application)()
result = views._add_hub_membership(hub, app)
mock_get.assert_called_once_with(hub=hub, application=app)
eq_(hub.record_activity.call_count, 0)
def test_membership_is_created_successfully(self, mock_get):
mock_get.return_value = ('instance', True)
hub = Mock(spec=Hub)()
hub.record_activity.return_value = True
app = Mock(spec=Application)()
result = views._add_hub_membership(hub, app)
mock_get.assert_called_once_with(hub=hub, application=app)
hub.record_activity.assert_called_once()
