def test_comments_stipped(self):
sa = SettingsAuditor('')
test_data = (
'#[foo]\n'
'#public.foo = SELECT\n')
data = sa._strip(test_data)
self.assertEqual('', data)
def test_extract_config_blocks(self):
test_settings = self.test_settings.replace(
'# This is the header.\n', '')
sa = SettingsAuditor(test_settings)
sa._separateConfigBlocks()
self.assertContentEqual(
['[good]', '[bad]'],
sa.config_blocks.keys())
def test_audit(self):
sa = SettingsAuditor(self.test_settings)
new_settings = sa.audit()
expected_settings = ('# This is the header.\n'
'[good]\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT\n'
'\n'
'[bad]\n'
'public.bar = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT')
self.assertEqual(expected_settings, new_settings)
def test_audit(self):
sa = SettingsAuditor(self.test_settings)
new_settings = sa.audit()
expected_settings = (
'# This is the header.\n'
'[good]\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT\n'
'\n'
'[bad]\n'
'public.bar = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT')
self.assertEqual(expected_settings, new_settings)
def test_audit_block(self):
sa = SettingsAuditor('')
test_block = (
'[bad]\n'
'public.foo = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.bar = SELECT\n'
'public.baz = SELECT\n')
sa.config_blocks = {'[bad]': test_block}
sa.config_labels = ['[bad]']
sa._processBlocks()
expected = (
'[bad]\n'
'public.bar = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT')
self.assertEqual(expected, sa.config_blocks['[bad]'])
expected_error = '[bad]\n\tDuplicate setting found: public.bar'
self.assertTrue(expected_error in sa.error_data)
def test_audit_block(self):
sa = SettingsAuditor('')
test_block = ('[bad]\n'
'public.foo = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.bar = SELECT\n'
'public.baz = SELECT\n')
sa.config_blocks = {'[bad]': test_block}
sa.config_labels = ['[bad]']
sa._processBlocks()
expected = ('[bad]\n'
'public.bar = SELECT\n'
'public.bar = SELECT, INSERT\n'
'public.baz = SELECT\n'
'public.foo = SELECT')
self.assertEqual(expected, sa.config_blocks['[bad]'])
expected_error = '[bad]\n\tDuplicate setting found: public.bar'
self.assertTrue(expected_error in sa.error_data)
def test_getHeader(self):
sa = SettingsAuditor(self.test_settings)
header = sa._getHeader()
self.assertEqual(
header,
'# This is the header.\n')
def main():
data = file(SECURITY_PATH).read()
auditor = SettingsAuditor(data)
settings = auditor.audit()
file(SECURITY_PATH, 'w').write(settings)
print auditor.error_data
def main():
data = file(SECURITY_PATH).read()
auditor = SettingsAuditor(data)
settings = auditor.audit()
file(SECURITY_PATH, 'w').write(settings)
print auditor.error_data
def test_comments_stipped(self):
sa = SettingsAuditor('')
test_data = ('#[foo]\n' '#public.foo = SELECT\n')
data = sa._strip(test_data)
self.assertEqual('', data)
def test_extract_config_blocks(self):
test_settings = self.test_settings.replace('# This is the header.\n',
'')
sa = SettingsAuditor(test_settings)
sa._separateConfigBlocks()
self.assertContentEqual(['[good]', '[bad]'], sa.config_blocks.keys())
def test_getHeader(self):
sa = SettingsAuditor(self.test_settings)
header = sa._getHeader()
self.assertEqual(header, '# This is the header.\n')
