forked from tableau/server-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_filesys_helpers.py
More file actions
107 lines (77 loc) · 3.69 KB
/
Copy pathtest_filesys_helpers.py
File metadata and controls
107 lines (77 loc) · 3.69 KB
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
import unittest
from io import BytesIO
import os
from xml.etree import ElementTree as ET
from zipfile import ZipFile
from tableauserverclient.filesys_helpers import get_file_object_size, get_file_type
from ._utils import asset, TEST_ASSET_DIR
class FilesysTests(unittest.TestCase):
def test_get_file_size_returns_correct_size(self):
target_size = 1000 # bytes
with BytesIO() as f:
f.seek(target_size - 1)
f.write(b"\0")
file_size = get_file_object_size(f)
self.assertEqual(file_size, target_size)
def test_get_file_size_returns_zero_for_empty_file(self):
with BytesIO() as f:
file_size = get_file_object_size(f)
self.assertEqual(file_size, 0)
def test_get_file_size_coincides_with_built_in_method(self):
asset_path = asset('SampleWB.twbx')
target_size = os.path.getsize(asset_path)
with open(asset_path, 'rb') as f:
file_size = get_file_object_size(f)
self.assertEqual(file_size, target_size)
def test_get_file_type_identifies_a_zip_file(self):
with BytesIO() as file_object:
with ZipFile(file_object, 'w') as zf:
with BytesIO() as stream:
stream.write('This is a zip file'.encode())
zf.writestr('dummy_file', stream.getbuffer())
file_object.seek(0)
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'zip')
def test_get_file_type_identifies_tdsx_as_zip_file(self):
with open(asset('World Indicators.tdsx'), 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'zip')
def test_get_file_type_identifies_twbx_as_zip_file(self):
with open(asset('SampleWB.twbx'), 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'zip')
def test_get_file_type_identifies_xml_file(self):
root = ET.Element('root')
child = ET.SubElement(root, 'child')
child.text = "This is a child element"
etree = ET.ElementTree(root)
with BytesIO() as file_object:
etree.write(file_object, encoding='utf-8',
xml_declaration=True)
file_object.seek(0)
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'xml')
def test_get_file_type_identifies_tds_as_xml_file(self):
with open(asset('World Indicators.tds'), 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'xml')
def test_get_file_type_identifies_twb_as_xml_file(self):
with open(asset('RESTAPISample.twb'), 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'xml')
def test_get_file_type_identifies_hyper_file(self):
with open(asset('World Indicators.hyper'), 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'hyper')
def test_get_file_type_identifies_tde_file(self):
asset_path = os.path.join(TEST_ASSET_DIR, 'Data', 'Tableau Samples', 'World Indicators.tde')
with open(asset_path, 'rb') as file_object:
file_type = get_file_type(file_object)
self.assertEqual(file_type, 'tde')
def test_get_file_type_handles_unknown_file_type(self):
# Create a dummy png file
with BytesIO() as file_object:
png_signature = bytes.fromhex("89504E470D0A1A0A")
file_object.write(png_signature)
file_object.seek(0)
self.assertRaises(ValueError, get_file_type, file_object)