Unify tests that use upload documents. Add explicit mode when opening a file.
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
@@ -12,7 +12,7 @@ TEST_VIEW_URL = 'test-view-url'
|
||||
TEST_FILENAME1 = 'test_file1.txt'
|
||||
TEST_FILENAME2 = 'test_file2.txt'
|
||||
TEST_FILENAME3 = 'test_file3.txt'
|
||||
TEST_FILE_CONTENTS_1 = 'TEST FILE 1\n'
|
||||
TEST_FILE_CONTENTS_1 = b'TEST FILE 1\n'
|
||||
TEST_FILE_CONTENTS_2 = 'TEST FILE 2\n'
|
||||
TEST_TAR_BZ2_FILE = 'test_file.tar.bz2'
|
||||
TEST_TAR_FILE = 'test_file.tar'
|
||||
|
||||
@@ -23,22 +23,22 @@ class TarArchiveClassTestCase(BaseTestCase):
|
||||
def test_add_file(self):
|
||||
archive = self.cls()
|
||||
archive.create()
|
||||
with open(self.file_path) as file_object:
|
||||
with open(self.file_path, 'rb') as file_object:
|
||||
archive.add_file(file_object=file_object, filename=self.filename)
|
||||
self.assertTrue(archive.members(), [self.filename])
|
||||
|
||||
def test_open(self):
|
||||
with open(self.archive_path) as file_object:
|
||||
with open(self.archive_path, 'rb') as file_object:
|
||||
archive = Archive.open(file_object=file_object)
|
||||
self.assertTrue(isinstance(archive, self.cls))
|
||||
|
||||
def test_members(self):
|
||||
with open(self.archive_path) as file_object:
|
||||
with open(self.archive_path, 'rb') as file_object:
|
||||
archive = Archive.open(file_object=file_object)
|
||||
self.assertEqual(archive.members(), self.members_list)
|
||||
|
||||
def test_member_contents(self):
|
||||
with open(self.archive_path) as file_object:
|
||||
with open(self.archive_path, 'rb') as file_object:
|
||||
archive = Archive.open(file_object=file_object)
|
||||
self.assertEqual(
|
||||
archive.member_contents(filename=self.member_name),
|
||||
@@ -46,7 +46,7 @@ class TarArchiveClassTestCase(BaseTestCase):
|
||||
)
|
||||
|
||||
def test_open_member(self):
|
||||
with open(self.archive_path) as file_object:
|
||||
with open(self.archive_path, 'rb') as file_object:
|
||||
archive = Archive.open(file_object=file_object)
|
||||
file_object = archive.open_member(filename=self.member_name)
|
||||
self.assertEqual(
|
||||
|
||||
Reference in New Issue
Block a user