Synced repos to: 58208
diff --git a/crosperf/experiment_factory_unittest.py b/crosperf/experiment_factory_unittest.py
new file mode 100755
index 0000000..e91295d
--- /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()