More failing tests

Author: aldexis

baseline-error-prone/src/test/java/com/palantir/baseline/errorprone/IllegalSafeLoggingArgumentTest.java

@@ -1383,6 +1383,49 @@ public void disagreeingSafetyAnnotations() {
                 .doTest();
     }
 
+    @Test
+    public void testDisagreeingSafetyAnnotations_argument() {
+        helper().addSourceLines(
+                        "Test.java",
+                        // language=Java
+                        """
+                        import com.palantir.logsafe.*;
+                        import java.util.function.*;
+
+                        @Unsafe
+                        class MyObject {}
+
+                        class Test {
+                          // BUG: Diagnostic contains: Dangerous
+                          void f(@Safe Object o) {}
+                        }
+                        """)
+                .doTest();
+    }
+
+    @Test
+    public void testDisagreeingSafetyAnnotations_return() {
+        helper().addSourceLines(
+                        "Test.java",
+                        // language=Java
+                        """
+                        import com.palantir.logsafe.*;
+                        import java.util.function.*;
+
+                        @Unsafe
+                        class MyObject {}
+
+                        class Test {
+                          // BUG: Diagnostic contains: Dangerous
+                          @Safe
+                          MyObject g() {
+                            return null;
+                          }
+                        }
+                        """)
+                .doTest();
+    }
+
     @Test
     public void testOptionalUnwrapping() {
         helper().addSourceLines(