aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--tests/test_relations.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_relations.py b/tests/test_relations.py
index a3672117..c29618ce 100644
--- a/tests/test_relations.py
+++ b/tests/test_relations.py
@@ -32,13 +32,13 @@ class TestPrimaryKeyRelatedField(APISimpleTestCase):
def test_pk_related_lookup_does_not_exist(self):
with pytest.raises(ValidationError) as excinfo:
self.field.to_internal_value(4)
- msg = excinfo.value.message
+ msg = excinfo.value.messages[0]
assert msg == "Invalid pk '4' - object does not exist."
def test_pk_related_lookup_invalid_type(self):
with pytest.raises(ValidationError) as excinfo:
self.field.to_internal_value(BadType())
- msg = excinfo.value.message
+ msg = excinfo.value.messages[0]
assert msg == 'Incorrect type. Expected pk value, received BadType.'
def test_pk_representation(self):
@@ -122,13 +122,13 @@ class TestSlugRelatedField(APISimpleTestCase):
def test_slug_related_lookup_does_not_exist(self):
with pytest.raises(ValidationError) as excinfo:
self.field.to_internal_value('doesnotexist')
- msg = excinfo.value.message
+ msg = excinfo.value.messages[0]
assert msg == 'Object with name=doesnotexist does not exist.'
def test_slug_related_lookup_invalid_type(self):
with pytest.raises(ValidationError) as excinfo:
self.field.to_internal_value(BadType())
- msg = excinfo.value.message
+ msg = excinfo.value.messages[0]
assert msg == 'Invalid value.'
def test_representation(self):