Index: gm/rebaseline_server/results_test.py |
diff --git a/gm/rebaseline_server/results_test.py b/gm/rebaseline_server/results_test.py |
index 958222c5166dd714c887491ec52044cbd45bebb1..7cbddefe23e01b91c4f2980b736fe84ec6b0e409 100755 |
--- a/gm/rebaseline_server/results_test.py |
+++ b/gm/rebaseline_server/results_test.py |
@@ -35,11 +35,18 @@ class ResultsTest(base_unittest.TestCase): |
actuals_root=os.path.join(self._input_dir, 'gm-actuals'), |
expected_root=os.path.join(self._input_dir, 'gm-expectations'), |
generated_images_root=self._temp_dir) |
+ results_obj.get_timestamp = mock_get_timestamp |
gm_json.WriteToFile( |
- results_obj.get_results_of_type(results.KEY__HEADER__RESULTS_ALL), |
+ results_obj.get_packaged_results_of_type( |
+ results.KEY__HEADER__RESULTS_ALL), |
os.path.join(self._output_dir_actual, 'gm.json')) |
+def mock_get_timestamp(): |
+ """Mock version of Results.get_timestamp() for testing.""" |
+ return 12345678 |
+ |
+ |
def main(): |
base_unittest.main(ResultsTest) |