Synced repos to: 58208

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()