~launchpad-pqm/launchpad/devel

# Copyright 2010-2011 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

"""Tests for `StructuralSubscription`."""

__metaclass__ = type

from storm.store import (
    EmptyResultSet,
    ResultSet,
    Store,
    )
from testtools.matchers import StartsWith
from zope.security.interfaces import Unauthorized

from lp.bugs.enum import BugNotificationLevel
from lp.bugs.interfaces.bugtask import (
    BugTaskImportance,
    BugTaskStatus,
    )
from lp.bugs.mail.bugnotificationrecipients import BugNotificationRecipients
from lp.bugs.model.bugsubscriptionfilter import (
    BugSubscriptionFilter,
    BugSubscriptionFilterMute,
    MuteNotAllowed,
    )
from lp.bugs.model.structuralsubscription import (
    get_structural_subscribers,
    get_structural_subscription_targets,
    get_structural_subscriptions,
    get_structural_subscriptions_for_bug,
    )
from lp.services.database.decoratedresultset import DecoratedResultSet
from lp.testing import (
    anonymous_logged_in,
    login_person,
    person_logged_in,
    TestCaseWithFactory,
    )
from lp.testing.factory import is_security_proxied_or_harmless
from lp.testing.layers import (
    DatabaseFunctionalLayer,
    LaunchpadFunctionalLayer,
    )


RESULT_SETS = ResultSet, EmptyResultSet, DecoratedResultSet


class TestStructuralSubscription(TestCaseWithFactory):

    layer = DatabaseFunctionalLayer

    def setUp(self):
        super(TestStructuralSubscription, self).setUp()
        self.product = self.factory.makeProduct()
        with person_logged_in(self.product.owner):
            self.subscription = self.product.addSubscription(
                self.product.owner, self.product.owner)
        self.original_filter = self.subscription.bug_filters[0]

    def test_delete_requires_Edit_permission(self):
        # delete() is only available to the subscriber.
        # We use a lambda here because a security proxy around
        # self.subscription is giving us the behavior we want to
        # demonstrate.  Merely accessing the "delete" name raises
        # Unauthorized, before the method is even called.  Therefore,
        # we use a lambda to make the trigger happen within "assertRaises".
        with anonymous_logged_in():
            self.assertRaises(Unauthorized, lambda: self.subscription.delete)
        with person_logged_in(self.factory.makePerson()):
            self.assertRaises(Unauthorized, lambda: self.subscription.delete)

    def test_simple_delete(self):
        with person_logged_in(self.product.owner):
            self.subscription.delete()
            self.assertEqual(
                self.product.getSubscription(self.product.owner), None)

    def test_delete_cascades_to_filters(self):
        with person_logged_in(self.product.owner):
            subscription_id = self.subscription.id
            self.subscription.newBugFilter()
            self.subscription.delete()
            self.assertEqual(
                self.product.getSubscription(self.product.owner), None)
            store = Store.of(self.product)
            # We know that the filter is gone, because we know the
            # subscription is gone, and the database would have
            # prevented the deletion of a subscription without first
            # deleting the filters.  We'll double-check, to be sure.
            self.assertEqual(
                store.find(
                    BugSubscriptionFilter,
                    BugSubscriptionFilter.structural_subscription_id ==
                        subscription_id).one(),
                None)

    def test_bug_filters_default(self):
        # The bug_filters attribute has a default non-filtering bug filter
        # to begin with.
        self.assertEqual([self.original_filter],
                         list(self.subscription.bug_filters))

    def test_bug_filters(self):
        # The bug_filters attribute returns the BugSubscriptionFilter records
        # associated with this subscription.
        subscription_filter = BugSubscriptionFilter()
        subscription_filter.structural_subscription = self.subscription
        self.assertContentEqual(
            [subscription_filter, self.original_filter],
            list(self.subscription.bug_filters))

    def test_newBugFilter(self):
        # newBugFilter() creates a new subscription filter linked to the
        # subscription.
        with person_logged_in(self.product.owner):
            subscription_filter = self.subscription.newBugFilter()
        self.assertEqual(
            self.subscription,
            subscription_filter.structural_subscription)
        self.assertContentEqual(
            [subscription_filter, self.original_filter],
            list(self.subscription.bug_filters))

    def test_newBugFilter_by_anonymous(self):
        # newBugFilter() is not available to anonymous users.
        with anonymous_logged_in():
            self.assertRaises(
                Unauthorized, lambda: self.subscription.newBugFilter)

    def test_newBugFilter_by_other_user(self):
        # newBugFilter() is only available to the subscriber.
        with person_logged_in(self.factory.makePerson()):
            self.assertRaises(
                Unauthorized, lambda: self.subscription.newBugFilter)


class FilteredStructuralSubscriptionTestBase:
    """Tests for filtered structural subscriptions."""

    layer = LaunchpadFunctionalLayer

    def makeTarget(self):
        raise NotImplementedError(self.makeTarget)

    def makeBugTask(self):
        return self.factory.makeBugTask(target=self.target)

    def setUp(self):
        super(FilteredStructuralSubscriptionTestBase, self).setUp()
        self.ordinary_subscriber = self.factory.makePerson()
        login_person(self.ordinary_subscriber)
        self.target = self.makeTarget()
        self.bugtask = self.makeBugTask()
        self.bug = self.bugtask.bug
        self.subscription = self.target.addSubscription(
            self.ordinary_subscriber, self.ordinary_subscriber)
        self.initial_filter = self.subscription.bug_filters[0]

    def assertSubscribers(
        self, expected_subscribers, level=BugNotificationLevel.LIFECYCLE):
        observed_subscribers = list(
            get_structural_subscribers(self.bugtask, None, level))
        self.assertEqual(expected_subscribers, observed_subscribers)

    def test_getStructuralSubscribers(self):
        # If no one has a filtered subscription for the given bug, the result
        # of get_structural_subscribers() is the same as for
        # the set of people from each subscription in getSubscriptions().
        subscriptions = self.target.getSubscriptions()
        self.assertSubscribers([sub.subscriber for sub in subscriptions])

    def test_getStructuralSubscribers_with_filter_on_status(self):
        # If a status filter exists for a subscription, the result of
        # get_structural_subscribers() may be a subset of getSubscriptions().

        # Without any filters the subscription is found.
        self.assertSubscribers([self.ordinary_subscriber])

        # Filter the subscription to bugs in the CONFIRMED state.
        self.initial_filter.statuses = [BugTaskStatus.CONFIRMED]

        # With the filter the subscription is not found.
        self.assertSubscribers([])

        # If the filter is adjusted, the subscription is found again.
        self.initial_filter.statuses = [self.bugtask.status]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_with_filter_on_importance(self):
        # If an importance filter exists for a subscription, the result of
        # get_structural_subscribers() may be a subset of getSubscriptions().

        # Without any filters the subscription is found.
        self.assertSubscribers([self.ordinary_subscriber])

        # Filter the subscription to bugs in the CRITICAL state.
        self.initial_filter.importances = [BugTaskImportance.CRITICAL]

        # With the filter the subscription is not found.
        self.assertSubscribers([])

        # If the filter is adjusted, the subscription is found again.
        self.initial_filter.importances = [self.bugtask.importance]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_with_filter_on_level(self):
        # All structural subscriptions have a level for bug notifications
        # which get_structural_subscribers() observes.

        # Adjust the subscription level to METADATA.
        self.initial_filter.bug_notification_level = (
            BugNotificationLevel.METADATA)

        # The subscription is found when looking for LIFECYCLE or above.
        self.assertSubscribers(
            [self.ordinary_subscriber], BugNotificationLevel.LIFECYCLE)
        # The subscription is found when looking for METADATA or above.
        self.assertSubscribers(
            [self.ordinary_subscriber], BugNotificationLevel.METADATA)
        # The subscription is not found when looking for COMMENTS or above.
        self.assertSubscribers(
            [], BugNotificationLevel.COMMENTS)

    def test_getStructuralSubscribers_with_filter_include_any_tags(self):
        # If a subscription filter has include_any_tags, a bug with one or
        # more tags is matched.

        self.initial_filter.include_any_tags = True

        # Without any tags the subscription is not found.
        self.assertSubscribers([])

        # With any tag the subscription is found.
        self.bug.tags = ["foo"]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_with_filter_exclude_any_tags(self):
        # If a subscription filter has exclude_any_tags, only bugs with no
        # tags are matched.

        self.initial_filter.exclude_any_tags = True

        # Without any tags the subscription is found.
        self.assertSubscribers([self.ordinary_subscriber])

        # With any tag the subscription is not found.
        self.bug.tags = ["foo"]
        self.assertSubscribers([])

    def test_getStructuralSubscribers_with_filter_for_any_tag(self):
        # If a subscription filter specifies that any of one or more specific
        # tags must be present, bugs with any of those tags are matched.

        # Looking for either the "foo" or the "bar" tag.
        self.initial_filter.tags = [u"foo", u"bar"]
        self.initial_filter.find_all_tags = False

        # Without either tag the subscription is not found.
        self.assertSubscribers([])

        # With either tag the subscription is found.
        self.bug.tags = ["bar", "baz"]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_with_filter_for_all_tags(self):
        # If a subscription filter specifies that all of one or more specific
        # tags must be present, bugs with all of those tags are matched.

        # Looking for both the "foo" and the "bar" tag.
        self.initial_filter.tags = [u"foo", u"bar"]
        self.initial_filter.find_all_tags = True

        # Without either tag the subscription is not found.
        self.assertSubscribers([])

        # Without only one of the required tags the subscription is not found.
        self.bug.tags = ["foo"]
        self.assertSubscribers([])

        # With both required tags the subscription is found.
        self.bug.tags = ["foo", "bar"]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_with_filter_for_not_any_tag(self):
        # If a subscription filter specifies that any of one or more specific
        # tags must not be present, bugs without any of those tags are
        # matched.

        # Looking to exclude the "foo" or "bar" tags.
        self.initial_filter.tags = [u"-foo", u"-bar"]
        self.initial_filter.find_all_tags = False

        # Without either tag the subscription is found.
        self.assertSubscribers([self.ordinary_subscriber])

        # With both tags, the subscription is omitted.
        self.bug.tags = ["foo", "bar"]
        self.assertSubscribers([])

        # With only one tag, the subscription is found again.
        self.bug.tags = ["foo"]
        self.assertSubscribers([self.ordinary_subscriber])

        # However, if find_all_tags is True, even a single excluded tag
        # causes the subscription to be skipped.
        self.initial_filter.find_all_tags = True
        self.assertSubscribers([])

        # This is also true, of course, if the bug has both tags.
        self.bug.tags = ["foo", "bar"]
        self.assertSubscribers([])

    def test_getStructuralSubscribers_with_filter_for_not_all_tags(self):
        # If a subscription filter specifies that all of one or more specific
        # tags must not be present, bugs without all of those tags are
        # matched.

        # Looking to exclude the "foo" and "bar" tags.
        self.initial_filter.tags = [u"-foo", u"-bar"]
        self.initial_filter.find_all_tags = True

        # Without either tag the subscription is found.
        self.assertSubscribers([self.ordinary_subscriber])

        # With only one of the excluded tags the subscription is not
        # found--we are saying that we want to find both an absence of foo
        # and an absence of bar, and yet foo exists.
        self.bug.tags = ["foo"]
        self.assertSubscribers([])

        # With both tags the subscription is also not found.
        self.bug.tags = ["foo", "bar"]
        self.assertSubscribers([])

    def test_getStructuralSubscribers_with_multiple_filters(self):
        # If multiple filters exist for a subscription, all filters must
        # match.

        # Add the "foo" tag to the bug.
        self.bug.tags = ["foo"]
        self.assertSubscribers([self.ordinary_subscriber])

        # Filter the subscription to bugs in the CRITICAL state.
        self.initial_filter.statuses = [BugTaskStatus.CONFIRMED]
        self.initial_filter.importances = [BugTaskImportance.CRITICAL]

        # With the filter the subscription is not found.
        self.assertSubscribers([])

        # If the filter is adjusted to match status but not importance, the
        # subscription is still not found.
        self.initial_filter.statuses = [self.bugtask.status]
        self.assertSubscribers([])

        # If the filter is adjusted to also match importance, the subscription
        # is found again.
        self.initial_filter.importances = [self.bugtask.importance]
        self.assertSubscribers([self.ordinary_subscriber])

        # If the filter is given some tag criteria, the subscription is not
        # found.
        self.initial_filter.tags = [u"-foo", u"bar", u"baz"]
        self.initial_filter.find_all_tags = False
        self.assertSubscribers([])

        # After removing the "foo" tag and adding the "bar" tag, the
        # subscription is found.
        self.bug.tags = ["bar"]
        self.assertSubscribers([self.ordinary_subscriber])

        # Requiring that all tag criteria are fulfilled causes the
        # subscription to no longer be found.
        self.initial_filter.find_all_tags = True
        self.assertSubscribers([])

        # After adding the "baz" tag, the subscription is found again.
        self.bug.tags = ["bar", "baz"]
        self.assertSubscribers([self.ordinary_subscriber])

    def test_getStructuralSubscribers_any_filter_is_a_match(self):
        # If a subscription has multiple filters, the subscription is selected
        # when any filter is found to match. Put another way, the filters are
        # ORed together.
        subscription_filter1 = self.initial_filter
        subscription_filter1.statuses = [BugTaskStatus.CONFIRMED]
        subscription_filter2 = self.subscription.newBugFilter()
        subscription_filter2.tags = [u"foo"]

        # With the filter the subscription is not found.
        self.assertSubscribers([])

        # If the bugtask is adjusted to match the criteria of the first filter
        # but not those of the second, the subscription is found.
        self.bugtask.transitionToStatus(
            BugTaskStatus.CONFIRMED, self.ordinary_subscriber)
        self.assertSubscribers([self.ordinary_subscriber])

        # If the filter is adjusted to also match the criteria of the second
        # filter, the subscription is still found.
        self.bugtask.bug.tags = [u"foo"]
        self.assertSubscribers([self.ordinary_subscriber])

        # If the bugtask is adjusted to no longer match the criteria of the
        # first filter, the subscription is still found.
        self.bugtask.transitionToStatus(
            BugTaskStatus.INPROGRESS, self.ordinary_subscriber)
        self.assertSubscribers([self.ordinary_subscriber])


