fix remaining assertRaises
This commit is contained in:
		| @@ -350,14 +350,11 @@ class TestInstance(MongoDBTestCase): | ||||
|             name = StringField() | ||||
|             meta = {"allow_inheritance": True} | ||||
|  | ||||
|         with pytest.raises(ValueError) as e: | ||||
|         with pytest.raises(ValueError, match="Cannot override primary key field") as e: | ||||
|  | ||||
|             class EmailUser(User): | ||||
|                 email = StringField(primary_key=True) | ||||
|  | ||||
|         exc = e.exception | ||||
|         assert str(exc) == "Cannot override primary key field" | ||||
|  | ||||
|     def test_custom_id_field_is_required(self): | ||||
|         """Ensure the custom primary key field is required.""" | ||||
|  | ||||
| @@ -365,10 +362,9 @@ class TestInstance(MongoDBTestCase): | ||||
|             username = StringField(primary_key=True) | ||||
|             name = StringField() | ||||
|  | ||||
|         with pytest.raises(ValidationError) as e: | ||||
|         with pytest.raises(ValidationError) as exc_info: | ||||
|             User(name="test").save() | ||||
|         exc = e.exception | ||||
|         assert "Field is required: ['username']" in str(exc) | ||||
|         assert "Field is required: ['username']" in str(exc_info.value) | ||||
|  | ||||
|     def test_document_not_registered(self): | ||||
|         class Place(Document): | ||||
| @@ -870,12 +866,12 @@ class TestInstance(MongoDBTestCase): | ||||
|  | ||||
|         t = TestDocument(doc=TestEmbeddedDocument(x=10, y=25, z=15)) | ||||
|  | ||||
|         with pytest.raises(ValidationError) as cm: | ||||
|         with pytest.raises(ValidationError) as exc_info: | ||||
|             t.save() | ||||
|  | ||||
|         expected_msg = "Value of z != x + y" | ||||
|         assert expected_msg in cm.exception.message | ||||
|         assert cm.exception.to_dict() == {"doc": {"__all__": expected_msg}} | ||||
|         assert expected_msg in str(exc_info.value) | ||||
|         assert exc_info.value.to_dict() == {"doc": {"__all__": expected_msg}} | ||||
|  | ||||
|         t = TestDocument(doc=TestEmbeddedDocument(x=10, y=25)).save() | ||||
|         assert t.doc.z == 35 | ||||
| @@ -3208,43 +3204,47 @@ class TestInstance(MongoDBTestCase): | ||||
|  | ||||
|     def test_positional_creation(self): | ||||
|         """Document cannot be instantiated using positional arguments.""" | ||||
|         with pytest.raises(TypeError) as e: | ||||
|         with pytest.raises(TypeError) as exc_info: | ||||
|             person = self.Person("Test User", 42) | ||||
|  | ||||
|         expected_msg = ( | ||||
|             "Instantiating a document with positional arguments is not " | ||||
|             "supported. Please use `field_name=value` keyword arguments." | ||||
|         ) | ||||
|         assert str(e.exception) == expected_msg | ||||
|         assert str(exc_info.value) == expected_msg | ||||
|  | ||||
|     def test_mixed_creation(self): | ||||
|         """Document cannot be instantiated using mixed arguments.""" | ||||
|         with pytest.raises(TypeError) as e: | ||||
|         with pytest.raises(TypeError) as exc_info: | ||||
|             person = self.Person("Test User", age=42) | ||||
|  | ||||
|         expected_msg = ( | ||||
|             "Instantiating a document with positional arguments is not " | ||||
|             "supported. Please use `field_name=value` keyword arguments." | ||||
|         ) | ||||
|         assert str(e.exception) == expected_msg | ||||
|         assert str(exc_info.value) == expected_msg | ||||
|  | ||||
|     def test_positional_creation_embedded(self): | ||||
|         """Embedded document cannot be created using positional arguments.""" | ||||
|         with pytest.raises(TypeError) as e: | ||||
|         with pytest.raises(TypeError) as exc_info: | ||||
|             job = self.Job("Test Job", 4) | ||||
|  | ||||
|         expected_msg = ( | ||||
|             "Instantiating a document with positional arguments is not " | ||||
|             "supported. Please use `field_name=value` keyword arguments." | ||||
|         ) | ||||
|         assert str(e.exception) == expected_msg | ||||
|         assert str(exc_info.value) == expected_msg | ||||
|  | ||||
|     def test_mixed_creation_embedded(self): | ||||
|         """Embedded document cannot be created using mixed arguments.""" | ||||
|         with pytest.raises(TypeError) as e: | ||||
|         with pytest.raises(TypeError) as exc_info: | ||||
|             job = self.Job("Test Job", years=4) | ||||
|  | ||||
|         expected_msg = ( | ||||
|             "Instantiating a document with positional arguments is not " | ||||
|             "supported. Please use `field_name=value` keyword arguments." | ||||
|         ) | ||||
|         assert str(e.exception) == expected_msg | ||||
|         assert str(exc_info.value) == expected_msg | ||||
|  | ||||
|     def test_data_contains_id_field(self): | ||||
|         """Ensure that asking for _data returns 'id'.""" | ||||
|   | ||||
		Reference in New Issue
	
	Block a user