diff --git a/test/Language/CQL/Parser/Generator.hs b/test/Language/CQL/Parser/Generator.hs index 4763ce2..b558484 100644 --- a/test/Language/CQL/Parser/Generator.hs +++ b/test/Language/CQL/Parser/Generator.hs @@ -48,7 +48,10 @@ lowerIdGen = (`notElem` reservedWords) specialIdGen :: Gen String -specialIdGen = (:) <$> idCharGen <*> listOf (oneof [idCharGen, digitCharGen]) +specialIdGen = + ((:) <$> idCharGen <*> + listOf (oneof [idCharGen, digitCharGen])) `suchThat` + (`notElem` reservedWords) identifierGen :: Gen String identifierGen = oneof [lowerIdGen, upperIdGen, specialIdGen]