← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to lib/lp/services/features/tests/test_xmlrpc.py

  • Committer: Jelmer Vernooij
  • Date: 2011-09-21 14:28:02 UTC
  • mfrom: (14006 devel)
  • mto: This revision was merged to the branch mainline in revision 14010.
  • Revision ID: jelmer@canonical.com-20110921142802-7ggkc204igsy532w
Merge lp:launchpad

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/lp/services/features/tests/test_xmlrpc.py
 
1
# Copyright 2011 Canonical Ltd.  This software is licensed under the
 
2
# GNU Affero General Public License version 3 (see the file LICENSE).
 
3
 
 
4
"""Tests for FeatureFlagApplication."""
 
5
 
 
6
__metaclass__ = type
 
7
 
 
8
import xmlrpclib
 
9
 
 
10
from canonical.testing.layers import DatabaseFunctionalLayer
 
11
from canonical.config import config
 
12
from lp.services import features
 
13
from lp.services.features.flags import FeatureController
 
14
from lp.services.features.rulesource import StormFeatureRuleSource
 
15
from lp.services.features.scopes import (
 
16
    DefaultScope,
 
17
    FixedScope,
 
18
    MultiScopeHandler,
 
19
    )
 
20
from lp.services.features.xmlrpc import FeatureFlagApplication
 
21
from lp.testing import (
 
22
    feature_flags,
 
23
    set_feature_flag,
 
24
    TestCaseWithFactory,
 
25
    )
 
26
from lp.testing.xmlrpc import XMLRPCTestTransport
 
27
 
 
28
 
 
29
class TestGetFeatureFlag(TestCaseWithFactory):
 
30
 
 
31
    layer = DatabaseFunctionalLayer
 
32
 
 
33
    def setUp(self):
 
34
        TestCaseWithFactory.setUp(self)
 
35
        self.endpoint = FeatureFlagApplication()
 
36
 
 
37
    def installFeatureController(self, feature_controller):
 
38
        old_features = features.get_relevant_feature_controller()
 
39
        features.install_feature_controller(feature_controller)
 
40
        self.addCleanup(
 
41
            features.install_feature_controller, old_features)
 
42
 
 
43
    def test_getFeatureFlag_returns_None_by_default(self):
 
44
        self.assertIs(None, self.endpoint.getFeatureFlag(u'unknown'))
 
45
 
 
46
    def test_getFeatureFlag_returns_true_for_set_flag(self):
 
47
        flag_name = u'flag'
 
48
        with feature_flags():
 
49
            set_feature_flag(flag_name, u'1')
 
50
            self.assertEqual(u'1', self.endpoint.getFeatureFlag(flag_name))
 
51
 
 
52
    def test_getFeatureFlag_ignores_relevant_feature_controller(self):
 
53
        # getFeatureFlag should only consider the scopes it is asked to
 
54
        # consider, not any that happen to be active due to the XML-RPC
 
55
        # request itself.
 
56
        flag_name = u'flag'
 
57
        scope_name = u'scope'
 
58
        self.installFeatureController(
 
59
            FeatureController(
 
60
                MultiScopeHandler(
 
61
                    [DefaultScope(), FixedScope(scope_name)]).lookup,
 
62
                StormFeatureRuleSource()))
 
63
        set_feature_flag(flag_name, u'1', scope_name)
 
64
        self.assertEqual(None, self.endpoint.getFeatureFlag(flag_name))
 
65
 
 
66
    def test_getFeatureFlag_considers_supplied_scope(self):
 
67
        flag_name = u'flag'
 
68
        scope_name = u'scope'
 
69
        with feature_flags():
 
70
            set_feature_flag(flag_name, u'value', scope_name)
 
71
            self.assertEqual(
 
72
                u'value',
 
73
                self.endpoint.getFeatureFlag(flag_name, [scope_name]))
 
74
 
 
75
    def test_getFeatureFlag_turns_user_into_team_scope(self):
 
76
        flag_name = u'flag'
 
77
        person = self.factory.makePerson()
 
78
        team = self.factory.makeTeam(members=[person])
 
79
        with feature_flags():
 
80
            set_feature_flag(flag_name, u'value', u'team:' + team.name)
 
81
            self.assertEqual(
 
82
                u'value',
 
83
                self.endpoint.getFeatureFlag(
 
84
                    flag_name, ['user:' + person.name]))
 
85
 
 
86
    def test_xmlrpc_interface_unset(self):
 
87
        sp = xmlrpclib.ServerProxy(
 
88
            config.launchpad.feature_flags_endpoint,
 
89
            transport=XMLRPCTestTransport(), allow_none=True)
 
90
        self.assertEqual(None, sp.getFeatureFlag(u'flag'))
 
91
 
 
92
    def test_xmlrpc_interface_set(self):
 
93
        sp = xmlrpclib.ServerProxy(
 
94
            config.launchpad.feature_flags_endpoint,
 
95
            transport=XMLRPCTestTransport(), allow_none=True)
 
96
        flag_name = u'flag'
 
97
        with feature_flags():
 
98
            set_feature_flag(flag_name, u'1')
 
99
            self.assertEqual(u'1', sp.getFeatureFlag(flag_name))