~launchpad-pqm/launchpad/devel

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Copyright 2010 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

"""Test Archive software center agent celebrity."""

from zope.component import getUtility

from canonical.launchpad.webapp.authorization import check_permission
from canonical.testing.layers import LaunchpadFunctionalLayer
from lp.app.interfaces.launchpad import ILaunchpadCelebrities
from lp.soyuz.interfaces.archivesubscriber import IArchiveSubscriberSet
from lp.testing import (
    login,
    login_person,
    TestCaseWithFactory,
    )


class TestArchivePrivacy(TestCaseWithFactory):
    layer = LaunchpadFunctionalLayer

    def setUp(self):
        super(TestArchivePrivacy, self).setUp()
        self.ppa = self._makePrivateArchive()
        self.ppa.commercial = True
        self.agent = getUtility(ILaunchpadCelebrities).software_center_agent
        self.joe = self.factory.makePerson(name='joe')

    def _makePrivateArchive(self):
        ppa = self.factory.makeArchive()
        login('admin@canonical.com')
        ppa.buildd_secret = 'blah'
        ppa.private = True
        return ppa

    def test_check_permission(self):
        """The software center agent has the relevant permissions for a
        commercial archive, but not a private one.
        """
        login_person(self.agent)
        self.assertEqual(
            check_permission('launchpad.View', self.ppa), True)
        self.assertEqual(
            check_permission('launchpad.Append', self.ppa), True)

    def test_check_permission_private(self):
        ppa = self._makePrivateArchive()
        login_person(self.agent)
        self.assertEqual(check_permission('launchpad.View', ppa), False)
        self.assertEqual(check_permission('launchpad.Append', ppa), False)

    def test_add_subscription(self):
        login_person(self.agent)
        self.ppa.newSubscription(self.joe, self.agent)
        subscription = getUtility(
            IArchiveSubscriberSet).getBySubscriber(
            self.joe, archive=self.ppa).one()
        self.assertEqual(subscription.registrant, self.agent)
        self.assertEqual(subscription.subscriber, self.joe)

    def test_getArchiveSubscriptionURL(self):
        login_person(self.agent)
        sources = self.joe.getArchiveSubscriptionURL(self.agent, self.ppa)
        authtoken = self.ppa.getAuthToken(self.joe).token
        url = self.ppa.archive_url.split('http://')[1]
        new_url = "http://joe:%s@%s" % (authtoken, url)
        self.assertEqual(sources, new_url)