NNTPGrab

/* 

    Copyright (C) 2005-2011  Erik van Pienbroek

    This program is free software; you can redistribute it and/or modify

    it under the terms of the GNU General Public License as published by

    the Free Software Foundation; either version 2 of the License, or

    (at your option) any later version.

    This program is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License

    along with this program; if not, write to the Free Software

    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

*/

#include 

#include 

#include 

#include 

#ifdef WIN32

#include 

#else

#include 

#include 

#endif

static char *

generate_tmpdir(void)

{

    char *tmpdir;

    pid_t pid;

    char pid_str[16];

    memset(&pid_str, 0, sizeof(pid_str));

    pid = getpid();

    snprintf(pid_str, sizeof(pid_str) - 1, "%i", (int) pid);

    tmpdir = g_build_filename(g_get_tmp_dir(), "nntpgrab_tests", pid_str, NULL);

    return tmpdir;

}

char *

get_location_of_download_queue(void)

{

    char *tmpdir = generate_tmpdir();

    char *ret;

    ret = g_build_filename(tmpdir, "NNTPGrab", "download_queue.db", NULL);

    g_free(tmpdir);

    return ret;

}

void

initialize_tests(void)

{

    char *tmpdir;

    char *errmsg = NULL;

    /* Create a temporary folder to perform the tests */

    tmpdir = generate_tmpdir();

    if (g_mkdir_with_parents(tmpdir, 0700) == -1) {

        g_error("Unable to create temporary folder named '%s'. Unable to continue tests: %s\n", tmpdir, errmsg);

    }

    g_setenv("NNTPGRAB_CONFIG_DIR", tmpdir, TRUE);

    g_free(tmpdir);

}

void

cleanup_tests(void)

{

    char *path;

    char *tmpdir = generate_tmpdir();

    /* Remove: /tmp/nntpgrab_tests/$PID/NNTPGrab/download_queue.db */

    path = g_build_filename(tmpdir, "NNTPGrab", "download_queue.db", NULL);

    if (g_unlink(path) == -1) {

        g_error("Unable to remove file '%s': %s\n", path, strerror(errno));

    }

    g_free(path);

    path = g_build_filename(tmpdir, "NNTPGrab", "nntpgrab.conf", NULL);

    if (g_unlink(path) == -1) {

        g_error("Unable to remove file '%s': %s\n", path, strerror(errno));

    }

    g_free(path);

    /* Remove: /tmp/nntpgrab_tests/$PID/NNTPGrab */

    path = g_build_filename(tmpdir, "NNTPGrab", NULL);

    if (g_rmdir(path) == -1) {

        g_error("Unable to remove folder '%s': %s\n", path, strerror(errno));

    }

    g_free(path);

    /* Remove: /tmp/nntpgrab_tests/$PID */

    if (g_rmdir(tmpdir) == -1) {

        g_error("Unable to remove folder '%s': %s\n", tmpdir, strerror(errno));

    }

    /* Remove: /tmp/nntpgrab_tests */

    path = g_path_get_dirname(tmpdir);

    if (g_rmdir(path) == -1) {

        /* No need to give an error here as other tests may still be running */

        //g_error("Unable to remove folder '%s': %s\n", path, strerror(errno));

    }

    g_free(path);

    g_free(tmpdir);

}