1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
import sys
import unittest
try:
# Python3
from unittest.mock import patch
except ImportError:
# Python2
from mock import patch
sys.path.insert(0, "..")
from androidtv import setup
from androidtv.androidtv.androidtv_sync import AndroidTVSync
from androidtv.firetv.firetv_sync import FireTVSync
from . import patchers
DEVICE_PROPERTIES_OUTPUT1 = "Amazon\n\n\n123"
DEVICE_PROPERTIES_DICT1 = {
"manufacturer": "Amazon",
"model": "",
"serialno": None,
"sw_version": "123",
"wifimac": None,
"ethmac": None,
}
DEVICE_PROPERTIES_OUTPUT2 = "Not Amazon\n\n\n456"
DEVICE_PROPERTIES_DICT2 = {
"manufacturer": "Not Amazon",
"model": "",
"serialno": None,
"sw_version": "456",
"wifimac": None,
"ethmac": None,
}
class TestSetup(unittest.TestCase):
PATCH_KEY = "python"
def test_setup(self):
"""Test that the ``setup`` function works correctly."""
with self.assertRaises(ValueError):
setup("HOST", 5555, device_class="INVALID")
with patchers.PATCH_ADB_DEVICE_TCP, patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(
DEVICE_PROPERTIES_OUTPUT1
)[self.PATCH_KEY]:
ftv = setup("HOST", 5555)
self.assertIsInstance(ftv, FireTVSync)
self.assertDictEqual(ftv.device_properties, DEVICE_PROPERTIES_DICT1)
with patchers.PATCH_ADB_DEVICE_TCP, patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(
DEVICE_PROPERTIES_OUTPUT2
)[self.PATCH_KEY]:
atv = setup("HOST", 5555)
self.assertIsInstance(atv, AndroidTVSync)
self.assertDictEqual(atv.device_properties, DEVICE_PROPERTIES_DICT2)
with patchers.PATCH_ADB_DEVICE_TCP, patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(
DEVICE_PROPERTIES_OUTPUT1
)[self.PATCH_KEY]:
ftv = setup("HOST", 5555, device_class="androidtv")
self.assertIsInstance(ftv, AndroidTVSync)
self.assertDictEqual(ftv.device_properties, DEVICE_PROPERTIES_DICT1)
with patchers.PATCH_ADB_DEVICE_TCP, patchers.patch_connect(True)[self.PATCH_KEY], patchers.patch_shell(
DEVICE_PROPERTIES_OUTPUT2
)[self.PATCH_KEY]:
atv = setup("HOST", 5555, device_class="firetv")
self.assertIsInstance(atv, FireTVSync)
self.assertDictEqual(atv.device_properties, DEVICE_PROPERTIES_DICT2)
if __name__ == "__main__":
unittest.main()
|
