old_contents: """
      Licensed under the Apache License, Version 2.0 (the "License");
      you may not use this file except in compliance with the License.
      You may obtain a copy of the License at
  
          http://www.apache.org/licenses/LICENSE-2.0
  
      Unless required by applicable law or agreed to in writing, software
      distributed under the License is distributed on an "AS IS" BASIS,
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      See the License for the specific language governing permissions and
      limitations under the License.
  """
  import collections
  from django.test import TestCase
  from django.urls import reverse
  from rest_framework.test import APITestCase
  
  from gwells.documents import MinioClient
  
  
  class DocumentTests(TestCase):
  
      def setUp(self):
          self.client = MinioClient(disable_private=True, disable_public=True)
          FileObject = collections.namedtuple(
              'FileObject', 'bucket_name object_name')
          self.bucket_name = 'test-bucket'
          self.file1 = FileObject(bucket_name=self.bucket_name,
                                  object_name='test-object')
          self.file2 = FileObject(bucket_name=self.bucket_name,
                                  object_name='test-object2')
          self.host = 'www.example.com'
  
      def test_create_url(self):
          url = self.client.create_url(self.file1, self.host, self.bucket_name)
          self.assertEqual(
              url, 'https://www.example.com/test-bucket/test-object')
  
      def test_create_url_list(self):
          file_list = [self.file1, self.file2]
          url_list = self.client.create_url_list(file_list, self.host, self.bucket_name)
  
          self.assertEqual(len(url_list), 2)
          self.assertEqual(url_list[1]['name'], 'test-object2')
  
  
  class TestDocumentsAPI(APITestCase):
      def test_documents_404(self):
          url = reverse('file-list', kwargs={'tag': 987654321})
          response = self.client.get(url)
          self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
new_contents: """
      Licensed under the Apache License, Version 2.0 (the "License");
      you may not use this file except in compliance with the License.
      You may obtain a copy of the License at
  
          http://www.apache.org/licenses/LICENSE-2.0
  
      Unless required by applicable law or agreed to in writing, software
      distributed under the License is distributed on an "AS IS" BASIS,
      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      See the License for the specific language governing permissions and
      limitations under the License.
  """
  import collections
  from django.test import TestCase
  from django.urls import reverse
  from rest_framework import status
  from rest_framework.test import APITestCase
  
  from gwells.documents import MinioClient
  
  
  class DocumentTests(TestCase):
  
      def setUp(self):
          self.client = MinioClient(disable_private=True, disable_public=True)
          FileObject = collections.namedtuple(
              'FileObject', 'bucket_name object_name')
          self.bucket_name = 'test-bucket'
          self.file1 = FileObject(bucket_name=self.bucket_name,
                                  object_name='test-object')
          self.file2 = FileObject(bucket_name=self.bucket_name,
                                  object_name='test-object2')
          self.host = 'www.example.com'
  
      def test_create_url(self):
          url = self.client.create_url(self.file1, self.host, self.bucket_name)
          self.assertEqual(
              url, 'https://www.example.com/test-bucket/test-object')
  
      def test_create_url_list(self):
          file_list = [self.file1, self.file2]
          url_list = self.client.create_url_list(file_list, self.host, self.bucket_name)
  
          self.assertEqual(len(url_list), 2)
          self.assertEqual(url_list[1]['name'], 'test-object2')
  
  
  class TestDocumentsAPI(APITestCase):
      def test_documents_404(self):
          url = reverse('file-list', kwargs={'tag': 987654321})
          response = self.client.get(url)
          self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
subject: fix missing import