def test_add_entry_twice(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 0)
s.add_entry('md5', 'afile', 'ahash')
self.assertEqual(len(s.entries), 1)
self.assertEqual(str(s.entries[-1]), 'MD5 (afile) = ahash\n')
s.add_entry('md5', 'afile', 'ahash')
self.assertEqual(len(s.entries), 1)
def test_parse_entry_line(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
line = 'MD5 (afile) = ahash\n'
entry = s.parse_line(line)
self.assertTrue(isinstance(entry, sources.SourceFileEntry))
self.assertEqual(entry.hashtype, 'md5')
self.assertEqual(entry.hash, 'ahash')
self.assertEqual(entry.file, 'afile')
self.assertEqual(str(entry), line)
def test_parse_old_style_line(self):
s = sources.SourcesFile(self.sourcesfile, 'old')
line = 'ahash afile\n'
entry = s.parse_line(line)
self.assertTrue(isinstance(entry, sources.SourceFileEntry))
self.assertEqual(entry.hashtype, 'md5')
self.assertEqual(entry.hash, 'ahash')
self.assertEqual(entry.file, 'afile')
self.assertEqual(str(entry), line)
def test_add_entry_mixing_hashtypes(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 0)
s.add_entry('md5', 'afile', 'ahash')
self.assertEqual(len(s.entries), 1)
self.assertEqual(str(s.entries[-1]), 'MD5 (afile) = ahash\n')
def raises():
s.add_entry('sha512', 'anotherfile', 'anotherhash')
self.assertRaises(sources.HashtypeMixingError, raises)
def test_open_existing_file(self):
lines = ['MD5 (afile) = ahash\n', 'MD5 (anotherfile) = anotherhash\n']
with open(self.sourcesfile, 'w') as f:
for line in lines:
f.write(line)
s = sources.SourcesFile(self.sourcesfile, 'bsd')
for i, entry in enumerate(s.entries):
self.assertTrue(isinstance(entry, sources.SourceFileEntry))
self.assertEqual(str(entry), lines[i])
def test_write_new_file(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 0)
s.add_entry('md5', 'afile', 'ahash')
s.add_entry('md5', 'anotherfile', 'anotherhash')
s.write()
with open(self.sourcesfile) as f:
lines = f.readlines()
self.assertEqual(len(lines), 2)
self.assertEqual(lines[0], 'MD5 (afile) = ahash\n')
self.assertEqual(lines[1], 'MD5 (anotherfile) = anotherhash\n')
def test_write_over(self):
lines = ['ahash afile\n', 'anotherhash anotherfile\n']
with open(self.sourcesfile, 'w') as f:
for line in lines:
f.write(line)
s = sources.SourcesFile(self.sourcesfile, 'bsd', replace=True)
s.add_entry('md5', 'thirdfile', 'thirdhash')
s.write()
with open(self.sourcesfile) as f:
lines = f.readlines()
self.assertEqual(len(lines), 1)
self.assertEqual(lines[0], 'MD5 (thirdfile) = thirdhash\n')
def test_parse_wrong_lines(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
lines = [
'ahash',
'ahash ',
'ahash afile',
'SHA512 (afile) = ahash garbage',
'MD5 SHA512 (afile) = ahash',
]
for line in lines:
def raises():
s.parse_line(line)
self.assertRaises(sources.MalformedLineError, raises)
def test_open_existing_file_with_identical_entries_old_and_new(self):
lines = [
'ahash afile\n',
'MD5 (afile) = ahash\n',
]
with open(self.sourcesfile, 'w') as f:
for line in lines:
f.write(line)
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 1)
self.assertEqual(s.entries[0].hashtype, 'md5')
self.assertEqual(s.entries[0].file, 'afile')
self.assertEqual(s.entries[0].hash, 'ahash')
self.assertEqual(str(s.entries[0]), lines[-1])
def test_open_existing_file_with_mixed_lines(self):
lines = [
'ahash afile\n',
'anotherhash anotherfile\n',
'MD5 (thirdfile) = thirdhash\n',
]
expected = [
'MD5 (afile) = ahash\n',
'MD5 (anotherfile) = anotherhash\n',
'MD5 (thirdfile) = thirdhash\n',
]
with open(self.sourcesfile, 'w') as f:
for line in lines:
f.write(line)
s = sources.SourcesFile(self.sourcesfile, 'bsd')
for i, entry in enumerate(s.entries):
self.assertTrue(isinstance(entry, sources.SourceFileEntry))
self.assertEqual(str(entry), expected[i])
def test_open_new_file(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 0)
def test_parse_whitespace_line(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
entry = s.parse_line(' \n')
self.assertTrue(entry is None)
def test_parse_empty_line(self):
s = sources.SourcesFile(self.sourcesfile, 'bsd')
entry = s.parse_line('')
self.assertTrue(entry is None)
def raises():
sources.SourcesFile(self.sourcesfile, 'bsd')
def test_open_empty_file(self):
with open(self.sourcesfile, 'w') as f:
f.write('')
s = sources.SourcesFile(self.sourcesfile, 'bsd')
self.assertEqual(len(s.entries), 0)
