src/test/run-pass/issue-16739.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 #![allow(unknown_features)]
  12 #![feature(box_syntax)]
  13 #![feature(unboxed_closures, core)]
  14
  15 // Test that unboxing shim for calling rust-call ABI methods through a
  16 // trait box works and does not cause an ICE.
  17
  18 struct Foo { foo: u32 }
  19
  20 impl FnMut<()> for Foo {
  21     type Output = u32;
  22     extern "rust-call" fn call_mut(&mut self, _: ()) -> u32 { self.foo }
  23 }
  24
  25 impl FnMut<(u32,)> for Foo {
  26     type Output = u32;
  27     extern "rust-call" fn call_mut(&mut self, (x,): (u32,)) -> u32 { self.foo + x }
  28 }
  29
  30 impl FnMut<(u32,u32)> for Foo {
  31     type Output = u32;
  32     extern "rust-call" fn call_mut(&mut self, (x, y): (u32, u32)) -> u32 { self.foo + x + y }
  33 }
  34
  35 fn main() {
  36     let mut f = box Foo { foo: 42 } as Box<FnMut() -> u32>;
  37     assert_eq!(f.call_mut(()), 42);
  38
  39     let mut f = box Foo { foo: 40 } as Box<FnMut(u32) -> u32>;
  40     assert_eq!(f.call_mut((2,)), 42);
  41
  42     let mut f = box Foo { foo: 40 } as Box<FnMut(u32, u32) -> u32>;
  43     assert_eq!(f.call_mut((1, 1)), 42);
  44 }