def test_parse_operating_system_repos(self, mock_init, mock_ub, mock_deb): mock_init.return_value = None data = { 'repos': REPOS_SAMPLE, 'codename': 'trusty' } driver = NailgunBuildImage() driver.data = data mock_deb_expected_calls = [] repos = [] for r in REPOS_SAMPLE: kwargs = { 'name': r['name'], 'uri': r['uri'], 'suite': r['suite'], 'section': r['section'], 'priority': r['priority'] } mock_deb_expected_calls.append(mock.call(**kwargs)) repos.append(objects.DEBRepo(**kwargs)) driver.parse_operating_system() mock_ub_instance = mock_ub.return_value mock_ub_instance.repos = repos mock_ub.assert_called_once_with( repos=repos, packages=NailgunBuildImage.DEFAULT_TRUSTY_PACKAGES) self.assertEqual(mock_deb_expected_calls, mock_deb.call_args_list[:len(REPOS_SAMPLE)]) self.assertEqual(driver.operating_system.repos, repos)
def test_parse_operating_system_packages_given(self, mock_init, mock_ub): mock_init.return_value = None data = { 'repos': [], 'codename': 'trusty', 'packages': ['pack'] } driver = NailgunBuildImage() driver.data = data mock_ub_instance = mock_ub.return_value mock_ub_instance.packages = data['packages'] driver.parse_operating_system() mock_ub.assert_called_once_with(repos=[], packages=data['packages']) self.assertEqual(driver.operating_system.packages, data['packages'])
def test_parse_operating_system_packages_not_given( self, mock_init, mock_ub): mock_init.return_value = None data = { 'repos': [], 'codename': 'trusty' } driver = NailgunBuildImage() driver.data = data mock_ub_instance = mock_ub.return_value mock_ub_instance.packages = NailgunBuildImage.DEFAULT_TRUSTY_PACKAGES driver.parse_operating_system() mock_ub.assert_called_once_with( repos=[], packages=NailgunBuildImage.DEFAULT_TRUSTY_PACKAGES) self.assertEqual(driver.operating_system.packages, NailgunBuildImage.DEFAULT_TRUSTY_PACKAGES)