~rvb/maas/transaction-1.7-bug-1409852

« back to all changes in this revision

Viewing changes to src/maasserver/tests/test_forms_nodeaction.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
45
45
 
46
46
    def test_get_action_form_creates_form_class(self):
47
47
        user = factory.make_admin()
48
 
        node = factory.make_node(status=NODE_STATUS.NEW)
 
48
        node = factory.make_Node(status=NODE_STATUS.NEW)
49
49
        form = get_action_form(user)(node)
50
50
 
51
51
        self.assertIsInstance(form, NodeActionForm)
53
53
 
54
54
    def test_get_action_form_for_admin(self):
55
55
        admin = factory.make_admin()
56
 
        node = factory.make_node(
 
56
        node = factory.make_Node(
57
57
            status=NODE_STATUS.NEW, boot_type=NODE_BOOT.DEBIAN)
58
58
        form = get_action_form(admin)(node)
59
59
 
62
62
            form.actions)
63
63
 
64
64
    def test_get_action_form_for_user(self):
65
 
        user = factory.make_user()
66
 
        node = factory.make_node(status=NODE_STATUS.NEW)
 
65
        user = factory.make_User()
 
66
        node = factory.make_Node(status=NODE_STATUS.NEW)
67
67
        form = get_action_form(user)(node)
68
68
 
69
69
        self.assertIsInstance(form, NodeActionForm)
72
72
 
73
73
    def test_save_performs_requested_action(self):
74
74
        admin = factory.make_admin()
75
 
        node = factory.make_node(status=NODE_STATUS.NEW)
 
75
        node = factory.make_Node(status=NODE_STATUS.NEW)
76
76
        form = get_action_form(admin)(
77
77
            node, {NodeActionForm.input_name: Commission.name})
78
78
        self.assertTrue(form.is_valid())
80
80
        self.assertEqual(NODE_STATUS.COMMISSIONING, node.status)
81
81
 
82
82
    def test_rejects_disallowed_action(self):
83
 
        user = factory.make_user()
84
 
        node = factory.make_node(status=NODE_STATUS.NEW)
 
83
        user = factory.make_User()
 
84
        node = factory.make_Node(status=NODE_STATUS.NEW)
85
85
        form = get_action_form(user)(
86
86
            node, {NodeActionForm.input_name: Commission.name})
87
87
        self.assertFalse(form.is_valid())
90
90
            form._errors)
91
91
 
92
92
    def test_rejects_unknown_action(self):
93
 
        user = factory.make_user()
94
 
        node = factory.make_node(status=NODE_STATUS.NEW)
 
93
        user = factory.make_User()
 
94
        node = factory.make_Node(status=NODE_STATUS.NEW)
95
95
        action = factory.make_string()
96
96
        form = get_action_form(user)(
97
97
            node, {NodeActionForm.input_name: action})
103
103
        error_text = factory.make_string(prefix="NodeActionError")
104
104
        exc = NodeActionError(error_text)
105
105
        self.patch(StartNode, "execute").side_effect = exc
106
 
        user = factory.make_user()
107
 
        node = factory.make_node(
 
106
        user = factory.make_User()
 
107
        node = factory.make_Node(
108
108
            status=NODE_STATUS.ALLOCATED, owner=user)
109
109
        action = StartNode.name
110
110
        # Required for messages to work: