def test_get_snapshots_size_empty_repository(self):
rss = Storage_stats_reporter(self.repo_names, self.repositories)
(items, size) = rss._get_repository_size(
self.repositories[self.repo_names[0]]
)
self.assertEqual(0, items)
self.assertEqual(0, size)
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)
def test_get_first_and_last_snapshot(self):
repo_ss = self.repositories[self.repo_names[0]]
self.fill_repositories_with_commits()
rss = Storage_stats_reporter(self.repo_names, self.repositories)
all_comm = [c for c in repo_ss]
first_comm = all_comm[0].tag
last_comm = all_comm[-1].tag
first, last = rss._get_first_and_last_commit_tags(repo_ss)
print(first_comm, first)
print(last_comm, last)
self.assertEqual(first_comm, first)
self.assertEqual(last_comm, last)
def test_get_repository_size_with_items(self):
repo_ss = self.repositories[self.repo_names[0]]
repo_wal = self.repositories[self.repo_names[1]]
self.fill_repositories_with_commits()
rss = Storage_stats_reporter(self.repo_names, self.repositories)
num_items, size_items = rss._get_repository_size(repo_ss)
self.assertEqual(
repo_ss.get_number_of_items(),
num_items
)
self.assertEqual(
repo_ss.get_repository_size(),
size_items
)
num_items, size_items = rss._get_repository_size(repo_wal)
self.assertEqual(
repo_wal.get_number_of_items(),
num_items
)
self.assertEqual(
repo_wal.get_repository_size(),
size_items
)
