blob: d2aeb3c8c9b2195a6eaf68cf51f310261214f377 (
plain)
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
|
"""The tests for discovering Xbox gaming consoles via SmartGlass protocol."""
import unittest
from binascii import unhexlify
from netdisco.smartglass import XboxSmartGlass
class TestXboxOne(unittest.TestCase):
"""Test the Xbox One Discoverable."""
def setUp(self):
"""
Setup test class
"""
with open('tests/xboxone_files/discovery_response', 'rb') as content:
packet = content.read()
if not packet:
raise Exception('Failed to read test data')
self.discovery_response = packet
def test_assemble_request(self):
"""
Test discovery request assembly
"""
packet = XboxSmartGlass.discovery_packet()
self.assertEqual(
packet,
unhexlify(b'dd00000a000000000000000400000002')
)
def test_parse_response(self):
"""
Test discovery response parsing
"""
response = XboxSmartGlass.parse_discovery_response(
self.discovery_response)
self.assertEqual(response['device_type'], 1)
self.assertEqual(response['flags'], 2)
self.assertEqual(response['name'], 'XboxOne')
self.assertEqual(response['uuid'],
'DE305D54-75B4-431B-ADB2-EB6B9E546014')
self.assertEqual(response['last_error'], 0)
self.assertEqual(response['certificate'][:8], '30820203')
self.assertEqual(len(unhexlify(response['certificate'])), 519)
def test_verify_response(self):
"""
Test discovery response verification
"""
valid_parse = XboxSmartGlass.verify_packet(self.discovery_response)
invalid_length = XboxSmartGlass.verify_packet(unhexlify(b'41'))
invalid_magic = XboxSmartGlass.verify_packet(
unhexlify(b'aabbccddeeff00'))
self.assertIsNotNone(valid_parse)
self.assertIsNone(invalid_length)
self.assertIsNone(invalid_magic)
|
