fix specialIdGen to avoid reserved words. Solves #151

statebox · Jan 21, 2020 · 6354b99 · 6354b99
1 parent f969b85
commit 6354b99
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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]