From 13b71e69b4a53aaeebbe2f93cf786ffb7f3b8e56 Mon Sep 17 00:00:00 2001
From: Ben Adida <ben@adida.net>
Date: Sun, 2 Dec 2012 12:02:24 -0800
Subject: [PATCH] fixed tests for version of test library
---
helios/tests.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/helios/tests.py b/helios/tests.py
index 8413236..85d9b81 100644
--- a/helios/tests.py
+++ b/helios/tests.py
@@ -353,15 +353,17 @@ class WebTest(django_webtest.WebTest):
assert url in response.location
else:
assert url in response._headers['location'][1]
- self.assertEqual(response.status_code, 302)
+
+ if hasattr(response, 'status_code'):
+ assert response.status_code == 302
+ else:
+ assert response.status_int == 302
+
+ #self.assertEqual(response.status_code, 302)
#return super(django_webtest.WebTest, self).assertRedirects(response, url)
#if hasattr(response, 'status_code') and hasattr(response, 'location'):
- #if hasattr(response, 'status_code'):
- # assert response.status_code == 302
- #else:
- # assert response.status_int == 302
#assert url in response.location, "redirected to %s instead of %s" % (response.location, url)
--
GitLab