Added api endpoints and tests

2022-01-24 12:59:43 +01:00
parent 10b768743b
commit cfbdcc7f82
36 changed files with 1781 additions and 315 deletions
--- a/api/assets_test.go
+++ b/api/assets_test.go
@@ -0,0 +1,138 @@
+package api
+
+import (
+	"TheAdversary/database"
+	"TheAdversary/schema"
+	"encoding/base64"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestAssetsGet(t *testing.T) {
+	if err := initTestDatabase("assets_get_test.sqlite3"); err != nil {
+		t.Fatal(err)
+	}
+
+	database.GetDB().Table("assets").Create([]map[string]interface{}{
+		{
+			"name": "linux",
+			"data": "this should be an image of tux :3",
+			"link": "/assets/linux",
+		},
+		{
+			"name": "get test",
+			"data": "",
+			"link": "/assets/get-test",
+		},
+	})
+
+	server := httptest.NewServer(http.HandlerFunc(assetsGet))
+	checkTestInformation(t, server.URL, []testInformation{
+		{
+			Method: http.MethodGet,
+			Query: map[string]interface{}{
+				"q": "linux",
+			},
+			ResultBody: []schema.Asset{
+				{
+					Id:   1,
+					Name: "linux",
+					Link: "/assets/linux",
+				},
+			},
+			Code: http.StatusOK,
+		},
+		{
+			Method: http.MethodGet,
+			Query: map[string]interface{}{
+				"limit": 1,
+			},
+			ResultBody: []schema.Asset{
+				{
+					Id:   1,
+					Name: "linux",
+					Link: "/assets/linux",
+				},
+			},
+			Code: http.StatusOK,
+		},
+		{
+			Method: http.MethodGet,
+			ResultBody: []schema.Asset{
+				{
+					Id:   1,
+					Name: "linux",
+					Link: "/assets/linux",
+				},
+				{
+					Id:   2,
+					Name: "get test",
+					Link: "/assets/get-test",
+				},
+			},
+			Code: http.StatusOK,
+		},
+	})
+}
+
+func TestAssetsPost(t *testing.T) {
+	if err := initTestDatabase("assets_post_test.sqlite3"); err != nil {
+		t.Fatal(err)
+	}
+
+	server := httptest.NewServer(http.HandlerFunc(assetsPost))
+	checkTestInformation(t, server.URL, []testInformation{
+		{
+			Method: http.MethodPost,
+			Body: assetsPostPayload{
+				Name:    "test",
+				Content: base64.StdEncoding.EncodeToString([]byte("test asset")),
+			},
+			ResultBody: schema.Asset{
+				Id:   1,
+				Name: "test",
+				Link: "/assets/test",
+			},
+			Code: http.StatusOK,
+		},
+		{
+			Method: http.MethodPost,
+			Body: assetsPostPayload{
+				Name:    "test",
+				Content: base64.StdEncoding.EncodeToString([]byte("test asset")),
+			},
+			Code: http.StatusConflict,
+		},
+	})
+}
+
+func TestAssetsDelete(t *testing.T) {
+	if err := initTestDatabase("assets_delete_test.sqlite3"); err != nil {
+		t.Fatal(err)
+	}
+
+	database.GetDB().Table("assets").Create(map[string]interface{}{
+		"name": "example",
+		"data": "just a normal string",
+		"link": "/assets/example",
+	})
+
+	server := httptest.NewServer(http.HandlerFunc(assetsDelete))
+	checkTestInformation(t, server.URL, []testInformation{
+		{
+			Method: http.MethodDelete,
+			Body: assetsDeletePayload{
+				Id: 1,
+			},
+			Code: http.StatusOK,
+		},
+		{
+			Method: http.MethodDelete,
+			Body: assetsDeletePayload{
+				Id: 69,
+			},
+			Code: http.StatusNotFound,
+		},
+	})
+}