~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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# Copyright 2009 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

"""Tests for BranchVisibilityPolicy."""

__metaclass__ = type

from lp.code.enums import BranchVisibilityRule
from lp.code.interfaces.branchvisibilitypolicy import (
    IHasBranchVisibilityPolicy,
    InvalidVisibilityPolicy,
    )
from lp.services.webapp.testing import verifyObject
from lp.testing import TestCaseWithFactory
from lp.testing.layers import DatabaseFunctionalLayer


class TestIHasBranchVisibilityPolicy(TestCaseWithFactory):
    """Tests for `IHasBranchVisibilityPolicy`."""

    layer = DatabaseFunctionalLayer

    def test_product_implements_IHasBranchVisibilityPolicy(self):
        # Products can have visibility policies.
        product = self.factory.makeProduct()
        verifyObject(IHasBranchVisibilityPolicy, product)

    def test_project_implements_IHasBranchVisibilityPolicy(self):
        # Products can have visibility policies.
        project = self.factory.makeProject()
        verifyObject(IHasBranchVisibilityPolicy, project)


class TestBaseBranchVisibilityRules(TestCaseWithFactory):
    """Tests for `getBaseBranchVisibilityRule`."""

    layer = DatabaseFunctionalLayer

    def test_no_rules(self):
        # If there are no rules specified, the default is PUBLIC.
        product = self.factory.makeProduct()
        self.assertEqual(
            BranchVisibilityRule.PUBLIC,
            product.getBaseBranchVisibilityRule())

    def test_explicit_public(self):
        # If there is an explicit public rule, then still public.
        product = self.factory.makeProduct()
        product.setBranchVisibilityTeamPolicy(
            None, BranchVisibilityRule.PUBLIC)
        self.assertEqual(
            BranchVisibilityRule.PUBLIC,
            product.getBaseBranchVisibilityRule())

    def test_explicit_forbidden(self):
        # If there is an explicit public rule, then still public.
        product = self.factory.makeProduct()
        product.setBranchVisibilityTeamPolicy(
            None, BranchVisibilityRule.FORBIDDEN)
        self.assertEqual(
            BranchVisibilityRule.FORBIDDEN,
            product.getBaseBranchVisibilityRule())

    def test_inherited_public(self):
        # If there is an explicit public rule, then still public.
        project = self.factory.makeProject()
        product = self.factory.makeProduct(project=project)
        project.setBranchVisibilityTeamPolicy(
            None, BranchVisibilityRule.PUBLIC)
        self.assertEqual(
            BranchVisibilityRule.PUBLIC,
            product.getBaseBranchVisibilityRule())

    def test_inherited_forbidden(self):
        # If there is an explicit public rule, then still public.
        project = self.factory.makeProject()
        product = self.factory.makeProduct(project=project)
        project.setBranchVisibilityTeamPolicy(
            None, BranchVisibilityRule.FORBIDDEN)
        self.assertEqual(
            BranchVisibilityRule.FORBIDDEN,
            product.getBaseBranchVisibilityRule())

    def test_no_default_private(self):
        # The default visibility rule cannot be PRIVATE or PRIVATE_ONLY.
        product = self.factory.makeProduct()
        self.assertRaises(
            InvalidVisibilityPolicy,
            product.setBranchVisibilityTeamPolicy,
            None, BranchVisibilityRule.PRIVATE)
        self.assertRaises(
            InvalidVisibilityPolicy,
            product.setBranchVisibilityTeamPolicy,
            None, BranchVisibilityRule.PRIVATE_ONLY)

    def test_no_forbidden_for_team(self):
        # Forbidden is only valid for everyone, not a specific person or team.
        product = self.factory.makeProduct()
        person = self.factory.makePerson()
        self.assertRaises(
            InvalidVisibilityPolicy,
            product.setBranchVisibilityTeamPolicy,
            person, BranchVisibilityRule.FORBIDDEN)
        team = self.factory.makeTeam(owner=person)
        self.assertRaises(
            InvalidVisibilityPolicy,
            product.setBranchVisibilityTeamPolicy,
            team, BranchVisibilityRule.FORBIDDEN)