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

« back to all changes in this revision

Viewing changes to src/maasserver/rpc/tests/test_bootsources.py

merged upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
23
23
 
24
24
    def test_returns_boot_sources_and_selections(self):
25
25
        keyring = factory.make_bytes()
26
 
        nodegroup = factory.make_node_group()
27
 
        source = factory.make_boot_source(keyring_data=keyring)
28
 
        factory.make_boot_source_selection(source)
 
26
        nodegroup = factory.make_NodeGroup()
 
27
        source = factory.make_BootSource(keyring_data=keyring)
 
28
        factory.make_BootSourceSelection(source)
29
29
 
30
30
        expected = source.to_dict()
31
31
        self.assertEqual([expected], get_boot_sources(nodegroup.uuid))
32
32
 
33
33
    def test_removes_os_from_boot_source_selections(self):
34
34
        keyring = factory.make_bytes()
35
 
        nodegroup = factory.make_node_group()
36
 
        source = factory.make_boot_source(keyring_data=keyring)
37
 
        factory.make_boot_source_selection(source)
 
35
        nodegroup = factory.make_NodeGroup()
 
36
        source = factory.make_BootSource(keyring_data=keyring)
 
37
        factory.make_BootSourceSelection(source)
38
38
 
39
39
        expected = source.to_dict()
40
40
        del expected['selections'][0]['os']