src/test/run-pass-fulldeps/issue-15149.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(unused_variables)]
  12 // no-prefer-dynamic
  13 // ignore-cross-compile
  14
  15 use std::env;
  16 use std::fs;
  17 use std::process;
  18 use std::str;
  19 use std::path::PathBuf;
  20
  21 fn main() {
  22     // If we're the child, make sure we were invoked correctly
  23     let args: Vec<String> = env::args().collect();
  24     if args.len() > 1 && args[1] == "child" {
  25         // FIXME: This should check the whole `args[0]` instead of just
  26         // checking that it ends_with the executable name. This
  27         // is needed because of Windows, which has a different behavior.
  28         // See #15149 for more info.
  29         return assert!(args[0].ends_with(&format!("mytest{}",
  30                                                   env::consts::EXE_SUFFIX)));
  31     }
  32
  33     test();
  34 }
  35
  36 fn test() {
  37     // If we're the parent, copy our own binary to a new directory.
  38     let my_path = env::current_exe().unwrap();
  39     let my_dir  = my_path.parent().unwrap();
  40
  41     let child_dir = PathBuf::from(env::var_os("RUST_TEST_TMPDIR").unwrap());
  42     let child_dir = child_dir.join("issue-15140-child");
  43     fs::create_dir_all(&child_dir).unwrap();
  44
  45     let child_path = child_dir.join(&format!("mytest{}",
  46                                              env::consts::EXE_SUFFIX));
  47     fs::copy(&my_path, &child_path).unwrap();
  48
  49     // Append the new directory to our own PATH.
  50     let path = {
  51         let mut paths: Vec<_> = env::split_paths(&env::var_os("PATH").unwrap()).collect();
  52         paths.push(child_dir.to_path_buf());
  53         env::join_paths(paths).unwrap()
  54     };
  55
  56     let child_output = process::Command::new("mytest").env("PATH", &path)
  57                                                       .arg("child")
  58                                                       .output().unwrap();
  59
  60     assert!(child_output.status.success(),
  61             format!("child assertion failed\n child stdout:\n {}\n child stderr:\n {}",
  62                     str::from_utf8(&child_output.stdout).unwrap(),
  63                     str::from_utf8(&child_output.stderr).unwrap()));
  64 }