def test_add_section_no_overwrite(ConfigParser: MagicMock, open: MagicMock, safe_print: MagicMock): parser = MagicMock() ConfigParser.return_value = parser parser.has_section.return_value = True aws_files.add_section('section-name', {'key': 'value', 'key2': 'value2'}, 'file-name', overwrite=False) parser.read.assert_called_once_with('file-name') parser.remove_section.assert_not_called() parser.add_section.assert_not_called() parser.set.assert_not_called()
def test_add_section_new_section(ConfigParser: MagicMock, open: MagicMock, safe_print: MagicMock): parser = MagicMock() ConfigParser.return_value = parser parser.has_section.return_value = False aws_files.add_section('section-name', {'key': 'value', 'key2': 'value2'}, 'file-name') parser.read.assert_called_once_with('file-name') parser.remove_section.assert_not_called() parser.add_section.assert_called_once_with('section-name') assert parser.set.call_count == 2 parser.write.assert_called_once() open.assert_called_once()
def test_add_section(ConfigParser: MagicMock, open: MagicMock): parser = MagicMock() ConfigParser.return_value = parser parser.has_section.return_value = True aws_files.add_section('section-name', { 'key': 'value', 'key2': 'value2' }, 'file-name', overwrite=True) parser.read.assert_called_once_with('file-name') parser.remove_section.assert_called_once_with('section-name') parser.add_section.assert_called_once_with('section-name') assert parser.set.call_count == 3 parser.set.assert_any_call('section-name', 'manager', 'awsume') parser.write.assert_called_once() open.assert_called_once()