~smoser/ubuntu/trusty/maas/lp-1172566

« back to all changes in this revision

Viewing changes to src/provisioningserver/tests/test_tasks.py

  • Committer: Package Import Robot
  • Author(s): Andres Rodriguez
  • Date: 2014-04-03 13:45:02 UTC
  • mto: This revision was merged to the branch mainline in revision 58.
  • Revision ID: package-import@ubuntu.com-20140403134502-8a6wvuqwyuekufh0
Tags: upstream-1.5+bzr2227
ImportĀ upstreamĀ versionĀ 1.5+bzr2227

Show diffs side-by-side

added added

removed removed

Lines of Context:
641
641
        self.assertThat(recorder, MockCalledOnceWith(
642
642
            ['sudo', '-n', '-E', 'maas-import-pxe-files'], env=expected_env))
643
643
 
644
 
    def test_import_boot_images_sets_archive_locations(self):
645
 
        self.patch(tasks, 'call_and_check')
646
 
        archives = {
647
 
            'main_archive': self.make_archive_url('main'),
648
 
            'ports_archive': self.make_archive_url('ports'),
649
 
            'cloud_images_archive': self.make_archive_url('cloud-images'),
650
 
        }
651
 
        expected_settings = {
652
 
            parameter.upper(): value
653
 
            for parameter, value in archives.items()}
654
 
        import_boot_images(**archives)
655
 
        env = tasks.call_and_check.call_args[1]['env']
656
 
        archive_settings = {
657
 
            variable: value
658
 
            for variable, value in env.items()
659
 
            if variable.endswith('_ARCHIVE')
660
 
        }
661
 
        self.assertEqual(expected_settings, archive_settings)
662
 
 
663
644
    def test_import_boot_images_calls_callback(self):
664
645
        self.patch(tasks, 'call_and_check')
665
646
        mock_callback = Mock()