Add conda_*() tests

1 files changed, 181 insertions, 0 deletions

diff --git a/tests/test_conda.c b/tests/test_conda.c
new file mode 100644
index 0000000..49cf48b
--- /dev/null
+++ b/tests/test_conda.c
@@ -0,0 +1,181 @@
+#include "testing.h"
+
+char cwd_start[PATH_MAX];
+char cwd_workspace[PATH_MAX];
+int conda_is_installed = 0;
+
+void test_micromamba() {
+    char mm_prefix[PATH_MAX] = {0};
+    char c_prefix[PATH_MAX] = {0};
+    snprintf(mm_prefix, strlen(cwd_workspace) + strlen("micromamba") + 2, "%s/%s", cwd_workspace, "micromamba");
+    snprintf(c_prefix, strlen(mm_prefix) + strlen("conda") + 2, "%s/%s", mm_prefix, "conda");
+
+    struct testcase {
+        struct MicromambaInfo mminfo;
+        const char *cmd;
+        int result;
+    };
+    struct testcase tc[] = {
+            {.mminfo = {.micromamba_prefix = mm_prefix, .conda_prefix = c_prefix}, .cmd = "info", .result = 0},
+            {.mminfo = {.micromamba_prefix = mm_prefix, .conda_prefix = c_prefix}, .cmd = "env list", .result = 0},
+            {.mminfo = {.micromamba_prefix = mm_prefix, .conda_prefix = c_prefix}, .cmd = "run python -V", .result = 0},
+            {.mminfo = {.micromamba_prefix = mm_prefix, .conda_prefix = c_prefix}, .cmd = "no_such_option", .result = 109},
+    };
+
+    for (size_t i = 0; i < sizeof(tc) / sizeof(*tc); i++) {
+        struct testcase *item = &tc[i];
+        int result = micromamba(&item->mminfo, (char *) item->cmd);
+        if (result > 0) {
+            result = result >> 8;
+        }
+        STASIS_ASSERT(result == item->result, "unexpected exit value");
+        SYSERROR("micromamba command: '%s' (returned: %d)", item->cmd, result);
+    }
+}
+
+static char conda_prefix[PATH_MAX] = {0};
+struct Delivery ctx;
+
+void test_conda_installation() {
+    char *install_url = calloc(255, sizeof(install_url));
+    delivery_get_installer_url(&ctx, install_url);
+    delivery_get_installer(&ctx, install_url);
+    delivery_install_conda(ctx.conda.installer_path, ctx.storage.conda_install_prefix);
+    STASIS_ASSERT_FATAL(access(ctx.storage.conda_install_prefix, F_OK) == 0, "conda was not installed correctly");
+    STASIS_ASSERT_FATAL(conda_activate(ctx.storage.conda_install_prefix, "base") == 0, "unable to activate base environment");
+    STASIS_ASSERT_FATAL(conda_exec("info") == 0, "conda was not installed correctly");
+    conda_is_installed = 1;
+}
+
+void test_conda_activate() {
+    struct testcase {
+        const char *key;
+        char *value;
+    };
+    struct testcase tc[] = {
+        {.key = "CONDA_DEFAULT_ENV",  .value = "base"},
+        {.key = "CONDA_SHLVL", .value = "1"},
+        {.key = "_CE_CONDA", ""},
+        {.key = "_CE_M", ""},
+        {.key = "CONDA_PREFIX", ctx.storage.conda_install_prefix},
+    };
+
+    STASIS_SKIP_IF(!conda_is_installed, "cannot run without conda");
+    STASIS_ASSERT_FATAL(conda_activate(ctx.storage.conda_install_prefix, "base") == 0, "unable to activate base environment");
+
+    for (size_t i = 0; i < sizeof(tc) / sizeof(*tc); i++) {
+        struct testcase *item = &tc[i];
+        char *value = getenv(item->key);
+        STASIS_ASSERT(value != NULL, "expected key to exist, but doesn't. conda is not activated.");
+        if (value) {
+            STASIS_ASSERT(strcmp(value, item->value) == 0, "conda variable value mismatch");
+        }
+    }
+}
+
+void test_conda_exec() {
+    STASIS_ASSERT(conda_exec("--version") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("info") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("search fitsverify") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("create -n testenv") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("install -n testenv fitsverify") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("update -n testenv fitsverify") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("list -n testenv") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("run -n testenv fitsverify") == 0, "conda is broken");
+    STASIS_ASSERT(conda_exec("notasubcommand --help") != 0, "conda is broken");
+}
+
+void test_python_exec() {
+    const char *python_system_path = find_program("python3");
+    char python_path[PATH_MAX];
+    sprintf(python_path, "%s/bin/python3", ctx.storage.conda_install_prefix);
+
+    STASIS_ASSERT(strcmp(python_path, python_system_path ? python_system_path : "/not/found") == 0, "conda is not configured correctly.");
+    STASIS_ASSERT(python_exec("-V") == 0, "python is broken");
+}
+
+void test_conda_setup_headless() {
+    globals.conda_packages = strlist_init();
+    globals.pip_packages = strlist_init();
+    strlist_append(&globals.conda_packages, "boa");
+    strlist_append(&globals.conda_packages, "conda-build");
+    strlist_append(&globals.conda_packages, "conda-verify");
+    strlist_append(&globals.pip_packages, "pytest");
+    STASIS_ASSERT(conda_setup_headless() == 0, "headless configuration failed");
+}
+
+void test_conda_env_create_from_uri() {
+    const char *url = "https://ssb.stsci.edu/jhunk/stasis_test/test_conda_env_create_from_uri.yml";
+    char *name = strdup(__FUNCTION__);
+    STASIS_ASSERT(conda_env_create_from_uri(name, (char *) url) == 0, "creating an environment from a remote source failed");
+    free(name);
+}
+
+void test_conda_env_create_export_remove() {
+    char *name = strdup(__FUNCTION__);
+    STASIS_ASSERT(conda_env_create(name, "3", "fitsverify") == 0, "unable to create a simple environment");
+    STASIS_ASSERT(conda_env_export(name, ".", name) == 0, "unable to export an environment");
+    STASIS_ASSERT(conda_env_remove(name) == 0, "unable to remove an environment");
+    free(name);
+}
+
+void test_conda_index() {
+    mkdirs("channel/noarch", 0755);
+    STASIS_ASSERT(conda_index("channel") == 0, "cannot index a simple conda channel");
+}
+
+int main(int argc, char *argv[]) {
+    STASIS_TEST_BEGIN_MAIN();
+    STASIS_TEST_FUNC *tests[] = {
+        test_micromamba,
+        test_conda_installation,
+        test_conda_activate,
+        test_conda_setup_headless,
+        test_conda_exec,
+        test_python_exec,
+        test_conda_env_create_from_uri,
+        test_conda_env_create_export_remove,
+        test_conda_index,
+    };
+
+    const char *ws = "workspace";
+    getcwd(cwd_start, sizeof(cwd_start) - 1);
+    mkdir(ws, 0755);
+    chdir(ws);
+    getcwd(cwd_workspace, sizeof(cwd_workspace) - 1);
+
+    snprintf(conda_prefix, strlen(cwd_workspace) + strlen("conda") + 2, "%s/conda", cwd_workspace);
+
+    const char *mockinidata = "[meta]\n"
+                              "name = mock\n"
+                              "version = 1.0.0\n"
+                              "rc = 1\n"
+                              "mission = generic\n"
+                              "python = 3.11\n"
+                              "[conda]\n"
+                              "installer_name = Miniforge3\n"
+                              "installer_version = 24.3.0-0\n"
+                              "installer_platform = {{env:STASIS_CONDA_PLATFORM}}\n"
+                              "installer_arch = {{env:STASIS_CONDA_ARCH}}\n"
+                              "installer_baseurl = https://github.com/conda-forge/miniforge/releases/download/24.3.0-0\n";
+    stasis_testing_write_ascii("mock.ini", mockinidata);
+    struct INIFILE *ini = ini_open("mock.ini");
+    ctx._stasis_ini_fp.delivery = ini;
+    ctx._stasis_ini_fp.delivery_path = realpath("mock.ini", NULL);
+
+    setenv("TMPDIR", cwd_workspace, 1);
+    globals.sysconfdir = getenv("STASIS_SYSCONFDIR");
+    ctx.storage.root = strdup(cwd_workspace);
+
+    setenv("LANG", "C", 1);
+    bootstrap_build_info(&ctx);
+    delivery_init(&ctx);
+
+    STASIS_TEST_RUN(tests);
+
+    chdir(cwd_start);
+    if (rmtree(cwd_workspace)) {
+        perror(cwd_workspace);
+    }
+    STASIS_TEST_END_MAIN();
+}
\ No newline at end of file