def test_export_package_changes_dpkg(self, mock_os_listdir, mock_os_path_exists, mock_command): command = Mock() command.output = 'changes' mock_command.return_value = command self.xml_state.get_package_manager = Mock(return_value='apt') mock_os_listdir.return_value = ['package_b', 'package_a'] mock_os_path_exists.return_value = True with patch('builtins.open', create=True) as mock_open: mock_open.return_value = MagicMock(spec=io.IOBase) result = self.setup.export_package_changes('target_dir') file_handle = mock_open.return_value.__enter__.return_value mock_open.assert_called_once_with( 'target_dir/some-image.x86_64-1.2.3.changes', 'w', encoding='utf-8') assert result == 'target_dir/some-image.x86_64-1.2.3.changes' assert file_handle.write.call_args_list == [ call('package_a|\n'), call('changes\n'), call('package_b|\n'), call('changes\n') ] self.runtime_config.get_package_changes.assert_called_once_with()
def test_add_repo_with_gpgchecks(self, mock_exists, mock_buildservice, mock_config): repo_config = mock.Mock() mock_buildservice.return_value = False mock_config.return_value = repo_config mock_exists.return_value = True with patch('builtins.open', create=True) as mock_open: self.repo.add_repo('foo', 'kiwi_iso_mount/uri', 'rpm-md', 42, repo_gpgcheck=False, pkg_gpgcheck=True) repo_config.add_section.assert_called_once_with('foo') assert repo_config.set.call_args_list == [ call('foo', 'name', 'foo'), call('foo', 'baseurl', 'file://kiwi_iso_mount/uri'), call('foo', 'priority', '42'), call('foo', 'repo_gpgcheck', '0'), call('foo', 'gpgcheck', '1') ] mock_open.assert_called_once_with('/shared-dir/dnf/repos/foo.repo', 'w')
def test_export_package_changes_rpm(self, mock_MountManager, mock_RpmDataBase, mock_command): rpmdb = Mock() rpmdb.rpmdb_image.expand_query.return_value = 'image_dbpath' rpmdb.rpmdb_host.expand_query.return_value = 'host_dbpath' rpmdb.has_rpm.return_value = True mock_RpmDataBase.return_value = rpmdb command = Mock() command.output = 'package|\nchanges' mock_command.return_value = command with patch('builtins.open', create=True) as mock_open: mock_open.return_value = MagicMock(spec=io.IOBase) result = self.setup.export_package_changes('target_dir') mock_open.assert_called_once_with( 'target_dir/some-image.x86_64-1.2.3.changes', 'w', encoding='utf-8') assert result == 'target_dir/some-image.x86_64-1.2.3.changes' mock_command.assert_called_once_with([ 'rpm', '--root', 'root_dir', '-qa', '--qf', '%{NAME}|\\n', '--changelog', '--dbpath', 'image_dbpath' ]) self.runtime_config.get_package_changes.assert_called_once_with()
def test_add_repo(self, mock_exists, mock_buildservice, mock_config): repo_config = mock.Mock() mock_buildservice.return_value = False mock_config.return_value = repo_config mock_exists.return_value = True with patch('builtins.open', create=True) as mock_open: self.repo.add_repo('foo', 'kiwi_iso_mount/uri', 'rpm-md', 42) repo_config.add_section.assert_called_once_with('foo') assert repo_config.set.call_args_list == [ call('foo', 'name', 'foo'), call('foo', 'baseurl', 'file://kiwi_iso_mount/uri'), call('foo', 'priority', '42') ] mock_open.assert_called_once_with('/shared-dir/dnf/repos/foo.repo', 'w') repo_config.add_section.reset_mock() repo_config.set.reset_mock() mock_exists.return_value = False with patch('builtins.open', create=True) as mock_open: self.repo.add_repo('bar', 'https://metalink', 'rpm-md', sourcetype='metalink') repo_config.add_section.assert_called_once_with('bar') assert repo_config.set.call_args_list == [ call('bar', 'name', 'bar'), call('bar', 'metalink', 'https://metalink') ] mock_open.assert_called_once_with('/shared-dir/dnf/repos/bar.repo', 'w')
def test_tfstate_bucket_set_up_in_dev_account_for_aslive_deployment( self, mock_open): # When cli.run(['deploy', 'live', '1.2.3']) # Then mock_open.assert_called_once_with('.terragrunt', 'w')
def test_loads_config_filepath(self, mock_yaml): mock_yaml.load.return_value = self.config with patch('__builtin__.open') as mock_open: r = self._C('/path/to/file.yaml') eq_(self.tempdir, r['NGSDATA']) mock_open.assert_called_once_with('/path/to/file.yaml') mock_yaml.load.assert_called_once_with(mock_open())
def test_render(self, mock_open, get_template_mock, get_context_mock): self.renderer.render('foo') mock_open.assert_called_once_with('foo', 'w') handle = mock_open() handle.write.assert_called_once() get_template_mock.assert_called_with('base.html') get_template_mock.return_value.render.assert_called_with( {'foo': 'bar'})
def test_get_board_as_rows(self, mocked_out): with patch('sudoku_solver.sudoku_solver.open', create=True) as mock_open: mocked_out.return_value = self.row_form_board mock_unsolved_puzzle = "mock_unsolved_puzzle.csv" test_rows_extracted = get_board_as_rows( mock_unsolved_puzzle) mock_open.assert_called_once_with(mock_unsolved_puzzle) mocked_out.assert_called_once(mock_unsolved_puzzle) self.assertEqual(self.row_form_board, test_rows_extracted)
def test_guardar_version_actual_fail(self, mock_open, mock_syslog): ''' Prueba el tratamiento del error en caso de no poder guardar la ultima version en el archivo de versiones definido en config.LOCAL_VERSION. ''' # preparo datos mock_open.side_effect = IOError() # llamo metodo a probar self.actualizador.guardar_version_actual() # verifico que todo este bien mock_open.assert_called_once_with(config.NETCOP['local_version'], 'w') mock_syslog.assert_called_once()
def test_wrtie_to_solved_board_to_csv(self, mocked_out): with patch('sudoku_solver.sudoku_solver.open', create=True) as mock_open: mock_open.return_value = MagicMock(spec=file) mocked_out.return_value = MagicMock() mock_solved_puzzle = "mock_solved_puzzle.csv" write_to_solved_board_to_csv( [[1, 2, 3]], mock_open) mock_open.assert_called_once_with(mock_open, 'w') mocked_out.assert_called_once() mocked_out.return_value.writerows.assert_called_once_with([[1, 2, 3]])
def test_service_metadata_loaded_with_default_ecs_cluster(self, mock_open): mock_file = MagicMock(spec=TextIOWrapper) expected_config = { 'TEAM': 'dummy-team', 'TYPE': 'docker', 'REGION': 'eu-west-1', 'ACCOUNT_PREFIX': 'mmg' } mock_file.read.return_value = json.dumps(expected_config) mock_open.return_value.__enter__.return_value = mock_file metadata = config.load_service_metadata() assert metadata.team == expected_config['TEAM'] assert metadata.type == expected_config['TYPE'] assert metadata.aws_region == expected_config['REGION'] assert metadata.account_prefix == expected_config['ACCOUNT_PREFIX'] assert metadata.ecs_cluster == 'default' mock_open.assert_called_once_with('service.json')
def test_writes_to_version_file(self, mock_get_version, mock_open): increment_version.increment_version('major') mock_open.assert_called_once_with('version.py', 'wb')
def test_open_right_file(self, mock_open): File(TEST_FILE_PATH) mock_open.assert_called_once_with(TEST_FILE_PATH, "r")