diff --git a/test/main/all-arg-test b/test/main/all-arg-test index a2e026510..b7327046a 100755 --- a/test/main/all-arg-test +++ b/test/main/all-arg-test @@ -21,7 +21,7 @@ in_work_dir create_repo_from_tgz "$base_dir/files/refs-repo.tgz" test_tig --all -assert_vars +assert_vars 1 assert_equals stderr <> "$HOME/$vars_file" -printf '%s\\n' "\$rhs" >> "$expected_var_file" +printf '%s\\n' "\$rhs" >> "$expected_vars_file" EOF assert_vars() { - if [ -e "$expected_var_file" ]; then - assert_equals "$vars_file" strict "message" < "$expected_var_file" + if [ -n "${1:-}" ]; then + printf '%s\n' "$1" > "$vars_count_file" + shift + else + die "Test must supply the expected count of assertions to assert_vars()" + fi + + grep -c . "$vars_file" | assert_equals "$vars_count_file" strict "$*" + + if [ -e "$expected_vars_file" ]; then + assert_equals "$vars_file" strict "$*" < "$expected_vars_file" else - printf '[FAIL] %s not found\n' "$expected_var_file" >> .test-result + printf '[FAIL] %s not found\n' "$expected_vars_file" >> .test-result fi }