def get_stats(self):
rp = ResultsPackage()
for dt in self.results[0].get_dates():
values = [input_rp.get_result(dt) for input_rp in self.results]
rp.add_result(dt, self.stat.get_stat(values))
return rp
def get_results_package(self, results_file):
rp = ResultsPackage()
jsontext = ''
with file(results_file, 'r') as f:
jsontext = f.read()
rp.add_results_from_json(jsontext)
return rp
def get_stats(self):
di = DateIterator(self.start, self.end, self.timedelta)
ri = RepositoryIterator(self.repo, di)
results = ResultsPackage()
for date in ri:
self.stat.set_files(self.files)
stat_result = self.stat.get_stat()
results.add_result(date, stat_result)
# self.print_one_day_results(date, stat_results)
return results
def setUp(self):
self.rp = ResultsPackage()
class ResultsPackageTests(TestCase):
def setUp(self):
self.rp = ResultsPackage()
def test_result_package_holds_single_stat(self):
dt = datetime(2011,03,18,18,44,0)
self.rp.add_result(dt, 123)
eq_(123, self.rp.get_result(dt))
def test_result_package_overwrites_multiple_stats_one_time(self):
dt = datetime(2011, 03, 18, 18, 52, 0)
self.rp.add_result(dt, 1234)
self.rp.add_result(dt, 2345)
eq_(2345, self.rp.get_result(dt))
def test_result_package_holds_multiple_stats_diff_time(self):
dt1 = datetime(2011, 03, 18, 18, 52, 0)
self.rp.add_result(dt1, 1234)
dt2 = datetime(2011, 03, 18, 18, 54, 0)
self.rp.add_result(dt2, 2345)
eq_(1234, self.rp.get_result(dt1))
eq_(2345, self.rp.get_result(dt2))
def test_result_package_returns_number_of_dates_stored(self):
dt1 = datetime(2011, 03, 18, 18, 52, 0)
self.rp.add_result(dt1, 1234)
eq_(1, self.rp.get_date_count())
dt2 = datetime(2011, 03, 18, 18, 54, 0)
self.rp.add_result(dt2, 2345)
eq_(2, self.rp.get_date_count())
def test_result_package_simple_json_output(self):
dt = datetime(2011, 03, 18, 19, 10, 0)
self.rp.add_result(dt, 1234)
expected='{\n "stats": {\n "1300489800000": 1234\n }\n}'
eq_(expected, self.rp.get_json())
def test_result_package_gives_list_of_dates(self):
dt1 = datetime(2011, 04, 1, 19, 34, 0)
self.rp.add_result(dt1, 1234)
dt2 = datetime(2011, 04, 1, 19, 35, 0)
self.rp.add_result(dt2, 2345)
eq_(set(self.rp.get_dates()), set([dt1, dt2]))
def test_result_package_writes_json_to_outfile(self):
directory = mkdtemp('-caboose-results-package-tests')
filename = path.join(directory, 'outfile')
dt = datetime(2011, 03, 18, 19, 10, 0)
self.rp.add_result(dt, 1234)
self.rp.write_json_results(filename)
with file(filename, "r") as f:
written = f.read()
expected='{\n "stats": {\n "1300489800000": 1234\n }\n}'
eq_(expected, written)
rmtree(directory)
def test_convert_from_js_time_to_datetime(self):
jstime = "1300489800000"
expected = datetime(2011, 3, 18, 19, 10, 0)
eq_(expected, self.rp._get_datetime_from_javascript_date(jstime))
def test_create_results_package_by_parsing_json(self):
inputjson = '{\n "stats": {\n "1300489800000": 1234\n }\n}'
self.rp.add_results_from_json(inputjson)
dates = self.rp.get_dates()
eq_(1, len(dates))
eq_(datetime(2011, 03, 18, 19, 10, 0), dates[0])
eq_(1234, self.rp.get_result(dates[0]))
def test_results_package_writes_out_description(self):
desc = "description goes here"
self.rp.set_description(desc)
dt = datetime(2011, 03, 18, 19, 10, 0)
self.rp.add_result(dt, 1234)
expected='{\n "stats": {\n "1300489800000": 1234\n }, \n "description": "description goes here"\n}'
eq_(expected, self.rp.get_json())
def test_results_package_writes_out_datatype(self):
datatype = "percentage"
self.rp.set_datatype(datatype)
dt = datetime(2011, 03, 18, 19, 10, 0)
self.rp.add_result(dt, 1234)
expected='{\n "datatype": "percentage", \n "stats": {\n "1300489800000": 1234\n }\n}'
eq_(expected, self.rp.get_json())