def test_get_stat(self): stat = MockResultsStat() rsc = ResultsStatCollector(stat) rp1 = MockResultsPackage() rp2 = MockResultsPackage() rsc.set_results([rp1, rp2]) stats = rsc.get_stats() eq_([1, 2, 3], stats.get_dates()) eq_(4, stats.get_result(1)) eq_(8, stats.get_result(2)) eq_(12, stats.get_result(3))
def test_create_result_packages_from_files(self): #write out a json file directory = mkdtemp('-caboose-results-stat-collector-tests') filename = path.join(directory, 'results.json') with file(filename, 'w') as f: f.write('{\n "stats": {\n "1300489800000": 1234\n }\n}') stat = MockResultsStat() rsc = ResultsStatCollector(stat) rsc.set_results_files([filename]) eq_(1, len(rsc.get_results())) eq_(1, rsc.get_results()[0].get_date_count()) rmtree(directory)