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)