mirror of
https://github.com/CCOSTAN/Home-AssistantConfig.git
synced 2025-11-06 09:45:07 +00:00
156 lines
4.7 KiB
Python
156 lines
4.7 KiB
Python
import unittest
|
|
from .core import Roku, COMMANDS
|
|
|
|
TEST_APPS = (
|
|
('11', '1.0.1', 'Fake Roku Channel Store'),
|
|
('22', '2.0.2', 'Fake Netflix'),
|
|
('33', '3.0.3', 'Fake YouTube'),
|
|
)
|
|
|
|
TEST_DEV_INFO = ''.join([
|
|
'<?xml version="1.0" encoding="UTF-8" ?>',
|
|
'<device-info>',
|
|
' <udn>00000000-1111-2222-3333-444444444444</udn>',
|
|
' <serial-number>111111111111</serial-number>',
|
|
' <device-id>222222222222</device-id>',
|
|
' <vendor-name>Roku</vendor-name>',
|
|
' <model-number>4200X</model-number>',
|
|
' <model-name>Roku 3</model-name>',
|
|
' <wifi-mac>00:11:22:33:44:55</wifi-mac>',
|
|
' <ethernet-mac>00:11:22:33:44:56</ethernet-mac>',
|
|
' <network-type>ethernet</network-type>',
|
|
' <user-device-name/>',
|
|
' <software-version>7.00</software-version>',
|
|
' <software-build>09044</software-build>',
|
|
' <secure-device>true</secure-device>',
|
|
' <language>en</language>',
|
|
' <country>US</country>',
|
|
' <locale>en_US</locale>',
|
|
' <time-zone>US/Eastern</time-zone>',
|
|
' <time-zone-offset>-300</time-zone-offset>',
|
|
' <power-mode>PowerOn</power-mode>',
|
|
' <developer-enabled>false</developer-enabled>',
|
|
' <search-enabled>true</search-enabled>',
|
|
' <voice-search-enabled>true</voice-search-enabled>',
|
|
' <notifications-enabled>true</notifications-enabled>',
|
|
' <notifications-first-use>false</notifications-first-use>',
|
|
' <headphones-connected>false</headphones-connected>',
|
|
'</device-info>'
|
|
])
|
|
|
|
|
|
class TestRoku(Roku):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(TestRoku, self).__init__(*args, **kwargs)
|
|
self._calls = []
|
|
|
|
def _call(self, method, path, *args, **kwargs):
|
|
|
|
resp = None
|
|
|
|
if path == '/query/apps':
|
|
fmt = '<app id="%s" version="%s">%s</app>'
|
|
resp = '<apps>%s</apps>' % "".join(fmt % a for a in TEST_APPS)
|
|
elif path == '/query/device-info':
|
|
resp = TEST_DEV_INFO
|
|
|
|
self._calls.append((method, path, args, kwargs, resp))
|
|
|
|
return resp
|
|
|
|
def calls(self):
|
|
return self._calls
|
|
|
|
def last_call(self):
|
|
return self._calls[-1]
|
|
|
|
|
|
class RokuTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.roku = TestRoku('0.0.0.0')
|
|
|
|
def testApps(self):
|
|
|
|
apps = self.roku.apps
|
|
self.assertEqual(len(apps), 3)
|
|
|
|
for i, app in enumerate(apps):
|
|
self.assertEqual(app.id, TEST_APPS[i][0])
|
|
self.assertEqual(app.version, TEST_APPS[i][1])
|
|
self.assertEqual(app.name, TEST_APPS[i][2])
|
|
|
|
app = self.roku['22']
|
|
self.assertEqual(app.id, TEST_APPS[1][0])
|
|
self.assertEqual(app.version, TEST_APPS[1][1])
|
|
self.assertEqual(app.name, TEST_APPS[1][2])
|
|
|
|
app = self.roku['Fake YouTube']
|
|
self.assertEqual(app.id, TEST_APPS[2][0])
|
|
self.assertEqual(app.version, TEST_APPS[2][1])
|
|
self.assertEqual(app.name, TEST_APPS[2][2])
|
|
|
|
def testDeviceInfo(self):
|
|
|
|
d = self.roku.device_info
|
|
|
|
self.assertEqual(d.modelname, 'Roku 3')
|
|
self.assertEqual(d.modelnum, '4200X')
|
|
self.assertEqual(d.swversion, '7.00.09044')
|
|
self.assertEqual(d.sernum, '111111111111')
|
|
|
|
def testCommands(self):
|
|
|
|
for cmd in self.roku.commands:
|
|
|
|
if cmd == 'literal':
|
|
continue
|
|
|
|
getattr(self.roku, cmd)()
|
|
call = self.roku.last_call()
|
|
|
|
self.assertEqual(call[0], 'POST')
|
|
self.assertEqual(call[1], '/keypress/%s' % COMMANDS[cmd])
|
|
self.assertEqual(call[2], ())
|
|
self.assertEqual(call[3], {})
|
|
|
|
def testLiteral(self):
|
|
|
|
text = 'Stargate'
|
|
self.roku.literal(text)
|
|
|
|
for i, call in enumerate(self.roku.calls()):
|
|
self.assertEqual(call[0], 'POST')
|
|
self.assertEqual(call[1], '/keypress/Lit_%s' % text[i].upper())
|
|
self.assertEqual(call[2], ())
|
|
self.assertEqual(call[3], {})
|
|
|
|
def testStore(self):
|
|
|
|
for app in self.roku.apps:
|
|
|
|
self.roku.store(app)
|
|
call = self.roku.last_call()
|
|
|
|
self.assertEqual(call[0], 'POST')
|
|
self.assertEqual(call[1], '/launch/11')
|
|
self.assertEqual(call[2], ())
|
|
self.assertEqual(call[3], {'params': {'contentID': app.id}})
|
|
|
|
def testLaunch(self):
|
|
|
|
for app in self.roku.apps:
|
|
|
|
self.roku.launch(app)
|
|
call = self.roku.last_call()
|
|
|
|
self.assertEqual(call[0], 'POST')
|
|
self.assertEqual(call[1], '/launch/%s' % app.id)
|
|
self.assertEqual(call[2], ())
|
|
self.assertEqual(call[3], {'params': {'contentID': app.id}})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|