# -*- coding:utf8 -*-

import unittest

from selenium import webdriver

class MobileTestDemo(unittest.TestCase):

    def setUp(self):

        fp = webdriver.FirefoxProfile()

        fp.set_preference("general.useragent.override", "Mozilla/5.0()\

            (iPhone; U; CPU iPhone OS 5_0_1 like Mac OS X;\

            en-us) AppleWebkit/533.17.0 (KHTML, like Gecko)\

            Version/5.0.2 Mobile/8H7 Safari/6533.18.")

        fp.update_preferences()

        cap = webdriver.DesiredCapabilities.FIREFOX

        cap['firefox_profile'] = fp.encoded

        self.driver = webdriver.Firefox(firefox_profile=fp)

    def tearDown(self):

        self.driver.close()

    def test_mobile(self):

        driver = self.driver

        driver.get("http://www.google.com")

        self.assertIn(u'Google', driver.title)

if __name__ == "__main__":

    unittest.main()