Add node test

uoo723 · Jun 21, 2017 · 9059b4f · 9059b4f
1 parent 922570c
commit 9059b4f
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 12 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -5,4 +5,5 @@ include_directories(include)
 
 file(GLOB SOURCES "src/*.c")
 
-add_executable(main ${SOURCES})
+add_executable(main ${SOURCES})
+add_executable(node_test "test/node_test.c" "src/node.c")
diff --git a/include/node.h b/include/node.h
@@ -9,9 +9,9 @@ typedef struct node {
     struct node *next;
 } node_t;
 
-node_t init();
-int insert(unsigned int, char[VALUESIZE]);
-int delete(unsigned int);
-node_t get(unsigned int);
+node_t *init();
+int insert(node_t *, unsigned int, char[VALUESIZE]);
+int delete(node_t *, unsigned int);
+node_t *get(node_t *, unsigned int);
 
 #endif
diff --git a/src/node.c b/src/node.c
@@ -4,7 +4,7 @@
 #include "node.h"
 #include "msg.h"
 
-node_t init() {
+node_t *init() {
     node_t *head = NULL;
     head = malloc(sizeof(node_t));
     if (head == NULL) {
@@ -16,12 +16,12 @@ node_t init() {
     return head;
 }
 
-int insert(node_t head, unsigned int key, char value[VALUESIZE]) {
+int insert(node_t *head, unsigned int key, char value[VALUESIZE]) {
     if (head == NULL) {
         return -1;
     }
 
-    node_t *node = get(key);
+    node_t *node = get(head, key);
 
     if (node != NULL) {
         return -1;
@@ -47,7 +47,7 @@ int insert(node_t head, unsigned int key, char value[VALUESIZE]) {
     return 0;
 }
 
-int delete(node_t head, unsigned int key) {
+int delete(node_t *head, unsigned int key) {
     if (head == NULL) {
         return -1;
     }
@@ -62,7 +62,7 @@ int delete(node_t head, unsigned int key) {
         iter = iter->next;
     }
 
-    if (iter->next == NULL) {
+    if (iter == NULL) {
         return -1;
     }
 
@@ -75,9 +75,9 @@ int delete(node_t head, unsigned int key) {
     return 0;
 }
 
-node_t get(node_t head, unsigned int key) {
+node_t *get(node_t *head, unsigned int key) {
     if (head == NULL) {
-        return;
+        return NULL;
     }
 
     node_t *iter = head;

diff --git a/test/node_test.c b/test/node_test.c
@@ -0,0 +1,43 @@
+#include <stdio.h>
+
+#include "node.h"
+
+void print(node_t *);
+
+int main() {
+    node_t *list = init();
+
+    if (list == NULL) {
+        return 0;
+    }
+
+    insert(list, 1, "hello");
+    insert(list, 2, "world");
+    insert(list, 3, "Han");
+    insert(list, 4, "Sangwoo");
+    insert(list, 5, "woo");
+    insert(list, 5, "aa");
+    print(list);
+
+    printf("delete test\n");
+    delete(list, 6);
+    delete(list, 4);
+    print(list);
+
+    printf("get test\n");
+    node_t *temp = get(list, 1);
+    printf("key: %d, value: %s\n", temp->key, temp->value);
+    temp = get(list, 6);
+    if (temp == NULL) {
+        printf("get is null\n");
+    }
+    return 0;
+}
+
+void print(node_t *list) {
+    node_t *iter = list;
+
+    while ((iter = iter->next) != NULL) {
+        printf("key: %d, value: %s\n", iter->key, iter->value);
+    }
+}