Ticket #15197: test_serializers_http.patch

tests/regressiontests/serializers_regress/tests.py

@@ -29 +29 @@
 from django.test import TestCase
 from django.utils.functional import curry
 from django.utils.unittest import skipUnless
+from django.http import HttpResponse
 
 from .models import (BooleanData, CharData, DateData, DateTimeData, EmailData,
     FileData, FilePathData, DecimalData, FloatData, IntegerData, IPAddressData,
@@ -469 +470 @@
     obj = ComplexModel(field1='first',field2='second',field3='third')
     obj.save_base(raw=True)
 
-    # Serialize the test database to a stream
-    stream = StringIO()
-    serializers.serialize(format, [obj], indent=2, stream=stream)
+    for stream in (StringIO(), HttpResponse()):
+        # Serialize the test database to a stream
+        serializers.serialize(format, [obj], indent=2, stream=stream)
 
-    # Serialize normally for a comparison
-    string_data = serializers.serialize(format, [obj], indent=2)
+        # Serialize normally for a comparison
+        string_data = serializers.serialize(format, [obj], indent=2)
 
-    # Check that the two are the same
-    self.assertEqual(string_data, stream.getvalue())
-    stream.close()
+        # Check that the two are the same
+        if callable(getattr(stream, 'getvalue', None)):
+            self.assertEqual(string_data, stream.getvalue())
+        elif callable(getattr(stream, 'content', None)):
+            self.assertEqual(string_data, stream.content())
+        stream.close()
 
 for format in serializers.get_serializer_formats():
     setattr(SerializerTests, 'test_' + format + '_serializer', curry(serializerTest, format))