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)