| Index: tools/telemetry/telemetry/internal/backends/chrome_inspector/memory_backend_unittest.py
|
| diff --git a/tools/telemetry/telemetry/internal/backends/chrome_inspector/memory_backend_unittest.py b/tools/telemetry/telemetry/internal/backends/chrome_inspector/memory_backend_unittest.py
|
| deleted file mode 100644
|
| index 5829d1d00e5d3d75f801316c599efdd153241f30..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/telemetry/internal/backends/chrome_inspector/memory_backend_unittest.py
|
| +++ /dev/null
|
| @@ -1,152 +0,0 @@
|
| -# Copyright 2015 The Chromium Authors. All rights reserved.
|
| -# Use of this source code is governed by a BSD-style license that can be
|
| -# found in the LICENSE file.
|
| -
|
| -import mock
|
| -import unittest
|
| -
|
| -
|
| -from telemetry.internal.backends.chrome_inspector import inspector_websocket
|
| -from telemetry.internal.backends.chrome_inspector import memory_backend
|
| -from telemetry.testing import fakes
|
| -from telemetry.testing import simple_mock
|
| -from telemetry.testing import tab_test_case
|
| -
|
| -
|
| -class MemoryBackendTest(tab_test_case.TabTestCase):
|
| -
|
| - def setUp(self):
|
| - super(MemoryBackendTest, self).setUp()
|
| - if not self._browser.supports_overriding_memory_pressure_notifications:
|
| - self.skipTest('Browser does not support overriding memory pressure '
|
| - 'notification signals, skipping test.')
|
| -
|
| - def testSetMemoryPressureNotificationsSuppressed(self):
|
| - def PerformCheck(suppressed):
|
| - # Check that the method sends the correct DevTools request.
|
| - with mock.patch.object(inspector_websocket.InspectorWebsocket,
|
| - 'SyncRequest') as mock_method:
|
| - self._browser.SetMemoryPressureNotificationsSuppressed(suppressed)
|
| - self.assertEqual(1, mock_method.call_count)
|
| - request = mock_method.call_args[0][0]
|
| - self.assertEqual('Memory.setPressureNotificationsSuppressed',
|
| - request['method'])
|
| - self.assertEqual(suppressed, request['params']['suppressed'])
|
| -
|
| - # Check that the request and the response from the browser are handled
|
| - # properly.
|
| - self._browser.SetMemoryPressureNotificationsSuppressed(suppressed)
|
| -
|
| - PerformCheck(True)
|
| - PerformCheck(False)
|
| -
|
| - def testSimulateMemoryPressureNotification(self):
|
| - def PerformCheck(pressure_level):
|
| - # Check that the method sends the correct DevTools request.
|
| - with mock.patch.object(inspector_websocket.InspectorWebsocket,
|
| - 'SyncRequest') as mock_method:
|
| - self._browser.SimulateMemoryPressureNotification(pressure_level)
|
| - self.assertEqual(1, mock_method.call_count)
|
| - request = mock_method.call_args[0][0]
|
| - self.assertEqual('Memory.simulatePressureNotification',
|
| - request['method'])
|
| - self.assertEqual(pressure_level, request['params']['level'])
|
| -
|
| - # Check that the request and the response from the browser are handled
|
| - # properly.
|
| - self._browser.SimulateMemoryPressureNotification(pressure_level)
|
| -
|
| - PerformCheck('moderate')
|
| - PerformCheck('critical')
|
| -
|
| -
|
| -class MemoryBackendUnitTest(unittest.TestCase):
|
| -
|
| - def setUp(self):
|
| - self._mock_timer = simple_mock.MockTimer()
|
| - self._inspector_socket = fakes.FakeInspectorWebsocket(self._mock_timer)
|
| -
|
| - def tearDown(self):
|
| - self._mock_timer.Restore()
|
| -
|
| - def testSetMemoryPressureNotificationsSuppressedSuccess(self):
|
| - response_handler = mock.Mock(return_value={'result': {}})
|
| - self._inspector_socket.AddResponseHandler(
|
| - 'Memory.setPressureNotificationsSuppressed', response_handler)
|
| - backend = memory_backend.MemoryBackend(self._inspector_socket)
|
| -
|
| - backend.SetMemoryPressureNotificationsSuppressed(True)
|
| - self.assertEqual(1, response_handler.call_count)
|
| - self.assertTrue(response_handler.call_args[0][0]['params']['suppressed'])
|
| -
|
| - backend.SetMemoryPressureNotificationsSuppressed(False)
|
| - self.assertEqual(2, response_handler.call_count)
|
| - self.assertFalse(response_handler.call_args[0][0]['params']['suppressed'])
|
| -
|
| - def testSetMemoryPressureNotificationsSuppressedFailure(self):
|
| - response_handler = mock.Mock()
|
| - backend = memory_backend.MemoryBackend(self._inspector_socket)
|
| - self._inspector_socket.AddResponseHandler(
|
| - 'Memory.setPressureNotificationsSuppressed', response_handler)
|
| -
|
| - # If the DevTools method is missing, the backend should fail silently.
|
| - response_handler.return_value = {
|
| - 'result': {},
|
| - 'error': {
|
| - 'code': -32601 # Method does not exist.
|
| - }
|
| - }
|
| - backend.SetMemoryPressureNotificationsSuppressed(True)
|
| - self.assertEqual(1, response_handler.call_count)
|
| -
|
| - # All other errors should raise an exception.
|
| - response_handler.return_value = {
|
| - 'result': {},
|
| - 'error': {
|
| - 'code': -32602 # Invalid method params.
|
| - }
|
| - }
|
| - self.assertRaises(memory_backend.MemoryUnexpectedResponseException,
|
| - backend.SetMemoryPressureNotificationsSuppressed, True)
|
| -
|
| - def testSimulateMemoryPressureNotificationSuccess(self):
|
| - response_handler = mock.Mock(return_value={'result': {}})
|
| - self._inspector_socket.AddResponseHandler(
|
| - 'Memory.simulatePressureNotification', response_handler)
|
| - backend = memory_backend.MemoryBackend(self._inspector_socket)
|
| -
|
| - backend.SimulateMemoryPressureNotification('critical')
|
| - self.assertEqual(1, response_handler.call_count)
|
| - self.assertEqual('critical',
|
| - response_handler.call_args[0][0]['params']['level'])
|
| -
|
| - backend.SimulateMemoryPressureNotification('moderate')
|
| - self.assertEqual(2, response_handler.call_count)
|
| - self.assertEqual('moderate',
|
| - response_handler.call_args[0][0]['params']['level'])
|
| -
|
| - def testSimulateMemoryPressureNotificationFailure(self):
|
| - response_handler = mock.Mock()
|
| - backend = memory_backend.MemoryBackend(self._inspector_socket)
|
| - self._inspector_socket.AddResponseHandler(
|
| - 'Memory.simulatePressureNotification', response_handler)
|
| -
|
| - # If the DevTools method is missing, the backend should fail silently.
|
| - response_handler.return_value = {
|
| - 'result': {},
|
| - 'error': {
|
| - 'code': -32601 # Method does not exist.
|
| - }
|
| - }
|
| - backend.SimulateMemoryPressureNotification('critical')
|
| - self.assertEqual(1, response_handler.call_count)
|
| -
|
| - # All other errors should raise an exception.
|
| - response_handler.return_value = {
|
| - 'result': {},
|
| - 'error': {
|
| - 'code': -32602 # Invalid method params.
|
| - }
|
| - }
|
| - self.assertRaises(memory_backend.MemoryUnexpectedResponseException,
|
| - backend.SimulateMemoryPressureNotification, 'critical')
|
|
|