Skip to content

Commit b5240d2

Browse files
committed
Removed unnecessary bytes prefix for ContentFile
1 parent 361d673 commit b5240d2

File tree

2 files changed

+27
-27
lines changed

2 files changed

+27
-27
lines changed

tests/modeltests/files/tests.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def tearDown(self):
2020
shutil.rmtree(temp_storage_location)
2121

2222
def test_files(self):
23-
temp_storage.save('tests/default.txt', ContentFile(b'default content'))
23+
temp_storage.save('tests/default.txt', ContentFile('default content'))
2424
# Attempting to access a FileField from the class raises a descriptive
2525
# error
2626
self.assertRaises(AttributeError, lambda: Storage.normal)
@@ -31,7 +31,7 @@ def test_files(self):
3131
self.assertRaises(ValueError, lambda: obj1.normal.size)
3232

3333
# Saving a file enables full functionality.
34-
obj1.normal.save("django_test.txt", ContentFile(b"content"))
34+
obj1.normal.save("django_test.txt", ContentFile("content"))
3535
self.assertEqual(obj1.normal.name, "tests/django_test.txt")
3636
self.assertEqual(obj1.normal.size, 7)
3737
self.assertEqual(obj1.normal.read(), b"content")
@@ -59,7 +59,7 @@ def test_files(self):
5959

6060
# Save another file with the same name.
6161
obj2 = Storage()
62-
obj2.normal.save("django_test.txt", ContentFile(b"more content"))
62+
obj2.normal.save("django_test.txt", ContentFile("more content"))
6363
self.assertEqual(obj2.normal.name, "tests/django_test_1.txt")
6464
self.assertEqual(obj2.normal.size, 12)
6565

@@ -70,13 +70,13 @@ def test_files(self):
7070

7171
# Deleting an object does not delete the file it uses.
7272
obj2.delete()
73-
obj2.normal.save("django_test.txt", ContentFile(b"more content"))
73+
obj2.normal.save("django_test.txt", ContentFile("more content"))
7474
self.assertEqual(obj2.normal.name, "tests/django_test_2.txt")
7575

7676
# Multiple files with the same name get _N appended to them.
7777
objs = [Storage() for i in range(3)]
7878
for o in objs:
79-
o.normal.save("multiple_files.txt", ContentFile(b"Same Content"))
79+
o.normal.save("multiple_files.txt", ContentFile("Same Content"))
8080
self.assertEqual(
8181
[o.normal.name for o in objs],
8282
["tests/multiple_files.txt", "tests/multiple_files_1.txt", "tests/multiple_files_2.txt"]
@@ -100,7 +100,7 @@ def test_files(self):
100100
# Verify the fix for #5655, making sure the directory is only
101101
# determined once.
102102
obj4 = Storage()
103-
obj4.random.save("random_file", ContentFile(b"random content"))
103+
obj4.random.save("random_file", ContentFile("random content"))
104104
self.assertTrue(obj4.random.name.endswith("/random_file"))
105105

106106
def test_max_length(self):

tests/regressiontests/file_storage/tests.py

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def test_file_accessed_time(self):
128128
"""
129129
self.assertFalse(self.storage.exists('test.file'))
130130

131-
f = ContentFile(b'custom contents')
131+
f = ContentFile('custom contents')
132132
f_name = self.storage.save('test.file', f)
133133
atime = self.storage.accessed_time(f_name)
134134

@@ -144,7 +144,7 @@ def test_file_created_time(self):
144144
"""
145145
self.assertFalse(self.storage.exists('test.file'))
146146

147-
f = ContentFile(b'custom contents')
147+
f = ContentFile('custom contents')
148148
f_name = self.storage.save('test.file', f)
149149
ctime = self.storage.created_time(f_name)
150150

@@ -161,7 +161,7 @@ def test_file_modified_time(self):
161161
"""
162162
self.assertFalse(self.storage.exists('test.file'))
163163

164-
f = ContentFile(b'custom contents')
164+
f = ContentFile('custom contents')
165165
f_name = self.storage.save('test.file', f)
166166
mtime = self.storage.modified_time(f_name)
167167

@@ -178,7 +178,7 @@ def test_file_save_without_name(self):
178178
"""
179179
self.assertFalse(self.storage.exists('test.file'))
180180

181-
f = ContentFile(b'custom contents')
181+
f = ContentFile('custom contents')
182182
f.name = 'test.file'
183183

184184
storage_f_name = self.storage.save(None, f)
@@ -195,7 +195,7 @@ def test_file_save_with_path(self):
195195
"""
196196
self.assertFalse(self.storage.exists('path/to'))
197197
self.storage.save('path/to/test.file',
198-
ContentFile(b'file saved with path'))
198+
ContentFile('file saved with path'))
199199

200200
self.assertTrue(self.storage.exists('path/to'))
201201
with self.storage.open('path/to/test.file') as f:
@@ -212,7 +212,7 @@ def test_file_path(self):
212212
"""
213213
self.assertFalse(self.storage.exists('test.file'))
214214

215-
f = ContentFile(b'custom contents')
215+
f = ContentFile('custom contents')
216216
f_name = self.storage.save('test.file', f)
217217

218218
self.assertEqual(self.storage.path(f_name),
@@ -247,8 +247,8 @@ def test_listdir(self):
247247
self.assertFalse(self.storage.exists('storage_test_2'))
248248
self.assertFalse(self.storage.exists('storage_dir_1'))
249249

250-
f = self.storage.save('storage_test_1', ContentFile(b'custom content'))
251-
f = self.storage.save('storage_test_2', ContentFile(b'custom content'))
250+
f = self.storage.save('storage_test_1', ContentFile('custom content'))
251+
f = self.storage.save('storage_test_2', ContentFile('custom content'))
252252
os.mkdir(os.path.join(self.temp_dir, 'storage_dir_1'))
253253

254254
dirs, files = self.storage.listdir('')
@@ -305,18 +305,18 @@ def fake_makedirs(path):
305305
os.makedirs = fake_makedirs
306306

307307
self.storage.save('normal/test.file',
308-
ContentFile(b'saved normally'))
308+
ContentFile('saved normally'))
309309
with self.storage.open('normal/test.file') as f:
310310
self.assertEqual(f.read(), b'saved normally')
311311

312312
self.storage.save('raced/test.file',
313-
ContentFile(b'saved with race'))
313+
ContentFile('saved with race'))
314314
with self.storage.open('raced/test.file') as f:
315315
self.assertEqual(f.read(), b'saved with race')
316316

