98 lines
3.0 KiB
C
98 lines
3.0 KiB
C
/*
|
|
* A test helper to exercise the progress display.
|
|
*
|
|
* Reads instructions from standard input, one instruction per line:
|
|
*
|
|
* "start <total>[ <title>]" - Call start_progress(title, total),
|
|
* Uses the default title of "Working hard"
|
|
* if the " <title>" is omitted.
|
|
* "progress <items>" - Call display_progress() with the given item count
|
|
* as parameter.
|
|
* "throughput <bytes> <millis> - Call display_throughput() with the given
|
|
* byte count as parameter. The 'millis'
|
|
* specify the time elapsed since the
|
|
* start_progress() call.
|
|
* "update" - Set the 'progress_update' flag.
|
|
* "stop" - Call stop_progress().
|
|
*
|
|
* See 't0500-progress-display.sh' for examples.
|
|
*/
|
|
#define GIT_TEST_PROGRESS_ONLY
|
|
#include "test-tool.h"
|
|
#include "parse-options.h"
|
|
#include "progress.h"
|
|
#include "strbuf.h"
|
|
#include "string-list.h"
|
|
|
|
int cmd__progress(int argc, const char **argv)
|
|
{
|
|
const char *const default_title = "Working hard";
|
|
struct string_list titles = STRING_LIST_INIT_DUP;
|
|
struct strbuf line = STRBUF_INIT;
|
|
struct progress *progress = NULL;
|
|
|
|
const char *usage[] = {
|
|
"test-tool progress <stdin",
|
|
NULL
|
|
};
|
|
struct option options[] = {
|
|
OPT_END(),
|
|
};
|
|
|
|
argc = parse_options(argc, argv, NULL, options, usage, 0);
|
|
if (argc)
|
|
usage_with_options(usage, options);
|
|
|
|
progress_testing = 1;
|
|
while (strbuf_getline(&line, stdin) != EOF) {
|
|
char *end;
|
|
|
|
if (skip_prefix(line.buf, "start ", (const char **) &end)) {
|
|
uint64_t total = strtoull(end, &end, 10);
|
|
const char *title;
|
|
|
|
/*
|
|
* We can't use "end + 1" as an argument to
|
|
* start_progress(), it doesn't xstrdup() its
|
|
* "title" argument. We need to hold onto a
|
|
* valid "char *" for it until the end.
|
|
*/
|
|
if (!*end)
|
|
title = default_title;
|
|
else if (*end == ' ')
|
|
title = string_list_insert(&titles, end + 1)->string;
|
|
else
|
|
die("invalid input: '%s'", line.buf);
|
|
|
|
progress = start_progress(title, total);
|
|
} else if (skip_prefix(line.buf, "progress ", (const char **) &end)) {
|
|
uint64_t item_count = strtoull(end, &end, 10);
|
|
if (*end != '\0')
|
|
die("invalid input: '%s'", line.buf);
|
|
display_progress(progress, item_count);
|
|
} else if (skip_prefix(line.buf, "throughput ",
|
|
(const char **) &end)) {
|
|
uint64_t byte_count, test_ms;
|
|
|
|
byte_count = strtoull(end, &end, 10);
|
|
if (*end != ' ')
|
|
die("invalid input: '%s'", line.buf);
|
|
test_ms = strtoull(end + 1, &end, 10);
|
|
if (*end != '\0')
|
|
die("invalid input: '%s'", line.buf);
|
|
progress_test_ns = test_ms * 1000 * 1000;
|
|
display_throughput(progress, byte_count);
|
|
} else if (!strcmp(line.buf, "update")) {
|
|
progress_test_force_update();
|
|
} else if (!strcmp(line.buf, "stop")) {
|
|
stop_progress(&progress);
|
|
} else {
|
|
die("invalid input: '%s'", line.buf);
|
|
}
|
|
}
|
|
strbuf_release(&line);
|
|
string_list_clear(&titles, 0);
|
|
|
|
return 0;
|
|
}
|