test_helpers: Move get_subscription out of actions.py.

It's no longer used by anything other than test code.
This commit is contained in:
Tim Abbott 2017-01-29 19:31:24 -08:00
parent 035e442b63
commit 01daa3e91a
4 changed files with 11 additions and 11 deletions

View File

@ -1409,13 +1409,6 @@ def pick_color_helper(user_profile, subs):
else:
return STREAM_ASSIGNMENT_COLORS[len(used_colors) % len(STREAM_ASSIGNMENT_COLORS)]
def get_subscription(stream_name, user_profile):
# type: (Text, UserProfile) -> Subscription
stream = get_stream(stream_name, user_profile.realm)
recipient = get_recipient(Recipient.STREAM, stream.id)
return Subscription.objects.get(user_profile=user_profile,
recipient=recipient, active=True)
def validate_user_access_to_subscribers(user_profile, stream):
# type: (Optional[UserProfile], Stream) -> None
""" Validates whether the user can view the subscribers of a stream. Raises a JsonableError if:

View File

@ -30,6 +30,7 @@ from zerver.lib.actions import (
)
from zerver.models import (
get_recipient,
get_stream,
get_user_profile_by_email,
Client,
@ -206,6 +207,13 @@ def most_recent_message(user_profile):
usermessage = most_recent_usermessage(user_profile)
return usermessage.message
def get_subscription(stream_name, user_profile):
# type: (Text, UserProfile) -> Subscription
stream = get_stream(stream_name, user_profile.realm)
recipient = get_recipient(Recipient.STREAM, stream.id)
return Subscription.objects.get(user_profile=user_profile,
recipient=recipient, active=True)
def get_user_messages(user_profile):
# type: (UserProfile) -> List[Message]
query = UserMessage.objects. \

View File

@ -60,11 +60,10 @@ from zerver.lib.actions import (
do_add_realm_alias,
do_remove_realm_alias,
fetch_initial_state_data,
get_subscription
)
from zerver.lib.message import render_markdown
from zerver.lib.test_helpers import POSTRequestMock
from zerver.lib.test_helpers import POSTRequestMock, get_subscription
from zerver.lib.test_classes import (
ZulipTestCase,
)

View File

@ -9,7 +9,7 @@ from django.utils.translation import ugettext as _
from zerver.lib import cache
from zerver.lib.test_helpers import (
queries_captured, tornado_redirected_to_list
get_subscription, queries_captured, tornado_redirected_to_list
)
from zerver.lib.test_classes import (
@ -39,7 +39,7 @@ from zerver.lib.actions import (
do_create_realm, do_remove_default_stream, do_set_realm_create_stream_by_admins_only,
gather_subscriptions_helper, bulk_add_subscriptions, bulk_remove_subscriptions,
gather_subscriptions, get_default_streams_for_realm, get_realm, get_stream,
get_user_profile_by_email, set_default_streams, get_subscription,
get_user_profile_by_email, set_default_streams,
create_stream_if_needed, create_streams_if_needed, active_user_ids
)