Qualify message field for assert statements

Reviewed By: connernilsen Differential Revision: D59820631 fbshipit-source-id: c93ca78a90baf425d2b506281c467b5bb0f4da9e
facebook · Jul 16, 2024 · fbedd7a · fbedd7a
1 parent 65be73a
commit fbedd7a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/source/analysis/preprocessing.ml b/source/analysis/preprocessing.ml
@@ -742,7 +742,8 @@ module Qualify (Context : QualifyContext) = struct
             Assert
               {
                 Assert.test = qualify_expression ~qualify_strings:DoNotQualify ~scope test;
-                message;
+                message =
+                  Option.map message ~f:(qualify_expression ~qualify_strings:DoNotQualify ~scope);
                 origin;
               } )
       | Class ({ name; _ } as definition) ->

diff --git a/source/analysis/test/preprocessingTest.ml b/source/analysis/test/preprocessingTest.ml
@@ -450,6 +450,7 @@ let test_qualify_source _ =
   assert_qualify_statement "(b, b)" "(a, a)";
   assert_qualify_statement "-b" "-a";
   assert_qualify_statement "assert b" "assert a";
+  assert_qualify_statement "assert b, b" "assert a, a";
   assert_qualify_statement "del b" "del a";
   assert_qualify_statement
     "b = 1\nfor b in []: pass"