[3.1.x] Fixed #31550 -- Adjusted ASGI test_file_response for various Windows content types.

chrisxkeith · felixxm · commit 5dec57a6fc5a · 2021-02-04T21:58:38.000+01:00
Backport of 7618130 from master
diff --git a/tests/asgi/tests.py b/tests/asgi/tests.py
@@ -72,14 +72,23 @@ async def test_file_response(self):
         response_start = await communicator.receive_output()
         self.assertEqual(response_start['type'], 'http.response.start')
         self.assertEqual(response_start['status'], 200)
-        self.assertEqual(
-            set(response_start['headers']),
-            {
-                (b'Content-Length', str(len(test_file_contents)).encode('ascii')),
-                (b'Content-Type', b'text/plain' if sys.platform == 'win32' else b'text/x-python'),
-                (b'Content-Disposition', b'inline; filename="urls.py"'),
-            },
-        )
+        headers = response_start['headers']
+        self.assertEqual(len(headers), 3)
+        expected_headers = {
+            b'Content-Length': str(len(test_file_contents)).encode('ascii'),
+            b'Content-Type': b'text/x-python',
+            b'Content-Disposition': b'inline; filename="urls.py"',
+        }
+        for key, value in headers:
+            try:
+                self.assertEqual(value, expected_headers[key])
+            except AssertionError:
+                # Windows registry may not be configured with correct
+                # mimetypes.
+                if sys.platform == 'win32' and key == b'Content-Type':
+                    self.assertEqual(value, b'text/plain')
+                else:
+                    raise
         response_body = await communicator.receive_output()
         self.assertEqual(response_body['type'], 'http.response.body')
         self.assertEqual(response_body['body'], test_file_contents)
