src/test/auxiliary/lint_group_plugin_test.rs

   1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 // force-host
  12
  13 #![feature(plugin_registrar)]
  14 #![feature(box_syntax, rustc_private)]
  15
  16 extern crate syntax;
  17
  18 // Load rustc as a plugin to get macros
  19 #[macro_use]
  20 extern crate rustc;
  21
  22 use syntax::ast;
  23 use syntax::parse::token;
  24 use rustc::lint::{Context, LintPass, LintPassObject, LintArray};
  25 use rustc::plugin::Registry;
  26
  27 declare_lint!(TEST_LINT, Warn, "Warn about items named 'lintme'");
  28
  29 declare_lint!(PLEASE_LINT, Warn, "Warn about items named 'pleaselintme'");
  30
  31 struct Pass;
  32
  33 impl LintPass for Pass {
  34     fn get_lints(&self) -> LintArray {
  35         lint_array!(TEST_LINT, PLEASE_LINT)
  36     }
  37
  38     fn check_item(&mut self, cx: &Context, it: &ast::Item) {
  39         let name = token::get_ident(it.ident);
  40         if &name[..] == "lintme" {
  41             cx.span_lint(TEST_LINT, it.span, "item is named 'lintme'");
  42         } else if &name[..] == "pleaselintme" {
  43             cx.span_lint(PLEASE_LINT, it.span, "item is named 'pleaselintme'");
  44         }
  45     }
  46 }
  47
  48 #[plugin_registrar]
  49 pub fn plugin_registrar(reg: &mut Registry) {
  50     reg.register_lint_pass(box Pass as LintPassObject);
  51     reg.register_lint_group("lint_me", vec![TEST_LINT, PLEASE_LINT]);
  52 }