317317
# Check that OSErrors aside from EEXIST are still raised.
318318
self.assertRaises(OSError,
319-
self.storage.save, 'error/test.file', ContentFile(b'not saved'))
319+
self.storage.save, 'error/test.file', ContentFile('not saved'))
320320
finally:
321321
os.makedirs = real_makedirs
322322

@@ -342,16 +342,16 @@ def fake_remove(path):
342342
try:
343343
os.remove = fake_remove
344344

345-
self.storage.save('normal.file', ContentFile(b'delete normally'))
345+
self.storage.save('normal.file', ContentFile('delete normally'))
346346
self.storage.delete('normal.file')
347347
self.assertFalse(self.storage.exists('normal.file'))
348348

349-
self.storage.save('raced.file', ContentFile(b'delete with race'))
349+
self.storage.save('raced.file', ContentFile('delete with race'))
350350
self.storage.delete('raced.file')
351351
self.assertFalse(self.storage.exists('normal.file'))
352352

353353
# Check that OSErrors aside from ENOENT are still raised.
354-
self.storage.save('error.file', ContentFile(b'delete with error'))
354+
self.storage.save('error.file', ContentFile('delete with error'))
355355
self.assertRaises(OSError, self.storage.delete, 'error.file')
356356
finally:
357357
os.remove = real_remove
@@ -375,9 +375,9 @@ class CustomStorageTests(FileStorageTests):
375375
storage_class = CustomStorage
376376

377377
def test_custom_get_available_name(self):
378-
first = self.storage.save('custom_storage', ContentFile(b'custom contents'))
378+
first = self.storage.save('custom_storage', ContentFile('custom contents'))
379379
self.assertEqual(first, 'custom_storage')
380-
second = self.storage.save('custom_storage', ContentFile(b'more contents'))
380+
second = self.storage.save('custom_storage', ContentFile('more contents'))
381381
self.assertEqual(second, 'custom_storage.2')
382382
self.storage.delete(first)
383383
self.storage.delete(second)
@@ -434,7 +434,7 @@ def tearDown(self):
434434
shutil.rmtree(self.storage_dir)
435435

436436
def test_file_upload_permissions(self):
437-
name = self.storage.save("the_file", ContentFile(b"data"))
437+
name = self.storage.save("the_file", ContentFile("data"))
438438
actual_mode = os.stat(self.storage.path(name))[0] & 0o777
439439
self.assertEqual(actual_mode, 0o666)
440440

@@ -454,8 +454,8 @@ def test_directory_with_dot(self):
454454
sure we still mangle the file name instead of the directory name.
455455
"""
456456

457-
self.storage.save('dotted.path/test', ContentFile(b"1"))
458-
self.storage.save('dotted.path/test', ContentFile(b"2"))
457+
self.storage.save('dotted.path/test', ContentFile("1"))
458+
self.storage.save('dotted.path/test', ContentFile("2"))
459459

460460
self.assertFalse(os.path.exists(os.path.join(self.storage_dir, 'dotted_.path')))
461461
self.assertTrue(os.path.exists(os.path.join(self.storage_dir, 'dotted.path/test')))
@@ -466,8 +466,8 @@ def test_first_character_dot(self):
466466
File names with a dot as their first character don't have an extension,
467467
and the underscore should get added to the end.
468468
"""
469-
self.storage.save('dotted.path/.test', ContentFile(b"1"))
470-
self.storage.save('dotted.path/.test', ContentFile(b"2"))
469+
self.storage.save('dotted.path/.test', ContentFile("1"))
470+
self.storage.save('dotted.path/.test', ContentFile("2"))
471471

472472
self.assertTrue(os.path.exists(os.path.join(self.storage_dir, 'dotted.path/.test')))
473473
self.assertTrue(os.path.exists(os.path.join(self.storage_dir, 'dotted.path/.test_1')))

0 commit comments

Comments
 (0)