From 6b8b00f4b58bd1e42251e49bc51fd4ffec9787b7 Mon Sep 17 00:00:00 2001 From: James Bowes Date: Sun, 8 Jun 2008 11:56:11 -0400 Subject: [PATCH] Clean up some segfaults in the test driver --- test-driver.c | 7 ++++++- 1 files changed, 6 insertions(+), 1 deletions(-) diff --git a/test-driver.c b/test-driver.c index 8ff184e..54dbde7 100644 --- a/test-driver.c +++ b/test-driver.c @@ -262,8 +262,12 @@ end_transaction(struct test_context *ctx) razor_transaction_install_package(ctx->trans, pkg); } for (i = 0; i < ctx->n_remove_pkgs; i++) { - pkg = razor_set_get_package(ctx->repo_set, + pkg = razor_set_get_package(ctx->system_set, ctx->remove_pkgs[i]); + if (!pkg) + pkg = razor_set_get_package(ctx->repo_set, + ctx->remove_pkgs[i]); + razor_transaction_remove_package(ctx->trans, pkg); } @@ -278,6 +282,7 @@ end_transaction(struct test_context *ctx) if (!errors) { struct razor_set *new; new = razor_transaction_finish(ctx->trans); + ctx->trans = NULL; ctx->system_set = new; } } -- 1.7.1