Auto merge of #2795 - alexcrichton:test-and-harness, r=brson

bors · web-flow · commit 80f2775c0730 · 2016-06-20T18:37:37.000-07:00
Fix `harness = false` on `[lib]` sections Now that this is fixed upstream, we can actually add a test for this! Closes #2305
diff --git a/src/cargo/ops/cargo_rustc/mod.rs b/src/cargo/ops/cargo_rustc/mod.rs
@@ -467,8 +467,10 @@ fn build_base_args(cx: &Context,
 
     cmd.arg("--crate-name").arg(&unit.target.crate_name());
 
-    for crate_type in crate_types.iter() {
-        cmd.arg("--crate-type").arg(crate_type);
+    if !test {
+        for crate_type in crate_types.iter() {
+            cmd.arg("--crate-type").arg(crate_type);
+        }
     }
 
     let prefer_dynamic = (unit.target.for_host() &&
@@ -515,6 +517,8 @@ fn build_base_args(cx: &Context,
 
     if test && unit.target.harness() {
         cmd.arg("--test");
+    } else if test {
+        cmd.arg("--cfg").arg("test");
     }
 
     if let Some(features) = cx.resolve.features(unit.pkg.package_id()) {
diff --git a/tests/test.rs b/tests/test.rs
@@ -2161,3 +2161,35 @@ fn test_panic_abort_with_dep() {
     assert_that(p.cargo_process("test").arg("-v"),
                 execs().with_status(0));
 }
+
+#[test]
+fn cfg_test_even_with_no_harness() {
+    if !is_nightly() {
+        return
+    }
+    let p = project("foo")
+        .file("Cargo.toml", r#"
+            [package]
+            name = "foo"
+            version = "0.0.1"
+            authors = []
+
+            [lib]
+            harness = false
+            doctest = false
+        "#)
+        .file("src/lib.rs", r#"
+            #[cfg(test)]
+            fn main() {
+                println!("hello!");
+            }
+        "#);
+    assert_that(p.cargo_process("test").arg("-v"),
+                execs().with_status(0)
+                       .with_stdout("hello!\n")
+                       .with_stderr("\
+[COMPILING] foo v0.0.1 ([..])
+[RUNNING] `rustc [..]`
+[RUNNING] `[..]`
+"));
+}
