Support Python 3 in tests.

carljm · Apr 18, 2013 · e610f1c · e610f1c
1 parent d858f97
commit e610f1c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/djangosecure/tests.py b/djangosecure/tests.py
@@ -23,7 +23,7 @@ def secure_request_kwargs(self):
     def response(self, *args, **kwargs):
         headers = kwargs.pop("headers", {})
         response = HttpResponse(*args, **kwargs)
-        for k, v in headers.iteritems():
+        for k, v in headers.items():
             response[k] = v
         return response
 
@@ -386,8 +386,8 @@ def test_returns_warnings(self):
 
 class CheckSettingsCommandTest(TestCase):
     def call(self, **options):
-        stdout = options.setdefault("stdout", StringIO.StringIO())
-        stderr = options.setdefault("stderr", StringIO.StringIO())
+        stdout = options.setdefault("stdout", StringIO())
+        stderr = options.setdefault("stderr", StringIO())
 
         call_command("checksecure", **options)