add new mode

dbt-labs · May 17, 2024 · 36d4f99 · 36d4f99
1 parent 341803d
commit 36d4f99
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/dbt/cli/requires.py b/core/dbt/cli/requires.py
@@ -107,14 +107,16 @@ def setup_record_replay():
         recorder = Recorder(RecorderMode.REPLAY, recording_path)
     elif rec_mode == RecorderMode.RECORD:
         recorder = Recorder(RecorderMode.RECORD)
+    elif rec_mode == RecorderMode.RECORD_QUERIES:
+        recorder = Recorder(RecorderMode.RECORD_QUERIES)
 
     get_invocation_context().recorder = recorder
 
 
 def tear_down_record_replay():
     recorder = get_invocation_context().recorder
     if recorder is not None:
-        if recorder.mode == RecorderMode.RECORD:
+        if recorder.mode == RecorderMode.RECORD or recorder.mode == RecorderMode.RECORD_QUERIES:
             recorder.write("recording.json")
         elif recorder.mode == RecorderMode.REPLAY:
             recorder.write_diffs("replay_diffs.json")