Andrew's git - gitweb.git/blob - t/helper/test-revision-walking.c

   1/*
   2 * test-revision-walking.c: test revision walking API.
   3 *
   4 * (C) 2012 Heiko Voigt <hvoigt@hvoigt.net>
   5 *
   6 * This code is free software; you can redistribute it and/or modify
   7 * it under the terms of the GNU General Public License version 2 as
   8 * published by the Free Software Foundation.
   9 */
  10
  11#include "test-tool.h"
  12#include "cache.h"
  13#include "commit.h"
  14#include "diff.h"
  15#include "revision.h"
  16
  17static void print_commit(struct commit *commit)
  18{
  19        struct strbuf sb = STRBUF_INIT;
  20        struct pretty_print_context ctx = {0};
  21        ctx.date_mode.type = DATE_NORMAL;
  22        format_commit_message(commit, " %m %s", &sb, &ctx);
  23        printf("%s\n", sb.buf);
  24        strbuf_release(&sb);
  25}
  26
  27static int run_revision_walk(void)
  28{
  29        struct rev_info rev;
  30        struct commit *commit;
  31        const char *argv[] = {NULL, "--all", NULL};
  32        int argc = ARRAY_SIZE(argv) - 1;
  33        int got_revision = 0;
  34
  35        repo_init_revisions(the_repository, &rev, NULL);
  36        setup_revisions(argc, argv, &rev, NULL);
  37        if (prepare_revision_walk(&rev))
  38                die("revision walk setup failed");
  39
  40        while ((commit = get_revision(&rev)) != NULL) {
  41                print_commit(commit);
  42                got_revision = 1;
  43        }
  44
  45        reset_revision_walk();
  46        return got_revision;
  47}
  48
  49int cmd__revision_walking(int argc, const char **argv)
  50{
  51        if (argc < 2)
  52                return 1;
  53
  54        setup_git_directory();
  55
  56        if (!strcmp(argv[1], "run-twice")) {
  57                printf("1st\n");
  58                if (!run_revision_walk())
  59                        return 1;
  60                printf("2nd\n");
  61                if (!run_revision_walk())
  62                        return 1;
  63
  64                return 0;
  65        }
  66
  67        fprintf(stderr, "check usage\n");
  68        return 1;
  69}