← Back to branch summary

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

« back to all changes in this revision

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

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/tests/test_third_party_drivers.py
39
39
 
40
40
    def test_uses_commissioning_modaliases(self):
41
41
        test_data = b'hulla\nbaloo'
42
 
        node = factory.make_node()
 
42
        node = factory.make_Node()
43
43
        NodeResult.objects.store_data(
44
44
            node, commissioningscript.LIST_MODALIASES_OUTPUT_NAME,
45
45
            0, RESULT_TYPE.COMMISSIONING, Bin(test_data))
48
48
        self.assertEqual(['hulla', 'baloo'], aliases)
49
49
 
50
50
    def test_survives_no_commissioning_data(self):
51
 
        node = factory.make_node()
 
51
        node = factory.make_Node()
52
52
        aliases = node_modaliases(node)
53
53
        self.assertEqual([], aliases)
54
54
 
89
89
class TestGetThirdPartyCode(MAASTestCase):
90
90
 
91
91
    def test_finds_match(self):
92
 
        node = factory.make_node()
 
92
        node = factory.make_Node()
93
93
        mock = self.patch(third_party_drivers, 'match_aliases_to_driver')
94
94
        base_driver = dict(comment='hooray')
95
95
        mock.return_value = base_driver
102
102
        self.assertEqual('hooray', driver['comment'])
103
103
 
104
104
    def test_finds_no_match(self):
105
 
        node = factory.make_node()
 
105
        node = factory.make_Node()
106
106
        mock = self.patch(third_party_drivers, 'match_aliases_to_driver')
107
107
        mock.return_value = None
108
108
        driver = get_third_party_driver(node)