class TestStructuralSubscriptionFiltersForDistro(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeDistribution()


class TestStructuralSubscriptionFiltersForProduct(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeProduct()


class TestStructuralSubscriptionFiltersForDistroSourcePackage(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeDistributionSourcePackage()


class TestStructuralSubscriptionFiltersForMilestone(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeMilestone()

    def makeBugTask(self):
        bug = self.factory.makeBug(milestone=self.target)
        return bug.bugtasks[0]


class TestStructuralSubscriptionFiltersForDistroSeries(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeDistroSeries()


class TestStructuralSubscriptionFiltersForProjectGroup(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeProject()

    def makeBugTask(self):
        return self.factory.makeBugTask(
            target=self.factory.makeProduct(project=self.target))


class TestStructuralSubscriptionFiltersForProductSeries(
    FilteredStructuralSubscriptionTestBase, TestCaseWithFactory):

    def makeTarget(self):
        return self.factory.makeProductSeries()


class TestGetStructuralSubscriptionTargets(TestCaseWithFactory):

    layer = DatabaseFunctionalLayer

    def test_product_target(self):
        product = self.factory.makeProduct()
        bug = self.factory.makeBug(product=product)
        bugtask = bug.bugtasks[0]
        result = get_structural_subscription_targets(bug.bugtasks)
        self.assertEqual(list(result), [(bugtask, product)])

    def test_milestone_target(self):
        actor = self.factory.makePerson()
        login_person(actor)
        product = self.factory.makeProduct()
        milestone = self.factory.makeMilestone(product=product)
        bug = self.factory.makeBug(product=product, milestone=milestone)
        bugtask = bug.bugtasks[0]
        result = get_structural_subscription_targets(bug.bugtasks)
        self.assertEqual(set(result), set(
            ((bugtask, product), (bugtask, milestone))))

    def test_sourcepackage_target(self):
        actor = self.factory.makePerson()
        login_person(actor)
        distroseries = self.factory.makeDistroSeries()
        sourcepackage = self.factory.makeSourcePackage(
            distroseries=distroseries, publish=True)
        product = self.factory.makeProduct()
        bug = self.factory.makeBug(product=product)
        bug.addTask(actor, sourcepackage)
        product_bugtask = bug.bugtasks[0]
        sourcepackage_bugtask = bug.bugtasks[1]
        result = get_structural_subscription_targets(bug.bugtasks)
        self.assertEqual(set(result), set(
            ((product_bugtask, product),
             (sourcepackage_bugtask, distroseries))))

    def test_distribution_source_package_target(self):
        actor = self.factory.makePerson()
        login_person(actor)
        distribution = self.factory.makeDistribution()
        dist_sourcepackage = self.factory.makeDistributionSourcePackage(
            distribution=distribution)
        product = self.factory.makeProduct()
        bug = self.factory.makeBug(product=product)
        bug.addTask(actor, dist_sourcepackage)
        product_bugtask = bug.bugtasks[0]
        dist_sourcepackage_bugtask = bug.bugtasks[1]
        result = get_structural_subscription_targets(bug.bugtasks)
        self.assertEqual(set(result), set(
            ((product_bugtask, product),
             (dist_sourcepackage_bugtask, dist_sourcepackage),
             (dist_sourcepackage_bugtask, distribution))))

    def test_product_with_project_group(self):
        # get_structural_subscription_targets() will yield both a
        # product and its parent project group if it has one.
        project = self.factory.makeProject()
        product = self.factory.makeProduct(
            project=project, owner=project.owner)
        subscriber = self.factory.makePerson()
        with person_logged_in(subscriber):
            project.addBugSubscription(subscriber, subscriber)
        # This is a sanity check.
        self.assertEqual(project, product.parent_subscription_target)
        bug = self.factory.makeBug(product=product)
        result = get_structural_subscription_targets(bug.bugtasks)
        self.assertEqual(
            set([(bug.bugtasks[0], product), (bug.bugtasks[0], project)]),
            set(result))


class TestGetStructuralSubscriptionsForBug(TestCaseWithFactory):

    layer = DatabaseFunctionalLayer

    def setUp(self):
        super(TestGetStructuralSubscriptionsForBug, self).setUp()
        self.subscriber = self.factory.makePerson()
        self.team = self.factory.makeTeam(members=[self.subscriber])
        login_person(self.subscriber)
        self.product = self.factory.makeProduct()
        self.milestone = self.factory.makeMilestone(product=self.product)
        self.bug = self.factory.makeBug(
            product=self.product, milestone=self.milestone)

    def getSubscriptions(self, person=None):
        result = get_structural_subscriptions_for_bug(self.bug, person)
        self.assertTrue(is_security_proxied_or_harmless(result))
        return result

    def test_no_subscriptions(self):
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual([], list(subscriptions))

    def test_one_subscription(self):
        sub = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual([sub], list(subscriptions))

    def test_two_subscriptions(self):
        sub1 = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        sub2 = self.milestone.addBugSubscription(
            self.subscriber, self.subscriber)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual(set([sub1, sub2]), set(subscriptions))

    def test_two_bugtasks_one_subscription(self):
        sub = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        product2 = self.factory.makeProduct()
        self.bug.addTask(self.subscriber, product2)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual([sub], list(subscriptions))

    def test_two_bugtasks_two_subscriptions(self):
        sub1 = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        product2 = self.factory.makeProduct()
        self.bug.addTask(self.subscriber, product2)
        sub2 = product2.addBugSubscription(
            self.subscriber, self.subscriber)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual(set([sub1, sub2]), set(subscriptions))

    def test_ignore_other_subscriptions(self):
        sub1 = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        another_subscriber = self.factory.makePerson()
        login_person(another_subscriber)
        sub2 = self.product.addBugSubscription(
            another_subscriber, another_subscriber)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual([sub1], list(subscriptions))
        subscriptions = self.getSubscriptions(another_subscriber)
        self.assertEqual([sub2], list(subscriptions))

    def test_team_subscription(self):
        with person_logged_in(self.team.teamowner):
            sub = self.product.addBugSubscription(
                self.team, self.team.teamowner)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual([sub], list(subscriptions))

    def test_both_subscriptions(self):
        self_sub = self.product.addBugSubscription(
            self.subscriber, self.subscriber)
        with person_logged_in(self.team.teamowner):
            team_sub = self.product.addBugSubscription(
                self.team, self.team.teamowner)
        subscriptions = self.getSubscriptions(self.subscriber)
        self.assertEqual(set([self_sub, team_sub]), set(subscriptions))

    def test_subscriptions_from_parent(self):
        # get_structural_subscriptions_for_bug() will return any
        # structural subscriptions from the parents of the targets of
        # that bug.
        project = self.factory.makeProject()
        product = self.factory.makeProduct(
            project=project, owner=project.owner)
        subscriber = self.factory.makePerson()
        self_sub = project.addBugSubscription(subscriber, subscriber)
        # This is a sanity check.
        self.assertEqual(project, product.parent_subscription_target)
        bug = self.factory.makeBug(product=product)
        subscriptions = get_structural_subscriptions_for_bug(
            bug, subscriber)
        self.assertEqual(set([self_sub]), set(subscriptions))


class TestGetStructuralSubscriptions(TestCaseWithFactory):

    layer = DatabaseFunctionalLayer

    def make_product_with_bug(self):
        product = self.factory.makeProduct()
        bug = self.factory.makeBug(product=product)
        return product, bug

    def test_get_structural_subscriptions_no_subscriptions(self):
        # If there are no subscriptions for any of the bug's targets then no
        # subscriptions will be returned by get_structural_subscriptions().
        product, bug = self.make_product_with_bug()
        subscriptions = get_structural_subscriptions(bug, None)
        self.assertIsInstance(subscriptions, RESULT_SETS)
        self.assertEqual([], list(subscriptions))

    def test_get_structural_subscriptions_single_target(self):
        # Subscriptions for any of the bug's targets are returned.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        subscription = product.addBugSubscription(subscriber, subscriber)
        self.assertContentEqual(
            [subscription], get_structural_subscriptions(bug, None))

    def test_get_structural_subscriptions_multiple_targets(self):
        # Subscriptions for any of the bug's targets are returned.
        actor = self.factory.makePerson()
        login_person(actor)

        subscriber1 = self.factory.makePerson()
        subscriber2 = self.factory.makePerson()

        product1 = self.factory.makeProduct(owner=actor)
        subscription1 = product1.addBugSubscription(subscriber1, subscriber1)
        product2 = self.factory.makeProduct(owner=actor)
        subscription2 = product2.addBugSubscription(subscriber2, subscriber2)

        bug = self.factory.makeBug(product=product1)
        bug.addTask(actor, product2)

        subscriptions = get_structural_subscriptions(bug, None)
        self.assertIsInstance(subscriptions, RESULT_SETS)
        self.assertContentEqual(
            [subscription1, subscription2], subscriptions)

    def test_get_structural_subscriptions_multiple_targets_2(self):
        # Only the first of multiple subscriptions for a person is returned
        # when they have multiple matching subscriptions.
        actor = self.factory.makePerson()
        login_person(actor)

        subscriber = self.factory.makePerson()
        product1 = self.factory.makeProduct(owner=actor)
        subscription1 = product1.addBugSubscription(subscriber, subscriber)
        product2 = self.factory.makeProduct(owner=actor)
        product2.addBugSubscription(subscriber, subscriber)

        bug = self.factory.makeBug(product=product1)
        bug.addTask(actor, product2)

        subscriptions = get_structural_subscriptions(bug, None)
        self.assertIsInstance(subscriptions, RESULT_SETS)
        self.assertContentEqual([subscription1], subscriptions)

    def test_get_structural_subscriptions_level(self):
        # get_structural_subscriptions() respects the given level.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        subscription = product.addBugSubscription(subscriber, subscriber)
        filter = subscription.bug_filters.one()
        filter.bug_notification_level = BugNotificationLevel.METADATA
        self.assertContentEqual(
            [subscription], get_structural_subscriptions(
                bug, BugNotificationLevel.METADATA))
        self.assertContentEqual(
            [], get_structural_subscriptions(
                bug, BugNotificationLevel.COMMENTS))

    def test_get_structural_subscriptions_exclude(self):
        # Subscriptions for any of the given excluded subscribers are not
        # returned.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        product.addBugSubscription(subscriber, subscriber)
        self.assertContentEqual(
            [], get_structural_subscriptions(
                bug, None, exclude=[subscriber]))


class TestGetStructuralSubscribers(TestCaseWithFactory):

    layer = DatabaseFunctionalLayer

    def make_product_with_bug(self):
        product = self.factory.makeProduct()
        bug = self.factory.makeBug(product=product)
        return product, bug

    def test_getStructuralSubscribers_no_subscribers(self):
        # If there are no subscribers for any of the bug's targets then no
        # subscribers will be returned by get_structural_subscribers().
        product, bug = self.make_product_with_bug()
        subscribers = get_structural_subscribers(bug, None, None, None)
        self.assertIsInstance(subscribers, RESULT_SETS)
        self.assertEqual([], list(subscribers))

    def test_getStructuralSubscribers_single_target(self):
        # Subscribers for any of the bug's targets are returned.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        product.addBugSubscription(subscriber, subscriber)
        self.assertEqual(
            [subscriber], list(
                get_structural_subscribers(bug, None, None, None)))

    def test_getStructuralSubscribers_multiple_targets(self):
        # Subscribers for any of the bug's targets are returned.
        actor = self.factory.makePerson()
        login_person(actor)

        subscriber1 = self.factory.makePerson()
        subscriber2 = self.factory.makePerson()

        product1 = self.factory.makeProduct(owner=actor)
        product1.addBugSubscription(subscriber1, subscriber1)
        product2 = self.factory.makeProduct(owner=actor)
        product2.addBugSubscription(subscriber2, subscriber2)

        bug = self.factory.makeBug(product=product1)
        bug.addTask(actor, product2)

        subscribers = get_structural_subscribers(bug, None, None, None)
        self.assertIsInstance(subscribers, RESULT_SETS)
        self.assertEqual(set([subscriber1, subscriber2]), set(subscribers))

    def test_getStructuralSubscribers_recipients(self):
        # If provided, get_structural_subscribers() calls the appropriate
        # methods on a BugNotificationRecipients object.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        product.addBugSubscription(subscriber, subscriber)
        recipients = BugNotificationRecipients()
        subscribers = get_structural_subscribers(bug, recipients, None, None)
        # The return value is a list only when populating recipients.
        self.assertIsInstance(subscribers, list)
        self.assertEqual([subscriber], recipients.getRecipients())
        reason, header = recipients.getReason(subscriber)
        self.assertThat(
            reason, StartsWith(
                u"You received this bug notification because "
                u"you are subscribed to "))
        self.assertThat(header, StartsWith(u"Subscriber "))

    def test_getStructuralSubscribers_level(self):
        # get_structural_subscribers() respects the given level.
        subscriber = self.factory.makePerson()
        login_person(subscriber)
        product, bug = self.make_product_with_bug()
        subscription = product.addBugSubscription(subscriber, subscriber)
        filter = subscription.bug_filters.one()
        filter.bug_notification_level = BugNotificationLevel.METADATA
        self.assertEqual(
            [subscriber], list(
                get_structural_subscribers(
                    bug, None, BugNotificationLevel.METADATA, None)))
        filter.bug_notification_level = BugNotificationLevel.METADATA
        self.assertEqual(
            [], list(
                get_structural_subscribers(
                    bug, None, BugNotificationLevel.COMMENTS, None)))


class TestBugSubscriptionFilterMute(TestCaseWithFactory):
    """Tests for the BugSubscriptionFilterMute class."""

    layer = DatabaseFunctionalLayer

    def setUp(self):
        super(TestBugSubscriptionFilterMute, self).setUp()
        self.target = self.factory.makeProduct()
        self.team = self.factory.makeTeam()
        self.team_member = self.factory.makePerson()
        with person_logged_in(self.team.teamowner):
            self.team.addMember(self.team_member, self.team.teamowner)
            self.team_subscription = self.target.addBugSubscription(
                self.team, self.team.teamowner)
            self.filter = self.team_subscription.bug_filters.one()

    def test_isMuteAllowed_returns_true_for_team_subscriptions(self):
        # BugSubscriptionFilter.isMuteAllowed() will return True for
        # subscriptions where the owner of the subscription is a team.
        self.assertTrue(self.filter.isMuteAllowed(self.team_member))

    def test_isMuteAllowed_returns_false_for_non_team_subscriptions(self):
        # BugSubscriptionFilter.isMuteAllowed() will return False for
        # subscriptions where the owner of the subscription is not a team.
        person = self.factory.makePerson()
        with person_logged_in(person):
            non_team_subscription = self.target.addBugSubscription(
                person, person)
        filter = non_team_subscription.bug_filters.one()
        self.assertFalse(filter.isMuteAllowed(person))

    def test_isMuteAllowed_returns_false_for_non_team_members(self):
        # BugSubscriptionFilter.isMuteAllowed() will return False if the
        # user passed to it is not a member of the subscribing team.
        non_team_person = self.factory.makePerson()
        self.assertFalse(self.filter.isMuteAllowed(non_team_person))

    def test_mute_adds_mute(self):
        # BugSubscriptionFilter.mute() adds a mute for the filter.
        filter_id = self.filter.id
        person_id = self.team_member.id
        store = Store.of(self.filter)
        mutes = store.find(
            BugSubscriptionFilterMute,
            BugSubscriptionFilterMute.filter == filter_id,
            BugSubscriptionFilterMute.person == person_id)
        self.assertTrue(mutes.is_empty())
        self.assertFalse(self.filter.muted(self.team_member))
        self.filter.mute(self.team_member)
        self.assertTrue(self.filter.muted(self.team_member))
        store.flush()
        self.assertFalse(mutes.is_empty())

    def test_unmute_removes_mute(self):
        # BugSubscriptionFilter.unmute() removes any mute for a given
        # person on that filter.
        filter_id = self.filter.id
        person_id = self.team_member.id
        store = Store.of(self.filter)
        self.filter.mute(self.team_member)
        store.flush()
        mutes = store.find(
            BugSubscriptionFilterMute,
            BugSubscriptionFilterMute.filter == filter_id,
            BugSubscriptionFilterMute.person == person_id)
        self.assertFalse(mutes.is_empty())
        self.assertTrue(self.filter.muted(self.team_member))
        self.filter.unmute(self.team_member)
        self.assertFalse(self.filter.muted(self.team_member))
        store.flush()
        self.assertTrue(mutes.is_empty())

    def test_mute_is_idempotent(self):
        # Muting works even if the user is already muted.
        store = Store.of(self.filter)
        mute = self.filter.mute(self.team_member)
        store.flush()
        second_mute = self.filter.mute(self.team_member)
        self.assertEqual(mute, second_mute)

    def test_unmute_is_idempotent(self):
        # Unmuting works even if the user is not muted
        store = Store.of(self.filter)
        mutes = store.find(
            BugSubscriptionFilterMute,
            BugSubscriptionFilterMute.filter == self.filter.id,
            BugSubscriptionFilterMute.person == self.team_member.id)
        self.assertTrue(mutes.is_empty())
        self.filter.unmute(self.team_member)
        self.assertTrue(mutes.is_empty())

    def test_mute_raises_error_for_non_team_subscriptions(self):
        # BugSubscriptionFilter.mute() will raise an error if called on
        # a non-team subscription.
        person = self.factory.makePerson()
        with person_logged_in(person):
            non_team_subscription = self.target.addBugSubscription(
                person, person)
        filter = non_team_subscription.bug_filters.one()
        self.assertFalse(filter.isMuteAllowed(person))
        self.assertRaises(MuteNotAllowed, filter.mute, person)

    def test_mute_raises_error_for_non_team_members(self):
        # BugSubscriptionFilter.mute() will raise an error if called on
        # a subscription of which the calling person is not a member.
        non_team_person = self.factory.makePerson()
        self.assertFalse(self.filter.isMuteAllowed(non_team_person))
        self.assertRaises(MuteNotAllowed, self.filter.mute, non_team_person)