test_google_assistant_mobile.py

#!/usr/bin/python

import time

from device.device import get_volume
from helper.basetest import BaseTest

GASSISTANT_PKG = 'com.google.android.googlequicksearchbox'
TARGET_MUSIC_PKG = 'com.miui.player'
TIMEOUT = 10


class TestGoogleAssistant(BaseTest):

    # Format comes from http://appium.io/docs/en/writing-running-appium/caps/
    LANG = 'en'
    LOCALE = 'US'

    @classmethod
    def setup_class(cls):
        BaseTest.pre_proc(cls.LANG, cls.LOCALE)

    def test_whats_the_weather(self):
        self._init_ok_google()

        # ***Ask weather
        self._say("what's the weather tomorrow")

        time.sleep(5)

    def test_volume_change(self):
        self._init_ok_google()

        # ***Ask "volume zero"
        self._say("volume zero", check_text=False)

        for _ in range(TIMEOUT):  # TODO Make better
            time.sleep(1)
            if get_volume(self.driver) == 0:
                break
        volume = get_volume(self.driver)
        assert volume == 0

        # ***Exit google assistant
        self.driver.back()

        # ***Ask "ok google"
        self._init_ok_google()

        # ***Ask "volume up"
        self._say("volume up", check_text=False)

        for _ in range(TIMEOUT):  # TODO Make better
            time.sleep(1)
            if get_volume(self.driver) > volume:
                break
        assert get_volume(self.driver) > volume

    def test_open_music(self):
        self._init_ok_google()

        # ***Ask weather
        self.voice.say("Launch music", self.LANG)

        for _ in range(TIMEOUT):  # TODO Make better
            time.sleep(1)
            if self.driver.current_package == TARGET_MUSIC_PKG:
                break
        assert self.driver.current_package == TARGET_MUSIC_PKG