~launchpad-pqm/launchpad/devel

# Copyright 2009 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""Tests for classes in the lp.code.browser.bazaar module."""

__metaclass__ = type

from zope.security.proxy import removeSecurityProxy

from lp.code.browser.bazaar import BazaarApplicationView

from lp.services.webapp.authorization import check_permission

from lp.services.webapp.servers import LaunchpadTestRequest

from lp.testing import (

    ANONYMOUS,

    login,

    login_person,

    TestCaseWithFactory,

    )

from lp.testing.layers import DatabaseFunctionalLayer

class TestBazaarViewPreCacheLaunchpadPermissions(TestCaseWithFactory):

    """Test the precaching of launchpad.View permissions."""

    layer = DatabaseFunctionalLayer

    def getViewBranches(self, attribute):

        """Create the view and get the branches for `attribute`."""

        request = LaunchpadTestRequest()

        login(ANONYMOUS, request)

        view = BazaarApplicationView(object(), request)

        return getattr(view, attribute)

    def test_recently_registered(self):

        # Create a some private branches (stacked and unstacked) that the

        # logged in user would not normally see.

        private_branch = self.factory.makeAnyBranch(private=True)

        self.factory.makeAnyBranch(stacked_on=private_branch)

        branch = self.factory.makeAnyBranch()

        recent_branches = self.getViewBranches('recently_registered_branches')

        self.assertEqual(branch, recent_branches[0])

        self.assertTrue(check_permission('launchpad.View', branch))

    def makeBranchScanned(self, branch):

        """Make the branch appear scanned."""

        revision = self.factory.makeRevision()

        # Login an administrator so they can update the branch's details.

        login('admin@canonical.com')

        branch.updateScannedDetails(revision, 1)

    def test_recently_changed(self):

        # Create a some private branches (stacked and unstacked) that the

        # logged in user would not normally see.

        private_branch = self.factory.makeAnyBranch(private=True)

        stacked_private_branch = self.factory.makeAnyBranch(

            stacked_on=private_branch)

        branch = self.factory.makeAnyBranch()

        self.makeBranchScanned(stacked_private_branch)

        self.makeBranchScanned(branch)

        recent_branches = self.getViewBranches('recently_changed_branches')

        self.assertEqual(branch, recent_branches[0])

        self.assertTrue(check_permission('launchpad.View', branch))

    def test_recently_imported(self):

        # Create an import branch that is stacked on a private branch that the

        # logged in user would not normally see.  This would never happen in

        # reality, but hey, lets test the function actually works.

        private_branch = self.factory.makeAnyBranch(private=True)

        # A new code import needs a real user as the sender for the outgoing

        # email.

        login_person(self.factory.makePerson())

        private_code_import = self.factory.makeCodeImport()

        stacked_private_branch = private_code_import.branch

        naked_branch = removeSecurityProxy(stacked_private_branch)

        naked_branch.stacked_on = private_branch

        code_import = self.factory.makeCodeImport()

        branch = code_import.branch

        self.makeBranchScanned(stacked_private_branch)

        self.makeBranchScanned(branch)

        recent_branches = self.getViewBranches('recently_imported_branches')

        self.assertEqual(branch, recent_branches[0])

        self.assertTrue(check_permission('launchpad.View', branch))