rz-test: Fix last line of diff when REGEXP_FILTER_OUT/_ERR is set

rizinorg · Dec 11, 2024 · 499ff25 · 499ff25
1 parent 19ff772
commit 499ff25
Showing 1 changed file with 10 additions and 11 deletions.
diff --git a/binrz/rz-test/rz-test.c b/binrz/rz-test/rz-test.c
@@ -755,14 +755,22 @@ static void *worker_th(RzTestState *state) {
 	return NULL;
 }
 
+static char *get_matched_str(const char *regexp, const char *str) {
+	RzStrBuf *match_str = rz_test_regex_full_match_str(regexp, str);
+	int len = rz_strbuf_length(match_str);
+	if (len && rz_strbuf_get(match_str)[len - 1] != '\n') { // empty matches are not changed
+		rz_strbuf_append(match_str, "\n");
+	}
+	return rz_strbuf_drain(match_str);
+}
+
 static void print_diff(const char *actual, const char *expected, const char *regexp) {
 	RzDiff *d = NULL;
 	char *uni = NULL;
 	const char *output = actual;
 
 	if (regexp) {
-		RzStrBuf *match_str = rz_test_regex_full_match_str(regexp, actual);
-		output = rz_strbuf_drain(match_str);
+		output = get_matched_str(regexp, actual);
 	}
 
 	d = rz_diff_lines_new(expected, output, NULL);
@@ -1236,15 +1244,6 @@ static void replace_cmd_kv_file(const char *path, ut64 line_begin, ut64 line_end
 	free(newc);
 }
 
-static char *get_matched_str(const char *regexp, const char *str) {
-	RzStrBuf *match_str = rz_test_regex_full_match_str(regexp, str);
-	int len = rz_strbuf_length(match_str);
-	if (len && rz_strbuf_get(match_str)[len - 1] != '\n') { // empty matches are not changed
-		rz_strbuf_append(match_str, "\n");
-	}
-	return rz_strbuf_drain(match_str);
-}
-
 static bool interact_fix_cmd(RzTestResultInfo *result, RzPVector /*<RzTestResultInfo *>*/ *fixup_results) {
 	assert(result->test->type == RZ_TEST_TYPE_CMD);
 	if (result->run_failed || result->proc_out->ret != 0) {