src/librustc_typeck/variance/test.rs

   1 use rustc::ty::TyCtxt;
   2 use rustc_errors::struct_span_err;
   3 use rustc_hir as hir;
   4 use rustc_hir::itemlikevisit::ItemLikeVisitor;
   5 use rustc_span::symbol::sym;
   6
   7 use rustc_error_codes::*;
   8
   9 pub fn test_variance(tcx: TyCtxt<'_>) {
  10     tcx.hir().krate().visit_all_item_likes(&mut VarianceTest { tcx });
  11 }
  12
  13 struct VarianceTest<'tcx> {
  14     tcx: TyCtxt<'tcx>,
  15 }
  16
  17 impl ItemLikeVisitor<'tcx> for VarianceTest<'tcx> {
  18     fn visit_item(&mut self, item: &'tcx hir::Item<'tcx>) {
  19         let item_def_id = self.tcx.hir().local_def_id(item.hir_id);
  20
  21         // For unit testing: check for a special "rustc_variance"
  22         // attribute and report an error with various results if found.
  23         if self.tcx.has_attr(item_def_id, sym::rustc_variance) {
  24             let variances_of = self.tcx.variances_of(item_def_id);
  25             struct_span_err!(self.tcx.sess, item.span, E0208, "{:?}", variances_of).emit();
  26         }
  27     }
  28
  29     fn visit_trait_item(&mut self, _: &'tcx hir::TraitItem<'tcx>) {}
  30     fn visit_impl_item(&mut self, _: &'tcx hir::ImplItem<'tcx>) {}
  31 }