0
|
1 |
#
|
|
2 |
# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
|
|
3 |
# All rights reserved.
|
|
4 |
# This component and the accompanying materials are made available
|
|
5 |
# under the terms of "Eclipse Public License v1.0"
|
|
6 |
# which accompanies this distribution, and is available
|
|
7 |
# at the URL "http://www.eclipse.org/legal/epl-v10.html".
|
|
8 |
#
|
|
9 |
# Initial Contributors:
|
|
10 |
# Nokia Corporation - initial contribution.
|
|
11 |
#
|
|
12 |
# Contributors:
|
|
13 |
#
|
|
14 |
# Description:
|
|
15 |
#
|
|
16 |
|
|
17 |
|
|
18 |
"""
|
|
19 |
Test the import export functionality
|
|
20 |
"""
|
|
21 |
|
|
22 |
import unittest
|
|
23 |
import sys, os, shutil
|
|
24 |
|
|
25 |
ROOT_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
26 |
|
|
27 |
from cone.public.api import Resource
|
|
28 |
from cone.storage import filestorage, zipstorage
|
|
29 |
|
|
30 |
from testautomation.base_testcase import BaseTestCase
|
|
31 |
from testautomation.unzip_file import unzip_file
|
|
32 |
|
|
33 |
def abspath(p):
|
|
34 |
return os.path.normpath(os.path.join(ROOT_PATH, p))
|
|
35 |
|
|
36 |
DATA_ZIP = abspath("file_vs_zip_data.zip")
|
|
37 |
TEMP_DIR = abspath("temp/file_vs_zip")
|
|
38 |
|
|
39 |
EXPECTED = [
|
|
40 |
# These two are not used every time
|
|
41 |
('folder', 'dir'),
|
|
42 |
('folder', 'empty2'),
|
|
43 |
|
|
44 |
('folder', 'dir/emptysub'),
|
|
45 |
('folder', 'empty'),
|
|
46 |
('folder', 'empty2/sub'),
|
|
47 |
('resource', 'root.txt'),
|
|
48 |
('resource', 'dir/test1.txt'),
|
|
49 |
('resource', 'dir/test2.txt'),
|
|
50 |
]
|
|
51 |
|
|
52 |
class TestFileStorageVsZipStorage(BaseTestCase):
|
|
53 |
def setUp(self):
|
|
54 |
pass
|
|
55 |
|
|
56 |
def get_temp_zip_storage(self, path, empty=False):
|
|
57 |
"""
|
|
58 |
Get a new zip storage located in the temp/ directory.
|
|
59 |
|
|
60 |
@param path: Path of the storage zip file relative to the temp/ directory.
|
|
61 |
@param empty: If False, the DATA_ZIP is copied to the location specified
|
|
62 |
by 'path' and returned as a read-only storage. If True, a new writable
|
|
63 |
empty storage is returned.
|
|
64 |
"""
|
|
65 |
full_path = os.path.join(TEMP_DIR, path)
|
|
66 |
if not empty:
|
|
67 |
self.create_dir_for_file_path(full_path)
|
|
68 |
shutil.copy2(DATA_ZIP, full_path)
|
|
69 |
return zipstorage.ZipStorage(full_path, 'r')
|
|
70 |
else:
|
|
71 |
if os.path.exists(full_path):
|
|
72 |
os.remove(full_path)
|
|
73 |
return zipstorage.ZipStorage(full_path, 'w')
|
|
74 |
|
|
75 |
def get_temp_file_storage(self, path, empty=False):
|
|
76 |
"""
|
|
77 |
Get a new file storage located in the temp/ directory.
|
|
78 |
|
|
79 |
@param path: Path of the storage directory relative to the temp/ directory.
|
|
80 |
@param empty: If False, the DATA_ZIP is extracted to the location specified
|
|
81 |
by 'path' and returned as a read-only storage. If True, a new writable
|
|
82 |
empty storage is returned.
|
|
83 |
"""
|
|
84 |
full_path = os.path.join(TEMP_DIR, path)
|
|
85 |
if not empty:
|
|
86 |
unzip_file(DATA_ZIP, full_path, delete_if_exists=True)
|
|
87 |
return filestorage.FileStorage(full_path, 'r')
|
|
88 |
else:
|
|
89 |
self.recreate_dir(full_path)
|
|
90 |
return filestorage.FileStorage(full_path, 'w')
|
|
91 |
|
|
92 |
def assert_storage_contains_expected(self, storage, empty_folders):
|
|
93 |
# Always list also empty folders, because empty_folders is False,
|
|
94 |
# they should not have been exported in the first place
|
|
95 |
resource_list = storage.list_resources('', recurse=True, empty_folders=True)
|
|
96 |
|
|
97 |
expected = list(EXPECTED)
|
|
98 |
|
|
99 |
# Remove empty folders from expected if necessary
|
|
100 |
if not empty_folders:
|
|
101 |
for i in reversed(xrange(len(expected))):
|
|
102 |
if expected[i][0] == 'folder':
|
|
103 |
del expected[i]
|
|
104 |
|
|
105 |
# If the actual list contains exactly 2 less entries, they are
|
|
106 |
# probably the root folders that are not present in every case
|
|
107 |
# (e.g. there is a resource 'my_dir/file.txt', so sometimes there
|
|
108 |
# is a folder 'my_dir' in the resource list, but not always)
|
|
109 |
if len(resource_list) == len(expected) - 2:
|
|
110 |
del expected[0:2]
|
|
111 |
|
|
112 |
# Check that there is the same amount of resources/folders
|
|
113 |
self.assertEquals(len(expected), len(resource_list), "Expected %r, actual %r" % (expected, resource_list))
|
|
114 |
|
|
115 |
# Check that all expected resources/folders are present in the storage
|
|
116 |
for type, ref in expected:
|
|
117 |
if type == 'resource':
|
|
118 |
self.assertTrue(storage.is_resource(ref), "(%r, %r) expected, but not a resource" % (type, ref))
|
|
119 |
elif type == 'folder':
|
|
120 |
self.assertTrue(storage.is_folder(ref), "(%r, %r) expected, but not a folder" % (type, ref))
|
|
121 |
else:
|
|
122 |
raise RuntimeError("Invalid type field in expected data")
|
|
123 |
|
|
124 |
def _run_test_storage_to_storage(self, source_storage, target_storage, empty_folders):
|
|
125 |
try:
|
|
126 |
# Check that the source storage contains all expected in the first place
|
|
127 |
self.assert_storage_contains_expected(source_storage, True)
|
|
128 |
|
|
129 |
# Export resources
|
|
130 |
resources = source_storage.list_resources('', recurse=True, empty_folders=empty_folders)
|
|
131 |
source_storage.export_resources(resources, target_storage, empty_folders=empty_folders)
|
|
132 |
|
|
133 |
# Check that resources have been exported properly
|
|
134 |
self.assert_storage_contains_expected(target_storage, empty_folders)
|
|
135 |
finally:
|
|
136 |
source_storage.close()
|
|
137 |
target_storage.close()
|
|
138 |
|
|
139 |
def test_export_file_to_file(self):
|
|
140 |
self._run_test_storage_to_storage(
|
|
141 |
source_storage = self.get_temp_file_storage('f2f/source'),
|
|
142 |
target_storage = self.get_temp_file_storage('f2f/target', empty=True),
|
|
143 |
empty_folders = False)
|
|
144 |
|
|
145 |
self._run_test_storage_to_storage(
|
|
146 |
source_storage = self.get_temp_file_storage('f2f/ef_source'),
|
|
147 |
target_storage = self.get_temp_file_storage('f2f/ef_target', empty=True),
|
|
148 |
empty_folders = True)
|
|
149 |
|
|
150 |
def test_export_zip_to_zip(self):
|
|
151 |
self._run_test_storage_to_storage(
|
|
152 |
source_storage = self.get_temp_zip_storage('z2z/source.zip'),
|
|
153 |
target_storage = self.get_temp_zip_storage('z2z/target.zip', empty=True),
|
|
154 |
empty_folders = False)
|
|
155 |
|
|
156 |
self._run_test_storage_to_storage(
|
|
157 |
source_storage = self.get_temp_zip_storage('z2z/ef_source.zip'),
|
|
158 |
target_storage = self.get_temp_zip_storage('z2z/ef_target.zip', empty=True),
|
|
159 |
empty_folders = True)
|
|
160 |
|
|
161 |
def test_export_zip_to_file(self):
|
|
162 |
self._run_test_storage_to_storage(
|
|
163 |
source_storage = self.get_temp_zip_storage('z2f/source.zip'),
|
|
164 |
target_storage = self.get_temp_file_storage('z2f/target', empty=True),
|
|
165 |
empty_folders = False)
|
|
166 |
|
|
167 |
self._run_test_storage_to_storage(
|
|
168 |
source_storage = self.get_temp_zip_storage('z2f/ef_source.zip'),
|
|
169 |
target_storage = self.get_temp_file_storage('z2f/ef_target', empty=True),
|
|
170 |
empty_folders = True)
|
|
171 |
|
5
|
172 |
#def test_export_file_to_zip(self):
|
|
173 |
# Currently fails, because zip storage will only contain exported file resources after closing
|
|
174 |
# self._run_test_storage_to_storage(
|
|
175 |
# source_storage = self.get_temp_file_storage('f2z/source'),
|
|
176 |
# target_storage = self.get_temp_zip_storage('f2z/target.zip', empty=True),
|
|
177 |
# empty_folders = False)
|
|
178 |
#
|
|
179 |
# self._run_test_storage_to_storage(
|
|
180 |
# source_storage = self.get_temp_file_storage('f2z/ef_source'),
|
|
181 |
# target_storage = self.get_temp_zip_storage('f2z/ef_target.zip', empty=True),
|
|
182 |
# empty_folders = True)
|
0
|
183 |
|
|
184 |
|
|
185 |
if __name__ == '__main__':
|
|
186 |
unittest.main()
|