← Back to branch summary

~allenap/maas/neighbours-service-live

« back to all changes in this revision

Viewing changes to src/maasserver/websockets/handlers/tests/test_general.py

  • Committer: Gavin Panella
  • Date: 2015-10-14 19:48:39 UTC
  • mfrom: (3852.1.524 maas)
  • Revision ID: gavin.panella@canonical.com-20151014194839-xgjmom6qzxyj71pn
Merge trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/websockets/handlers/tests/test_general.py
14
14
__metaclass__ = type
15
15
__all__ = []
16
16
 
17
 
from maasserver.enum import NODE_PERMISSION
 
17
from maasserver.enum import (
 
18
    BOND_LACP_RATE_CHOICES,
 
19
    BOND_MODE_CHOICES,
 
20
    BOND_XMIT_HASH_POLICY_CHOICES,
 
21
    NODE_PERMISSION,
 
22
)
 
23
from maasserver.models.config import Config
18
24
from maasserver.node_action import ACTIONS_DICT
19
25
from maasserver.testing.factory import factory
20
26
from maasserver.testing.osystems import make_osystem_with_releases
48
54
    def test_osinfo(self):
49
55
        handler = GeneralHandler(factory.make_User(), {})
50
56
        osystem = make_osystem_with_releases(self)
51
 
        releases = [("", "Default OS Release")]
52
 
        for release in osystem["releases"]:
53
 
            releases.append((
54
 
                "%s/%s" % (osystem["name"], release["name"]),
55
 
                release["title"]))
 
57
        releases = [
 
58
            ("%s/%s" % (osystem["name"], release["name"]), release["title"])
 
59
            for release in osystem["releases"]
 
60
        ]
56
61
        expected_osinfo = {
57
62
            "osystems": [
58
 
                ("", "Default OS"),
59
63
                (osystem["name"], osystem["title"]),
60
 
                ],
 
64
            ],
61
65
            "releases": releases,
62
 
            }
 
66
            "kernels": None,
 
67
            "default_osystem": Config.objects.get_config("default_osystem"),
 
68
            "default_release": Config.objects.get_config(
 
69
                "default_distro_series"),
 
70
        }
63
71
        self.assertItemsEqual(expected_osinfo, handler.osinfo({}))
64
72
 
65
73
    def test_node_actions_for_admin(self):
105
113
            lambda: iter(hostnames))
106
114
        handler = GeneralHandler(factory.make_User(), {})
107
115
        self.assertEqual("", handler.random_hostname({}))
 
116
 
 
117
    def test_bond_options(self):
 
118
        handler = GeneralHandler(factory.make_User(), {})
 
119
        self.assertEquals({
 
120
            "modes": BOND_MODE_CHOICES,
 
121
            "lacp_rates": BOND_LACP_RATE_CHOICES,
 
122
            "xmit_hash_policies": BOND_XMIT_HASH_POLICY_CHOICES,
 
123
            }, handler.bond_options({}))