/tests/modeltests/files/tests_25.py

 1from __future__ import with_statement
 2
 3import tempfile
 4
 5from django.core.files import File
 6from django.utils.unittest import TestCase
 7
 8
 9class FileObjTests(TestCase):
10    def test_context_manager(self):
11        orig_file = tempfile.TemporaryFile()
12        base_file = File(orig_file)
13        with base_file as f:
14            self.assertIs(base_file, f)
15            self.assertFalse(f.closed)
16        self.assertTrue(f.closed)
17        self.assertTrue(orig_file.closed)