diff --git a/set.c b/set.c index 6464991..d52e039 100644 --- a/set.c +++ b/set.c @@ -109,7 +109,13 @@ int main(int argc, char** argv) printf("\n\n%s", BANNER); #endif /* ifndef NO_BANNER */ - + + if (!set_up()) + { + fprintf(stderr, "%s\n", "Setup call returned with error. Aborting execution."); + return 1; + } + int counter = 0; set_bundle_suits(NULL, &counter, true); @@ -123,4 +129,11 @@ int main(int argc, char** argv) dispatch_test_suits(suits, counter); free(suits); + + if (!tear_down()) + { + fprintf(stderr, "%s\n", "Teardown returned with error."); + return 1; + } + return 0; } diff --git a/set.h b/set.h index b72aa09..67de8be 100644 --- a/set.h +++ b/set.h @@ -26,6 +26,13 @@ char *format_string(const char *fmt, ...); // To be implemented by user. void set_bundle_suits(struct SETSuit **suits, int *counter, bool count); +bool set_up(); + +bool tear_down(); + +#define SETUP() bool set_up() +#define TEAR_DOWN() bool tear_down() + #define BUNDLE() \ void set_bundle_suits(struct SETSuit **suits, int *counter, bool count)