1 files changed, 124 insertions, 1 deletions
diff --git a/tests/ds/hashmap_tests.c b/tests/ds/hashmap_tests.c
index 83efe03..49e4261 100644
--- a/tests/ds/hashmap_tests.c
+++ b/tests/ds/hashmap_tests.c
@@ -1,7 +1,130 @@
 #include "hashmap_tests.h"
 #include "moslib/ds/hashmap.h"
 
+#include <stdlib.h>
+#include <string.h>
+
+#include "moslib/ds/array.h"
+#include "moslib/string.h"
+
+Test put_get() {
+    HM_DEF(E, float, int);
+    E *hm = NULL;
+
+    hm_put(hm, 0.0f, 69);
+    hm_put(hm, 1.0f, 420);
+    test_assert(hm_get(hm, 0.0f) == 69, "hm_get got value not from hm_put");
+    test_assert(hm_get(hm, 1.0f) == 420, "hm_get got value not from hm_put");
+
+    hm_put(hm, 0.0f, 2137);
+    test_assert(hm_get(hm, 0.0f) == 2137, "hm_put didn't update value properly");
+
+    // a lot of puts and gets to test expansion of hashmap
+    for (int i = 2; i < 100; i++) {
+        hm_put(hm, (float)i, i);
+    }
+    int flag = 1;
+    for (int i = 2; i < 100; i++) {
+        if (hm_get(hm, (float)i) != i)
+            flag = 0;
+    }
+    test_assert(flag, "hm_put / hm_get failed when expansion was needed");
+
+    hm_free(hm);
+}
+
+Test put_e_get_e() {
+    HM_DEF(E, float, int);
+    E *hm = NULL;
+
+    hm_put_e(hm, (E){0.0f, 69});
+    hm_put_e(hm, (E){1.0f, 420});
+    E e = {0.0f, 69};
+    test_assert(memcmp(&e, hm_get_e(hm, 0.0f), sizeof(E)) == 0, "hm_get_e got entry not from hm_put_e");
+    e = (E){1.0f, 420};
+    test_assert(memcmp(&e, hm_get_e(hm, 1.0f), sizeof(E)) == 0, "hm_get_e got entry not from hm_put_e");
+
+    e = (E){1.0f, 2137};
+    hm_put_e(hm, e);
+    test_assert(memcmp(&e, hm_get_e(hm, 1.0f), sizeof(E)) == 0, "hm_put_e didn't update entry properly");
+
+    hm_free(hm);
+}
+
+Test str_key() {
+    HM_DEF(E, char *, int);
+    E *hm = hm_new(.key = STR);
+
+    char *str = "key";
+    char *str2 = strdup(str);
+    hm_put(hm, str, 69);
+    hm_put(hm, "hey", 420);
+    test_assert(hm_get(hm, str2) == 69, "hm_get got value not from hm_put");
+    test_assert(hm_get(hm, "hey") == 420, "hm_get got value not from hm_put");
+
+    hm_put(hm, str2, 2137);
+    test_assert(hm_get(hm, str) == 2137, "hm_put didn't update value properly");
+
+    free(str2);
+    hm_free(hm);
+}
+
+Test ss_key() {
+    HM_DEF(E, StringSlice, int);
+    E *hm = hm_new(.key = SS);
+
+    char *str = "key";
+    char *str2 = strdup(str);
+    StringSlice ss = ss_from_str(str);
+    StringSlice ss2 = ss_from_str(str2);
+    hm_put(hm, ss, 69);
+    hm_put(hm, SS_LIT("hey"), 420);
+    test_assert(hm_get(hm, ss2) == 69, "hm_get got value not from hm_put");
+    test_assert(hm_get(hm, SS_LIT("hey")) == 420, "hm_get got value not from hm_put");
+
+    hm_put(hm, ss2, 2137);
+    test_assert(hm_get(hm, ss) == 2137, "hm_put didn't update value properly");
+
+    free(str2);
+    hm_free(hm);
+}
+
+Test looping() {
+    HM_DEF(E, int, int);
+    E *hm = NULL;
+
+    hm_put(hm, 1, 10);
+    hm_put(hm, 2, 20);
+    hm_put(hm, 3, 30);
+
+    E *allowed = NULL;
+    E items[] = {{1, 10}, {2, 20}, {3, 30}};
+    arr_append_n(allowed, items, 3);
+
+    test_assert(memcmp(hm_first(hm), hm_next(hm, NULL), sizeof(E)), "hm_next(hm, NULL) != hm_first(hm)");
+    E *e = NULL;
+    while ((e = hm_next(hm, e))) {
+        int flag = 0;
+        for (size_t i = 0; i < arr_len(allowed); i++) {
+            if (memcmp(e, allowed + i, sizeof(E)) == 0) {
+                flag = 1;
+                arr_del(allowed, i);
+                break;
+            }
+        }
+        test_assert(flag, "hm_next produced an entry that wasn't put in the hashmap");
+    }
+    test_assert(arr_len(allowed) == 0, "hm_next didn't produce every entry from hashmap when it looped over it");
+
+    arr_free(allowed);
+    hm_free(hm);
+}
+
 void ds_hashmap_tests(Tester *tester) {
     TestGroup *group = add_test_group(tester, "moslib/ds/hashmap.h");
-    (void)group;
+    add_test(group, put_get);
+    add_test(group, put_e_get_e);
+    add_test(group, str_key);
+    add_test(group, ss_key);
+    add_test(group, looping);
 }