diff options
Diffstat (limited to 'src/remembering-c.c')
| -rw-r--r-- | src/remembering-c.c | 146 |
1 files changed, 141 insertions, 5 deletions
diff --git a/src/remembering-c.c b/src/remembering-c.c index 2e7395f..a6bd8c2 100644 --- a/src/remembering-c.c +++ b/src/remembering-c.c @@ -7,6 +7,12 @@ #include <string.h> #include <unistd.h> +#ifdef TEST +#include "unit-test.h" +#include <stdbool.h> +#endif + + const int EXIT_ERROR = 1; const int EXIT_USAGE = 2; @@ -52,11 +58,11 @@ static int missing(FILE *stream, const char *const argument) { return 0; } -static char *get_profile_file(const char *const profile) { - char *home = NULL; - if (!(home = getenv("XDG_DATA_HOME"))) { +static char *get_profile_file(const char *const profile, FILE *stream) { + char *home = getenv("XDG_DATA_HOME"); + if (!home || (strcmp(home, "")) == 0) { if (!(home = getenv("HOME"))) { - fprintf(stderr, "Unable to get $XDG_DATA_HOME or $HOME environment variables\n"); + fprintf(stream, "Unable to get $XDG_DATA_HOME or $HOME environment variables\n"); return NULL; } const char *const suffix = "/.local/share/remembering"; @@ -97,8 +103,138 @@ static char *get_profile_file(const char *const profile) { return profile_file; } +#ifdef TEST +void test_get_profile_file() { + #define EXPECTED_FMT "\nexpected: %s\ngot: %s\n" + FILE *f = fopen("/dev/null", "w"); + assert(f); + + { + testing("get_profile_file() when $XDG_DATA_HOME is *unset*"); + const char *const previous_xdg_data_home = getenv("XDG_DATA_HOME"); + unsetenv("XDG_DATA_HOME"); + + const char *const HOME = getenv("HOME"); + const char *const SUFFIX = "/.local/share/remembering/"; + const char *const A_PROFILE = "A-PROFILE-NAME"; + assert(HOME); + + const size_t expected_size = + strlen(HOME) + + strlen(SUFFIX) + + strlen(A_PROFILE) + + sizeof(char); + char *const expected = malloc(expected_size); + assert(expected); + strcat(expected, HOME); + strcat(expected, SUFFIX); + strcat(expected, A_PROFILE); + + char *const profile_file = get_profile_file(A_PROFILE, f); + assert(profile_file); + assertf( + strcmp(profile_file, expected) == 0, + EXPECTED_FMT, + expected, + profile_file + ); + + free(profile_file); + free(expected); + if (previous_xdg_data_home) { + setenv("XDG_DATA_HOME", previous_xdg_data_home, true); + } + test_ok(); + } + { + testing("get_profile_file() when $XDG_DATA_HOME is *empty*"); + const char *const previous_xdg_data_home = getenv("XDG_DATA_HOME"); + setenv("XDG_DATA_HOME", "", true); + + const char *const HOME = getenv("HOME"); + const char *const SUFFIX = "/.local/share/remembering/"; + const char *const A_PROFILE = "ANOTHER-PROFILE-NAME"; + assert(HOME); + + const size_t expected_size = + strlen(HOME) + + strlen(SUFFIX) + + strlen(A_PROFILE) + + sizeof(char); + char *const expected = malloc(expected_size); + assert(expected); + strcat(expected, HOME); + strcat(expected, SUFFIX); + strcat(expected, A_PROFILE); + + char *const profile_file = get_profile_file(A_PROFILE, f); + assert(profile_file); + assertf( + strcmp(profile_file, expected) == 0, + EXPECTED_FMT, + expected, + profile_file + ); + + free(profile_file); + free(expected); + if (previous_xdg_data_home) { + setenv("XDG_DATA_HOME", previous_xdg_data_home, true); + } else { + unsetenv("XDG_DATA_HOME"); + } + test_ok(); + } + { + testing("get_profile_file() when $XDG_DATA_HOME is *set*"); + const char *const previous_xdg_data_home = getenv("XDG_DATA_HOME"); + + const char *const XDG_DATA_HOME = "/a/custom/path"; + const char *const A_PROFILE = "YET-ANOTHER-PROFILE-NAME"; + const char *const EXPECTED = "/a/custom/path/YET-ANOTHER-PROFILE-NAME"; + + setenv("XDG_DATA_HOME", XDG_DATA_HOME, true); + + char *const profile_file = get_profile_file(A_PROFILE, f); + assert(profile_file); + assertf( + strcmp(profile_file, EXPECTED) == 0, + EXPECTED_FMT, + EXPECTED, + profile_file + ); + + free(profile_file); + if (previous_xdg_data_home) { + setenv("XDG_DATA_HOME", previous_xdg_data_home, true); + } else { + unsetenv("XDG_DATA_HOME"); + } + test_ok(); + } + { + testing("get_profile_file() when $HOME is *unset*"); + const char *const previous_home = getenv("HOME"); + const char *const previous_xdg_data_home = getenv("XDG_DATA_HOME"); + unsetenv("XDG_DATA_HOME"); + unsetenv("HOME"); + + assert(get_profile_file("ANY-PROFILE", f) == NULL); + + setenv("HOME", previous_home, true); + if (previous_xdg_data_home) { + setenv("XDG_DATA_HOME", previous_xdg_data_home, true); + } + test_ok(); + } + + fclose(f); +} +#endif + int main(int argc, char *argv[]) { #ifdef TEST + test_get_profile_file(); return EXIT_SUCCESS; #endif @@ -151,7 +287,7 @@ int main(int argc, char *argv[]) { return EXIT_USAGE; } - char *const profile_file = get_profile_file(profile); + char *const profile_file = get_profile_file(profile, stderr); if(!profile_file) { return EXIT_ERROR; } |