py/tools/chrome_debugger_unittest.py

#!/usr/bin/env python
# Copyright 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import print_function

import json
import logging
import StringIO
import unittest
import urllib2

import mox
from ws4py.client.threadedclient import WebSocketClient

import factory_common  # pylint: disable=W0611
from cros.factory.tools import chrome_debugger


class ChromeRemoteDebuggerTest(unittest.TestCase):

  def setUp(self):
    self.chrome = chrome_debugger.ChromeRemoteDebugger()
    self.mox = mox.Mox()
    self.mox.StubOutWithMock(urllib2, "urlopen")
    self.mox.StubOutWithMock(chrome_debugger, "WebSocketClient")
    self.mock_pageset_url = chrome_debugger.DEFAULT_CHROME_DEBUG_URL + "/json"
    self.mock_pageset = [{
        "id": "6CA5278B-EEC7-48F7-BBE2-297C6DEFB59A",
        "title": "FactoryTestExtension",
        "type": "background_page",
        "url": "chrome-extension://pngocaclmlmihmhokaeejfiklacihcmb/....html",
        "webSocketDebuggerUrl": "ws://6CA5278B-EEC7-48F7-BBE2-297C6DEFB59A"
    }, {
        "id": "79589549-AD32-4AC8-B4EE-6B06EFF2D7CF",
        "title": "Blank page",
        "type": "other",
        "url": "about:blank",
        "webSocketDebuggerUrl": "ws://79589549-AD32-4AC8-B4EE-6B06EFF2D7CF",
    }, {
        "id": "7711AF2B-B1CF-40C4-B269-0D29A72BEBBC",
        "title": "Why Should I Care What Color the Bikeshed Is?",
        "type": "other",
        "url": "http://cyan.bikeshed.com/",
        "webSocketDebuggerUrl": "ws://7711AF2B-B1CF-40C4-B269-0D29A72BEBBC"
    }]
    self.mock_pageset_stream = StringIO.StringIO(json.dumps(self.mock_pageset))
    self.mock_websocket = self.mox.CreateMock(WebSocketClient)

  def tearDown(self):
    self.mox.UnsetStubs()

  def testIsReady(self):
    urllib2.urlopen(self.mock_pageset_url).AndRaise(
        urllib2.URLError("Cannot connect"))
    urllib2.urlopen(self.mock_pageset_url).AndReturn(self.mock_pageset_stream)
    self.mox.ReplayAll()
    self.assertFalse(self.chrome.IsReady())
    self.assertTrue(self.chrome.IsReady())
    self.mox.VerifyAll()

  def testGetPages(self):
    urllib2.urlopen(self.mock_pageset_url).AndReturn(self.mock_pageset_stream)
    urllib2.urlopen(self.mock_pageset_url).AndReturn(self.mock_pageset_stream)
    urllib2.urlopen(self.mock_pageset_url).AndReturn(self.mock_pageset_stream)
    self.mox.ReplayAll()
    self.assertEquals(self.chrome.GetPages(), self.mock_pageset)
    self.mock_pageset_stream.seek(0)
    self.assertEquals(self.chrome.GetPages("background_page"),
                      [self.mock_pageset[0]])
    self.mock_pageset_stream.seek(0)
    self.assertEquals(self.chrome.GetPages("no-such-page"), [])
    self.mox.VerifyAll()

  def testSetActivePage(self):
    self.mox.StubOutWithMock(self.chrome, "GetPages")
    self.chrome.GetPages("other").AndReturn(self.mock_pageset[1:])
    chrome_debugger.WebSocketClient(
        self.mock_pageset[1]["webSocketDebuggerUrl"]).AndReturn(
            self.mock_websocket)
    self.mock_websocket.connect()
    self.mock_websocket.close()
    self.mox.ReplayAll()

    self.chrome.SetActivePage()
    self.chrome.SetActivePage(None)
    self.mox.VerifyAll()

  def testSendCommand(self):
    command = {"method": "test", "params": {"param1": "value1"}}
    expected = command.copy()
    expected.update({"id": 1})
    self.chrome.active_websocket = self.mock_websocket
    self.chrome.active_websocket.send(json.dumps(expected))
    self.mox.ReplayAll()

    self.assertEquals(1, self.chrome.id)
    self.chrome.SendCommand(command)
    self.assertEquals(2, self.chrome.id)
    self.chrome.active_websocket = None
    self.mox.VerifyAll()

  def testPageNavigate(self):
    url = "http://blah"
    expected = {"method": "Page.navigate", "params": {"url": url}}
    expected.update({"id": 1})
    self.chrome.active_websocket = self.mock_websocket
    self.chrome.active_websocket.send(json.dumps(expected))
    self.mox.ReplayAll()

    self.chrome.PageNavigate(url)
    self.chrome.active_websocket = None
    self.mox.VerifyAll()


if __name__ == "__main__":
  logging.basicConfig(level=logging.INFO)
  unittest.main()