END_TEST
START_TEST(cleanup_single) {
// Test cleanup before we initialize it
cleanup_run(cleanup_func);
// Test cleanup it self
int one = 1, two = 2;
cleaned = 2;
cleanup_register(cleanup_func, &one);
cleanup_register(cleanup_func, &two);
cleanup_run(cleanup_func);
ck_assert_int_eq(1, cleaned);
cleanup_run(cleanup_func);
ck_assert_int_eq(0, cleaned);
// Both functions should be unregisterd so this should fail
ck_assert(!cleanup_unregister(cleanup_func));
// Check if we don't fail
cleanup_run(cleanup_func);
}
int main(int argc, char** argv)
{
const char *tmp = getenv ("TMPDIR");
char file[PATH_MAX];
char dir[PATH_MAX];
char dir2[PATH_MAX];
struct stat sb;
int fd, len;
plan (NO_PLAN);
/* Independent file and dir
*/
len = snprintf (file, sizeof (file), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (file)))
BAIL_OUT ("snprintf failed creating tmp file path");
if (!(fd = mkstemp (file)))
BAIL_OUT ("could not create tmp file");
close (fd);
len = snprintf (dir, sizeof (dir), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (dir)))
BAIL_OUT ("snprintf failed creating tmp directory");
if (!mkdtemp (dir))
BAIL_OUT ("could not create tmp directory");
cleanup_push_string (cleanup_file, file);
cleanup_push_string (cleanup_directory, dir);
cleanup_run ();
ok (stat (file, &sb) < 0 && errno == ENOENT,
"cleanup removed independent file");
ok (stat (dir, &sb) < 0 && errno == ENOENT,
"cleanup removed independent dir");
/* This time put file inside directory
*/
len = snprintf (dir, sizeof (dir), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (dir)))
BAIL_OUT ("snprintf failed creating tmp directory");
if (!mkdtemp (dir))
BAIL_OUT ("could not create tmp directory");
len = snprintf (file, sizeof (file), "%s/file", dir);
if ((len < 0) || (len >= sizeof (file)))
BAIL_OUT ("snprintf failed creating tmp file path");
if (!(fd = open (file, O_CREAT, 0644)))
BAIL_OUT ("could not create tmp file");
close (fd);
cleanup_push_string (cleanup_directory, dir);
cleanup_push_string (cleanup_file, file);
cleanup_run ();
ok (stat (file, &sb) < 0 && errno == ENOENT,
"cleanup removed file pushed second");
ok (stat (dir, &sb) < 0 && errno == ENOENT,
"cleanup removed dir pushed first");
/* Same but reverse push order
*/
len = snprintf (dir, sizeof (dir), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (dir)))
BAIL_OUT ("snprintf failed creating tmp directory");
if (!mkdtemp (dir))
BAIL_OUT ("could not create tmp directory");
len = snprintf (file, sizeof (file), "%s/file", dir);
if ((len < 0) || (len >= sizeof (file)))
BAIL_OUT ("snprintf failed creating tmp file path");
if (!(fd = open (file, O_CREAT, 0644)))
BAIL_OUT ("could not create tmp file");
close (fd);
cleanup_push_string (cleanup_file, file);
cleanup_push_string (cleanup_directory, dir);
cleanup_run ();
ok (stat (dir, &sb) == 0,
"cleanup failed to remove dir pushed first");
ok (stat (file, &sb) < 0 && errno == ENOENT,
"cleanup removed file pushed second (1 deep)");
if (rmdir (dir) < 0)
BAIL_OUT ("rmdir %s failed", dir);
/* Same but recursive removal
*/
len = snprintf (dir, sizeof (dir), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (dir)))
BAIL_OUT ("snprintf failed creating tmp directory");
if (!mkdtemp (dir))
BAIL_OUT ("could not create tmp directory");
len = snprintf (file, sizeof (file), "%s/file", dir);
if ((len < 0) || (len >= sizeof (file)))
BAIL_OUT ("snprintf failed creating tmp file path");
if (!(fd = open (file, O_CREAT, 0644)))
BAIL_OUT ("could not create tmp file");
close (fd);
cleanup_push_string (cleanup_directory_recursive, dir);
cleanup_run ();
ok (stat (file, &sb) < 0 && errno == ENOENT,
"cleanup removed file not pushed (1 deep)");
ok (stat (dir, &sb) < 0 && errno == ENOENT,
"cleanup removed pushed dir recursively");
/* Try couple levels deep
*/
len = snprintf (dir, sizeof (dir), "%s/cleanup_test.XXXXXX", tmp ? tmp : "/tmp");
if ((len < 0) || (len >= sizeof (dir)))
BAIL_OUT ("snprintf failed creating tmp dir path");
if (!mkdtemp (dir))
BAIL_OUT ("could not create tmp directory");
len = snprintf (dir2, sizeof (dir2), "%s/dir", dir);
if ((len < 0) || (len >= sizeof (dir2)))
BAIL_OUT ("snprintf failed creating tmp dir path");
if (mkdir (dir2, 0755) < 0)
BAIL_OUT ("mkdir failed");
len = snprintf (file, sizeof (file), "%s/file", dir2);
if ((len < 0) || (len >= sizeof (file)))
BAIL_OUT ("snprintf failed creating tmp file path");
if (!(fd = open (file, O_CREAT, 0644)))
BAIL_OUT ("could not create tmp file");
close (fd);
cleanup_push_string (cleanup_directory_recursive, dir);
cleanup_run ();
ok (stat (file, &sb) < 0 && errno == ENOENT,
"cleanup removed file not pushed (2 deep)");
ok (stat (dir2, &sb) < 0 && errno == ENOENT,
"cleanup removed dir not pushed (1 deep)");
ok (stat (dir, &sb) < 0 && errno == ENOENT,
"cleanup removed pushed dir recursively");
done_testing();
}
