chore: Use Hypothesis' InvalidArgument exception when an invalid input is passed to the generator functions

Author: Stranger6667

CHANGELOG.md

@@ -6,6 +6,10 @@
 
 - The `from_schema` function which takes a GraphQL schema and returns a Hypothesis strategy for defined queries and mutations.
 
+### Changed
+
+- Use Hypothesis' `InvalidArgument` exception when an invalid input is passed to the generator functions.
+
 ### Removed
 
 - `hypothesis_graphql.schemas` as it is not complete and not tested well.

src/hypothesis_graphql/_strategies/strategy.py

@@ -374,7 +374,7 @@ def queries(
     """
     parsed_schema = validation.maybe_parse_schema(schema)
     if parsed_schema.query_type is None:
-        raise ValueError("Query type is not defined in the schema")
+        raise InvalidArgument("Query type is not defined in the schema")
     return (
         _make_strategy(parsed_schema, type_=parsed_schema.query_type, fields=fields, custom_scalars=custom_scalars)
         .map(make_query)
@@ -401,7 +401,7 @@ def mutations(
     """
     parsed_schema = validation.maybe_parse_schema(schema)
     if parsed_schema.mutation_type is None:
-        raise ValueError("Mutation type is not defined in the schema")
+        raise InvalidArgument("Mutation type is not defined in the schema")
     return (
         _make_strategy(parsed_schema, type_=parsed_schema.mutation_type, fields=fields, custom_scalars=custom_scalars)
         .map(make_mutation)
@@ -454,5 +454,5 @@ def from_schema(
         if type_ is not None and (type_fields is None or len(type_fields) > 0)
     ]
     if not strategies:
-        raise ValueError("Query or Mutation type must be provided")
+        raise InvalidArgument("Query or Mutation type must be provided")
     return reduce(or_, strategies)

test/test_from_schema.py

@@ -1,6 +1,7 @@
 import pytest
 from hypothesis import given
 from hypothesis import strategies as st
+from hypothesis.errors import InvalidArgument
 
 from hypothesis_graphql import from_schema
 
@@ -41,5 +42,5 @@ def test_from_schema(data, schema, validate_operation, types, available_fields):
 
 
 def test_no_query_no_mutation(schema, validate_operation):
-    with pytest.raises(ValueError, match="Query or Mutation type must be provided"):
+    with pytest.raises(InvalidArgument, match="Query or Mutation type must be provided"):
         from_schema(schema)

test/test_mutations.py

@@ -1,6 +1,7 @@
 import pytest
 from hypothesis import given
 from hypothesis import strategies as st
+from hypothesis.errors import InvalidArgument
 
 from hypothesis_graphql import mutations
 
@@ -33,5 +34,5 @@ def test_mutation(data, mutation, validate_operation, fields):
 
 
 def test_no_mutation(schema):
-    with pytest.raises(ValueError, match="Mutation type is not defined in the schema"):
+    with pytest.raises(InvalidArgument, match="Mutation type is not defined in the schema"):
         mutations(schema)

test/test_queries.py

@@ -3,6 +3,7 @@
 from graphql import GraphQLNamedType
 from hypothesis import assume, find, given, settings
 from hypothesis import strategies as st
+from hypothesis.errors import InvalidArgument
 
 from hypothesis_graphql import nodes, queries
 from hypothesis_graphql._strategies.strategy import GraphQLStrategy
@@ -172,7 +173,7 @@ def test_missing_query():
     schema = """type Author {
       name: String
     }"""
-    with pytest.raises(ValueError, match="Query type is not defined in the schema"):
+    with pytest.raises(InvalidArgument, match="Query type is not defined in the schema"):
         queries(schema)