src/test/run-pass/overloaded-index-autoderef.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 // Test overloaded indexing combined with autoderef.
  12
  13 #![allow(unknown_features)]
  14 #![feature(box_syntax, core)]
  15
  16 use std::ops::{Index, IndexMut};
  17
  18 struct Foo {
  19     x: int,
  20     y: int,
  21 }
  22
  23 impl Index<int> for Foo {
  24     type Output = int;
  25
  26     fn index(&self, z: &int) -> &int {
  27         if *z == 0 {
  28             &self.x
  29         } else {
  30             &self.y
  31         }
  32     }
  33 }
  34
  35 impl IndexMut<int> for Foo {
  36     fn index_mut(&mut self, z: &int) -> &mut int {
  37         if *z == 0 {
  38             &mut self.x
  39         } else {
  40             &mut self.y
  41         }
  42     }
  43 }
  44
  45 trait Int {
  46     fn get(self) -> int;
  47     fn get_from_ref(&self) -> int;
  48     fn inc(&mut self);
  49 }
  50
  51 impl Int for int {
  52     fn get(self) -> int { self }
  53     fn get_from_ref(&self) -> int { *self }
  54     fn inc(&mut self) { *self += 1; }
  55 }
  56
  57 fn main() {
  58     let mut f: Box<_> = box Foo {
  59         x: 1,
  60         y: 2,
  61     };
  62
  63     assert_eq!(f[1], 2);
  64
  65     f[0] = 3;
  66
  67     assert_eq!(f[0], 3);
  68
  69     // Test explicit IndexMut where `f` must be autoderef:
  70     {
  71         let p = &mut f[1];
  72         *p = 4;
  73     }
  74
  75     // Test explicit Index where `f` must be autoderef:
  76     {
  77         let p = &f[1];
  78         assert_eq!(*p, 4);
  79     }
  80
  81     // Test calling methods with `&mut self`, `self, and `&self` receivers:
  82     f[1].inc();
  83     assert_eq!(f[1].get(), 5);
  84     assert_eq!(f[1].get_from_ref(), 5);
  85 }