| Index: dashboard/dashboard/pinpoint/models/quest/read_value_test.py
|
| diff --git a/dashboard/dashboard/pinpoint/models/quest/read_value_test.py b/dashboard/dashboard/pinpoint/models/quest/read_value_test.py
|
| index 972eef56e7da7edd20406de44dc4295eee929a58..b0cf86b3679114683ba5252c685f9d359b395876 100644
|
| --- a/dashboard/dashboard/pinpoint/models/quest/read_value_test.py
|
| +++ b/dashboard/dashboard/pinpoint/models/quest/read_value_test.py
|
| @@ -105,3 +105,25 @@ class ReadChartJsonValueTest(unittest.TestCase):
|
| execution.Poll()
|
|
|
| self.assertEqual(execution.result_values, (2.5,))
|
| +
|
| +
|
| +@mock.patch('dashboard.services.isolate_service.Retrieve')
|
| +class ReadGraphJsonValueTest(unittest.TestCase):
|
| +
|
| + def testReadGraphJsonValue(self, retrieve):
|
| + retrieve.side_effect = (
|
| + {'files': {'chartjson-output.json': {'h': 'graphjson hash'}}},
|
| + json.dumps({'chart': {'traces': {'trace': ['126444.869721', '0.0']}}}),
|
| + )
|
| +
|
| + quest = read_value.ReadGraphJsonValue('chart', 'trace')
|
| + execution = quest.Start('output hash')
|
| + execution.Poll()
|
| +
|
| + self.assertTrue(execution.completed)
|
| + self.assertFalse(execution.failed)
|
| + self.assertEqual(execution.result_values, (126444.869721,))
|
| + self.assertEqual(execution.result_arguments, {})
|
| +
|
| + expected_calls = [mock.call('output hash'), mock.call('graphjson hash')]
|
| + self.assertEqual(retrieve.mock_calls, expected_calls)
|
|
|