def test_filled_repository_produces_correct_output(self):
repo_ss = self.repositories[self.repo_names[0]]
repo_wal = self.repositories[self.repo_names[1]]
self.fill_repositories_with_commits()
self.num_snapshots = repo_ss.get_number_of_items()
self.size_snapshots = repo_ss.get_repository_size()
self.num_walfiles = repo_wal.get_number_of_items()
self.size_walfiles = repo_wal.get_repository_size()
total = self.size_snapshots + self.size_walfiles
self.snapshots = self.item.format(
self.repo_names[0],
'%s' % self.num_snapshots,
'%s' % (self.size_snapshots / self.ONE_MEBIBYTE)
)
self.walfiles = self.item.format(
self.repo_names[1],
'%s' % self.num_walfiles,
'%s' % (self.size_walfiles / self.ONE_MEBIBYTE)
)
self.total = self.total_templ.format(
'Total Size',
'',
'%s' % (total / self.ONE_MEBIBYTE)
)
all = [c for c in repo_ss]
self.first = self.firstlast_templ.format(
'First Snapshot:',
all[0].tag
)
self.last = self.firstlast_templ.format(
'Last Snapshot:',
all[-1].tag
)
self.expected_output_lines = [
self.topbottom_dashes,
self.column_headers,
self.middle_dashes,
self.snapshots,
self.walfiles,
self.middle_dashes,
self.total,
self.middle_dashes,
self.first,
self.last,
self.topbottom_dashes,
]
self.expected_output = ''.join(self.expected_output_lines)
rss = Storage_stats_reporter(self.repo_names, self.repositories)
myout = StringIO.StringIO()
rss.write_report(myout)
output = myout.getvalue()
print(output)
myout.close()
stdout.write(self.expected_output)
stdout.write(output)
self.assertEqual(self.expected_output, output)
def test_write_report_empty_repository(self):
rss = Storage_stats_reporter(self.repo_names, self.repositories)
myout = StringIO.StringIO()
rss.write_report(myout)
output = myout.getvalue()
print(output)
#assert False
myout.close()
stdout.write(self.expected_output)
stdout.write(output)
self.assertEqual(self.expected_output, output)
