Some helpful debug functions

Reviewed By: simonmar Differential Revision: D68561917 fbshipit-source-id: e8cf0841afe4e0475a1cf64da5150e8c8ce3c7ec
facebookincubator · Jan 23, 2025 · 48f2bdb · 48f2bdb
1 parent 7824af3
commit 48f2bdb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/glean/test/lib/Glean/Database/Test.hs b/glean/test/lib/Glean/Database/Test.hs
@@ -19,6 +19,8 @@ module Glean.Database.Test
   , setDBVersion
   , setCompactOnCompletion
   , setMaxSetSize
+  , enableTcDebug
+  , enableQueryDebug
   , withTestEnv
   , kickOffTestDB
   , waitUntilComplete
@@ -105,6 +107,14 @@ setMaxSetSize i cfg = cfg
   { cfgServerConfig = cfgServerConfig cfg <&> \scfg -> scfg
       { ServerConfig.config_max_set_size_bytes = Just i } }
 
+enableTcDebug :: Setting
+enableTcDebug cfg = cfg
+  { cfgDebug = (cfgDebug cfg) { tcDebug = True } }
+
+enableQueryDebug :: Setting
+enableQueryDebug cfg = cfg
+  { cfgDebug = (cfgDebug cfg) { queryDebug = True } }
+
 withTestEnv
   :: [Setting]
   -> (Env -> IO a)