1 files changed, 33 insertions, 0 deletions
diff --git a/build/clang-plugin/tests/TestKnownLive.cpp b/build/clang-plugin/tests/TestKnownLive.cpp
new file mode 100644
index 0000000000..8d01160d1d
--- /dev/null
+++ b/build/clang-plugin/tests/TestKnownLive.cpp
@@ -0,0 +1,33 @@
+#include <mozilla/RefPtr.h>
+
+#define MOZ_KNOWN_LIVE __attribute__((annotate("moz_known_live")))
+
+class Foo {
+  // dummy refcounting
+public:
+  uint32_t AddRef() { return 0; }
+  uint32_t Release() { return 0; }
+
+private:
+  ~Foo() = default;
+};
+
+class Bar {
+  MOZ_KNOWN_LIVE RefPtr<Foo> mFoo;
+  Bar() : mFoo(new Foo()) {}
+  ~Bar() { mFoo = nullptr; }
+
+  void Baz() {
+    mFoo = nullptr; // expected-error {{MOZ_KNOWN_LIVE members can only be modified by constructors and destructors}}
+  }
+};
+
+class Bar2 {
+  MOZ_KNOWN_LIVE Foo *mFoo;
+  Bar2() : mFoo(new Foo()) {}
+  ~Bar2() { mFoo = nullptr; }
+
+  void Baz() {
+    mFoo = nullptr; // expected-error {{MOZ_KNOWN_LIVE members can only be modified by constructors and destructors}}
+  }
+};