src/test/run-pass/process-spawn-with-unicode-params.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 // no-prefer-dynamic
  12
  13 // The test copies itself into a subdirectory with a non-ASCII name and then
  14 // runs it as a child process within the subdirectory.  The parent process
  15 // also adds an environment variable and an argument, both containing
  16 // non-ASCII characters.  The child process ensures all the strings are
  17 // intact.
  18
  19 extern crate native;
  20
  21 use std::io;
  22 use std::io::fs;
  23 use std::io::Command;
  24 use std::os;
  25 use std::path::Path;
  26
  27 fn main() {
  28     let my_args = os::args();
  29     let my_cwd  = os::getcwd();
  30     let my_env  = os::env();
  31     let my_path = Path::new(os::self_exe_name().unwrap());
  32     let my_dir  = my_path.dir_path();
  33     let my_ext  = my_path.extension_str().unwrap_or("");
  34
  35     // some non-ASCII characters
  36     let blah       = "\u03c0\u042f\u97f3\u00e6\u221e";
  37
  38     let child_name = "child";
  39     let child_dir  = format_strbuf!("process-spawn-with-unicode-params-{}",
  40                                     blah);
  41
  42     // parameters sent to child / expected to be received from parent
  43     let arg = blah;
  44     let cwd = my_dir.join(Path::new(child_dir.clone()));
  45     let env = ("RUST_TEST_PROC_SPAWN_UNICODE".to_string(), blah.to_string());
  46
  47     // am I the parent or the child?
  48     if my_args.len() == 1 {             // parent
  49
  50         let child_filestem = Path::new(child_name);
  51         let child_filename = child_filestem.with_extension(my_ext);
  52         let child_path     = cwd.join(child_filename.clone());
  53
  54         // make a separate directory for the child
  55         drop(fs::mkdir(&cwd, io::UserRWX).is_ok());
  56         assert!(fs::copy(&my_path, &child_path).is_ok());
  57
  58         // run child
  59         let p = Command::new(&child_path)
  60                         .arg(arg)
  61                         .cwd(&cwd)
  62                         .env(my_env.append_one(env).as_slice())
  63                         .spawn().unwrap().wait_with_output().unwrap();
  64
  65         // display the output
  66         assert!(io::stdout().write(p.output.as_slice()).is_ok());
  67         assert!(io::stderr().write(p.error.as_slice()).is_ok());
  68
  69         // make sure the child succeeded
  70         assert!(p.status.success());
  71
  72     } else {                            // child
  73
  74         // check working directory (don't try to compare with `cwd` here!)
  75         assert!(my_cwd.ends_with_path(&Path::new(child_dir)));
  76
  77         // check arguments
  78         assert_eq!(my_args.get(1).as_slice(), arg);
  79
  80         // check environment variable
  81         assert!(my_env.contains(&env));
  82
  83     };
  84 }