def test_write_appends_to_file_at_given_offset(self):
tus = TUSUpload('myuid', {'length': 1024})
tus.write(StringIO('0123456789'))
tus.write(StringIO('abc'), 10)
self.assertEqual(13, tus.offset())
with open(tus.filepath, 'rb') as f:
data = f.read()
self.assertEqual('0123456789abc', data)
def test_write_appends_to_file_at_given_offset(self):
tus = TUSUpload("myuid", {"length": 1024})
tus.write(BytesIO(b"0123456789"))
tus.write(BytesIO(b"abc"), 10)
self.assertEqual(13, tus.offset())
with open(tus.filepath, "rb") as f:
data = f.read()
self.assertEqual(b"0123456789abc", data)
tus.cleanup()
def test_cleanup_removes_upload_file(self):
tus = TUSUpload('myuid', {'length': 1024})
tus.write(StringIO('0123456789'))
tus.cleanup()
self.assertFalse(os.path.exists(tus.filepath))
def test_expires_returns_expiration_time_of_current_upload(self):
tus = TUSUpload('myuid', {'length': 1024})
tus.write(StringIO('0123456789'))
self.assertGreater(DateTime(tus.expires()), DateTime())
tus.cleanup()
def test_write_sets_finished_flag(self):
tus = TUSUpload('myuid', {'length': 10})
tus.write(StringIO('0123456789'))
self.assertTrue(tus.finished)
tus.cleanup()
def test_write_creates_new_file(self):
tus = TUSUpload('myuid', {'length': 1024})
tus.write(StringIO('0123456789'))
self.assertTrue(os.path.isfile(tus.filepath))
tus.cleanup()
def test_offset_returns_size_of_current_file(self):
tus = TUSUpload('myuid', {'length': 1024})
tus.write(StringIO('0123456789'))
self.assertEqual(10, tus.offset())
tus.cleanup()
def test_cleanup_removes_upload_file(self):
tus = TUSUpload("myuid", {"length": 1024})
tus.write(BytesIO(b"0123456789"))
tus.cleanup()
self.assertFalse(os.path.exists(tus.filepath))
def test_expires_returns_expiration_time_of_current_upload(self):
tus = TUSUpload("myuid", {"length": 1024})
tus.write(BytesIO(b"0123456789"))
self.assertGreater(DateTime(tus.expires()), DateTime())
tus.cleanup()
def test_write_sets_finished_flag(self):
tus = TUSUpload("myuid", {"length": 10})
tus.write(BytesIO(b"0123456789"))
self.assertTrue(tus.finished)
tus.cleanup()
def test_write_creates_new_file(self):
tus = TUSUpload("myuid", {"length": 1024})
tus.write(BytesIO(b"0123456789"))
self.assertTrue(os.path.isfile(tus.filepath))
tus.cleanup()
def test_offset_returns_size_of_current_file(self):
tus = TUSUpload("myuid", {"length": 1024})
tus.write(BytesIO(b"0123456789"))
self.assertEqual(10, tus.offset())
tus.cleanup()