annotate contrib/merge-lists/tests/test-merge-lists.rs @ 49011:b999edb15f8c

merge-lists: make it possible to specify pattern to match

The `merge-lists` tool doesn't know anything about Python other than
its regex that attempts to match import lines. Let's make it possible
to pass in a custom regex so it's easy to use the tool for e.g. C/C++
`#include` lines or Rust `use` lines (given the limited).

Differential Revision: https://phab.mercurial-scm.org/D12392

use similar::DiffableStr;
use std::ffi::OsStr;
use tempdir::TempDir;

fn run_test(arg: &str, input: &str) -> String {
    let mut cmd = assert_cmd::Command::cargo_bin("merge-lists").unwrap();
    let temp_dir = TempDir::new("test").unwrap();
    let base_path = temp_dir.path().join("base");
    let local_path = temp_dir.path().join("local");
    let other_path = temp_dir.path().join("other");

    let rest = input.strip_prefix("\nbase:\n").unwrap();
    let mut split = rest.split("\nlocal:\n");
    std::fs::write(&base_path, split.next().unwrap()).unwrap();
    let rest = split.next().unwrap();
    let mut split = rest.split("\nother:\n");
    std::fs::write(&local_path, split.next().unwrap()).unwrap();
    std::fs::write(&other_path, split.next().unwrap()).unwrap();
    cmd.args(&[
        OsStr::new(arg),
        local_path.as_os_str(),
        base_path.as_os_str(),
        other_path.as_os_str(),
    ])
    .assert()
    .success();

    let new_base_bytes = std::fs::read(&base_path).unwrap();
    let new_local_bytes = std::fs::read(&local_path).unwrap();
    let new_other_bytes = std::fs::read(&other_path).unwrap();
    // No newline before "base:" because of https://github.com/mitsuhiko/insta/issues/117
    format!(
        "base:\n{}\nlocal:\n{}\nother:\n{}",
        new_base_bytes.as_str().unwrap(),
        new_local_bytes.as_str().unwrap(),
        new_other_bytes.as_str().unwrap()
    )
}

#[test]
fn test_merge_lists_basic() {
    let output = run_test(
        "--python-imports",
        r"
base:
import lib1
import lib2

local:
import lib2
import lib3

other:
import lib3
import lib4
",
    );
    insta::assert_snapshot!(output, @r###"
    base:
    import lib3
    import lib4

    local:
    import lib3
    import lib4

    other:
    import lib3
    import lib4
    "###);
}

#[test]
fn test_merge_lists_from() {
    // Test some "from x import y" statements and some non-import conflicts
    // (unresolvable)
    let output = run_test(
        "--python-imports",
        r"
base:
from . import x

1+1

local:
from . import x
from a import b

2+2

other:
from a import c

3+3
",
    );
    insta::assert_snapshot!(output, @r###"
    base:
    from a import b
    from a import c

    1+1

    local:
    from a import b
    from a import c

    2+2

    other:
    from a import b
    from a import c

    3+3
    "###);
}

#[test]
fn test_merge_lists_not_sorted() {
    // Test that nothing is done if the elements in the conflicting hunks are
    // not sorted
    let output = run_test(
        "--python-imports",
        r"
base:
import x

1+1

local:
import a
import x

2+2

other:
import z
import y

3+3
",
    );
    insta::assert_snapshot!(output, @r###"
    base:
    import x

    1+1

    local:
    import a
    import x

    2+2

    other:
    import z
    import y

    3+3
    "###);
}

#[test]
fn test_custom_regex() {
    // Test merging of all lines (by matching anything)
    let output = run_test(
        "--pattern=.*",
        r"
base:
aardvark
baboon
camel

local:
aardvark
camel
eagle

other:
aardvark
camel
deer
",
    );
    insta::assert_snapshot!(output, @r###"
    base:
    aardvark
    camel
    deer
    eagle

    local:
    aardvark
    camel
    deer
    eagle

    other:
    aardvark
    camel
    deer
    eagle
    "###);
}