Expand - add #[test] handler that removes all tests

1 files changed, 27 insertions, 0 deletions

diff --git a/src/expand/test.cpp b/src/expand/test.cpp
new file mode 100644
index 00000000..ba18fd24
--- /dev/null
+++ b/src/expand/test.cpp
@@ -0,0 +1,27 @@
+/*
+ * MRustC - Rust Compiler
+ * - By John Hodge (Mutabah/thePowersGang)
+ *
+ * expand/test.cpp
+ * - #[test] handling
+ */
+#include <synext_decorator.hpp>
+#include <ast/ast.hpp>
+
+class CTestHandler:
+    public ExpandDecorator
+{
+    AttrStage   stage() const override { return AttrStage::Pre; }
+    
+    void handle(const Span& sp, const AST::MetaItem& mi, ::AST::Crate& crate, const AST::Path& path, AST::Module& mod, AST::Item&i) const override {
+        if( ! i.is_Function() ) {
+            ERROR(sp, E0000, "#[test] can only be put on functions - found on " << i.tag_str());
+        }
+        
+        // TODO: Proper #[test] support, for now just remove them
+        i = AST::Item::make_None({});
+    }
+};
+
+STATIC_DECORATOR("test", CTestHandler);
+