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()