diff options
Diffstat (limited to 'crosperf/experiment_factory_unittest.py')
-rwxr-xr-x | crosperf/experiment_factory_unittest.py | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/crosperf/experiment_factory_unittest.py b/crosperf/experiment_factory_unittest.py new file mode 100755 index 00000000..e91295da --- /dev/null +++ b/crosperf/experiment_factory_unittest.py @@ -0,0 +1,50 @@ +#!/usr/bin/python + +# Copyright 2011 Google Inc. All Rights Reserved. + +import StringIO +import unittest +from experiment_factory import ExperimentFactory +from experiment_file import ExperimentFile +from utils.file_utils import FileUtils + + +EXPERIMENT_FILE_1 = """ + board: x86-alex + remote: chromeos-alex3 + + benchmark: PageCycler { + iterations: 3 + } + + image1 { + chromeos_image: /usr/local/google/cros_image1.bin + } + + image2 { + chromeos_image: /usr/local/google/cros_image2.bin + } + """ + + +class ExperimentFactoryTest(unittest.TestCase): + def testLoadExperimentFile1(self): + experiment_file = ExperimentFile(StringIO.StringIO(EXPERIMENT_FILE_1)) + experiment = ExperimentFactory().GetExperiment(experiment_file, "") + self.assertEqual(experiment.remote, ["chromeos-alex3"]) + + self.assertEqual(len(experiment.benchmarks), 1) + self.assertEqual(experiment.benchmarks[0].name, "PageCycler") + self.assertEqual(experiment.benchmarks[0].autotest_name, "PageCycler") + self.assertEqual(experiment.benchmarks[0].iterations, 3) + + self.assertEqual(len(experiment.labels), 2) + self.assertEqual(experiment.labels[0].chromeos_image, + "/usr/local/google/cros_image1.bin") + self.assertEqual(experiment.labels[0].board, + "x86-alex") + + +if __name__ == "__main__": + FileUtils.Configure(True) + unittest.main() |