示例#1
0
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()
示例#2
0
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()
示例#3
0
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()