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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
#!/usr/bin/python3
# Copyright 2020 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import unittest
import common
from autotest_lib.server.cros.servo.topology import servo_topology
# pylint: disable=missing-docstring
class ConvertDataToStringTests(unittest.TestCase):
def test_expect_empty_when_data_is_none(self):
data = servo_topology._convert_topology_to_string(None)
self.assertEqual(data, '')
def test_expect_empty_when_data_is_empty(self):
data = servo_topology._convert_topology_to_string({})
self.assertEqual(data, '')
def test_expect_string_when_data_is_provided(self):
expected_string = (
'eyJtYWluIjp7InVzYl9odWJfcG9ydCI6IjYuNC4xIiwic2VyaWFsIjoiQzE'
'5MDMxNDU1OTEhQCMkJV4qKClcIic6OycnXFw/L2BcIiwiLCJ0eXBlIjoic2'
'Vydm9fdjQiLCJzeXNmc19wcm9kdWN0IjoiU2Vydm8gVjQifSwiY2hpbGRyZ'
'W4iOlt7InVzYl9odWJfcG9ydCI6IjYuNC4yIiwic2VyaWFsIjoiMDY4MUQw'
'M0EtOTJEQ0NENjQiLCJ0eXBlIjoiY2NkX2NyNTAiLCJzeXNmc19wcm9kdWN'
'0IjoiQ3I1MCJ9XX0=')
data = {
"main": {
"usb_hub_port": "6.4.1",
"serial": "C1903145591!@#$%^*()\"':;''\\?/`\",",
"type": "servo_v4",
"sysfs_product": "Servo V4"
},
"children": [{
"usb_hub_port": "6.4.2",
"serial": "0681D03A-92DCCD64",
"type": "ccd_cr50",
"sysfs_product": "Cr50"
}]
}
result = servo_topology._convert_topology_to_string(data)
self.assertEqual(result, expected_string)
def test_expect_string_when_data_is_provided2(self):
expected_string = (
'eyJtYWluIjp7InVzYl9odWJfcG9ydCI6IiIsInNlcmlhbCI6IiIsInR5cGU'
'iOiIiLCJzeXNmc19wcm9kdWN0IjoiIn0sImNoaWxkcmVuIjpbXX0=')
data = {
"main": {
"usb_hub_port": "",
"serial": "",
"type": "",
"sysfs_product": ""
},
"children": []
}
result = servo_topology._convert_topology_to_string(data)
self.assertEqual(result, expected_string)
class ParseStringToDataTests(unittest.TestCase):
def test_expect_none_when_string_empty(self):
data = servo_topology._parse_string_as_topology("")
self.assertEqual(data, None)
def test_expect_empty_data_when_string_empty_instance(self):
data = servo_topology._parse_string_as_topology('{}')
self.assertEqual(data, None)
def test_expect_data_when_string_is_provided(self):
b64_data = (
'eyJtYWluIjp7InVzYl9odWJfcG9ydCI6IjYuNC4xIiwic2VyaWFsIjoiQzE5'
'MDMxNDU1OTEhQCMkJV4mKigpXCInOjsnJ1xcPy9gflwiLCIsInR5cGUiOiJz'
'ZXJ2b192NCIsInN5c2ZzX3Byb2R1Y3QiOiJTZXJ2byBWNCJ9LCJjaGlsZHJl'
'biI6W3sidXNiX2h1Yl9wb3J0IjoiNi40LjIiLCJzZXJpYWwiOiIwNjgxRDAz'
'QS05MkRDQ0Q2NCIsInR5cGUiOiJjY2RfY3I1MCIsInN5c2ZzX3Byb2R1Y3Qi'
'OiJDcjUwIn1dfQ==')
data = {
"main": {
"usb_hub_port": "6.4.1",
"serial": "C1903145591!@#$%^&*()\"':;''\\?/`~\",",
"type": "servo_v4",
"sysfs_product": "Servo V4"
},
"children": [{
"usb_hub_port": "6.4.2",
"serial": "0681D03A-92DCCD64",
"type": "ccd_cr50",
"sysfs_product": "Cr50"
}]
}
result = servo_topology._parse_string_as_topology(b64_data)
self.assertEqual(result, data)
def test_expect_data_when_string_is_provided2(self):
b64_data = (
'eyJtYWluIjp7InVzYl9odWJfcG9ydCI6IiIsInNlcmlhbCI6IiIsInR5cGU'
'iOiIiLCJzeXNmc19wcm9kdWN0IjoiIn0sImNoaWxkcmVuIjpbXX0=')
data = {
"main": {
"usb_hub_port": "",
"serial": "",
"type": "",
"sysfs_product": ""
},
"children": []
}
result = servo_topology._parse_string_as_topology(b64_data)
self.assertEqual(result, data)
if __name__ == '__main__':
unittest.main()
|