src/test/run-pass/task-comm-7.rs

   1 // Copyright 2012-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(dead_assignment)]
  12
  13 use std::sync::mpsc::{channel, Sender};
  14 use std::task;
  15
  16 pub fn main() { test00(); }
  17
  18 fn test00_start(c: &Sender<int>, start: int,
  19                 number_of_messages: int) {
  20     let mut i: int = 0;
  21     while i < number_of_messages { c.send(start + i).unwrap(); i += 1; }
  22 }
  23
  24 fn test00() {
  25     let mut r: int = 0;
  26     let mut sum: int = 0;
  27     let (tx, rx) = channel();
  28     let number_of_messages: int = 10;
  29
  30     let tx2 = tx.clone();
  31     task::spawn(move|| {
  32         test00_start(&tx2, number_of_messages * 0, number_of_messages);
  33     });
  34     let tx2 = tx.clone();
  35     task::spawn(move|| {
  36         test00_start(&tx2, number_of_messages * 1, number_of_messages);
  37     });
  38     let tx2 = tx.clone();
  39     task::spawn(move|| {
  40         test00_start(&tx2, number_of_messages * 2, number_of_messages);
  41     });
  42     let tx2 = tx.clone();
  43     task::spawn(move|| {
  44         test00_start(&tx2, number_of_messages * 3, number_of_messages);
  45     });
  46
  47     let mut i: int = 0;
  48     while i < number_of_messages {
  49         r = rx.recv().unwrap();
  50         sum += r;
  51         r = rx.recv().unwrap();
  52         sum += r;
  53         r = rx.recv().unwrap();
  54         sum += r;
  55         r = rx.recv().unwrap();
  56         sum += r;
  57         i += 1;
  58     }
  59
  60     assert_eq!(sum, number_of_messages * 4 * (number_of_messages * 4 - 1) / 2);
  61 }