-
Notifications
You must be signed in to change notification settings - Fork 19
/
output_writer.py
88 lines (66 loc) · 2.91 KB
/
output_writer.py
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Tests for the output writer."""
import unittest
from winregrc import output_writers
from tests import test_lib as shared_test_lib
class StdoutOutputWriterTest(shared_test_lib.BaseTestCase):
"""Tests for the stdout output writer."""
# pylint: disable=protected-access
def testFormatDataInHexadecimal(self):
"""Tests the _FormatDataInHexadecimal function."""
test_output_writer = output_writers.StdoutOutputWriter()
data = b'\x00\x01\x02\x03\x04\x05\x06'
expected_formatted_data = (
'0x00000000 00 01 02 03 04 05 06 '
'.......\n'
'\n')
formatted_data = test_output_writer._FormatDataInHexadecimal(data)
self.assertEqual(formatted_data, expected_formatted_data)
data = b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09'
expected_formatted_data = (
'0x00000000 00 01 02 03 04 05 06 07 08 09 '
'..........\n'
'\n')
formatted_data = test_output_writer._FormatDataInHexadecimal(data)
self.assertEqual(formatted_data, expected_formatted_data)
data = b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f'
expected_formatted_data = (
'0x00000000 00 01 02 03 04 05 06 07 08 09 0a 0b 0c 0d 0e 0f '
'................\n'
'\n')
formatted_data = test_output_writer._FormatDataInHexadecimal(data)
self.assertEqual(formatted_data, expected_formatted_data)
data = (
b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f'
b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f'
b'\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f')
expected_formatted_data = (
'0x00000000 00 01 02 03 04 05 06 07 08 09 0a 0b 0c 0d 0e 0f '
'................\n'
'...\n'
'0x00000020 00 01 02 03 04 05 06 07 08 09 0a 0b 0c 0d 0e 0f '
'................\n'
'\n')
formatted_data = test_output_writer._FormatDataInHexadecimal(data)
self.assertEqual(formatted_data, expected_formatted_data)
def testOpenClose(self):
"""Tests the Open and Close functions."""
test_output_writer = output_writers.StdoutOutputWriter()
result = test_output_writer.Open()
self.assertTrue(result)
test_output_writer.Close()
def testWriteDebugData(self):
"""Tests the WriteDebugData function."""
test_output_writer = output_writers.StdoutOutputWriter()
test_output_writer.WriteDebugData('Description', b'DATA')
def testWriteValue(self):
"""Tests the WriteValue function."""
test_output_writer = output_writers.StdoutOutputWriter()
test_output_writer.WriteValue('Description', 'Value')
def testWriteText(self):
"""Tests the WriteText function."""
test_output_writer = output_writers.StdoutOutputWriter()
test_output_writer.WriteText('Test')
if __name__ == '__main__':
unittest.main()