src/test/run-pass/macro-at-most-once-rep.rs

   1 // Copyright 2012 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 // The logic for parsing Kleene operators in macros has a special case to disambiguate `?`.
  12 // Specifically, `$(pat)?` is the ZeroOrOne operator whereas `$(pat)?+` or `$(pat)?*` are the
  13 // ZeroOrMore and OneOrMore operators using `?` as a separator. These tests are intended to
  14 // exercise that logic in the macro parser.
  15 //
  16 // Moreover, we also throw in some tests for using a separator with `?`, which is meaningless but
  17 // included for consistency with `+` and `*`.
  18 //
  19 // This test focuses on non-error cases and making sure the correct number of repetitions happen.
  20
  21 #![feature(macro_at_most_once_rep)]
  22
  23 macro_rules! foo {
  24     ($($a:ident)? ; $num:expr) => { {
  25         let mut x = 0;
  26
  27         $(
  28             x += $a;
  29          )?
  30
  31         assert_eq!(x, $num);
  32     } }
  33 }
  34
  35 macro_rules! barplus {
  36     ($($a:ident)?+ ; $num:expr) => { {
  37         let mut x = 0;
  38
  39         $(
  40             x += $a;
  41          )?
  42
  43         assert_eq!(x, $num);
  44     } }
  45 }
  46
  47 macro_rules! barstar {
  48     ($($a:ident)?* ; $num:expr) => { {
  49         let mut x = 0;
  50
  51         $(
  52             x += $a;
  53          )?
  54
  55         assert_eq!(x, $num);
  56     } }
  57 }
  58
  59 pub fn main() {
  60     let a = 1;
  61
  62     // accept 0 or 1 repetitions
  63     foo!( ; 0);
  64     foo!(a ; 1);
  65
  66     // Make sure using ? as a separator works as before
  67     barplus!(+ ; 0);
  68     barplus!(a + ; 1);
  69     barstar!(* ; 0);
  70     barstar!(a * ; 1);
  71 }