windwalker/main_test.go at main · TeamOpencampus/windwalker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main

import (
	"bytes"
	"context"
	"encoding/json"
	"io"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"
	"windwalker/ent"

	"github.com/stretchr/testify/assert"
)

type AuthResponse struct {
	ID    string `json:"id"`
	Role  string `json:"role"`
	Token string `json:"token"`
}

type ErrorResponse struct {
	Code    string `json:"code"`
	Message string `json:"message"`
}

func TestMain(t *testing.T) {
	client, err := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
	assert.NoError(t, err, "failed to open database")
	// Run the auto migration tool.
	err = client.Schema.Create(context.Background())
	assert.NoError(t, err, "failed to migrate database")

	r := SetupRouter(client)
	t.Run("HealthCheckHandler", func(t *testing.T) {
		req, _ := http.NewRequest("GET", "/health", nil)
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		res, _ := ioutil.ReadAll(w.Body)
		assert.Equal(t, `"all systems operational"`, string(res))
		assert.Equal(t, http.StatusOK, w.Code)
	})

	t.Run("RegisterHandler should create account", func(t *testing.T) {
		form, _ := json.Marshal(map[string]string{"email": "someone@example.com", "password": "12345678"})
		req, _ := http.NewRequest(http.MethodPost, "/v1/register", bytes.NewBuffer(form))
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		var resp AuthResponse

		body, _ := io.ReadAll(w.Body)
		err = json.Unmarshal(body, &resp)
		assert.NoError(t, err)

		assert.NotEmpty(t, resp.Token)
		assert.Equal(t, "student", resp.Role)
		assert.Equal(t, http.StatusOK, w.Code)
	})

	t.Run("RegisterHandler should not create account with existing email", func(t *testing.T) {
		form, _ := json.Marshal(map[string]string{"email": "someone@example.com", "password": "12345678"})
		req, _ := http.NewRequest(http.MethodPost, "/v1/register", bytes.NewBuffer(form))
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		var resp ErrorResponse

		body, _ := io.ReadAll(w.Body)
		err = json.Unmarshal(body, &resp)
		assert.NoError(t, err)

		assert.Equal(t, "auth/email-already-registered", resp.Code)
	})

	t.Run("LoginHandler should success", func(t *testing.T) {
		form, _ := json.Marshal(map[string]string{"email": "someone@example.com", "password": "12345678"})
		req, _ := http.NewRequest(http.MethodPost, "/v1/login", bytes.NewBuffer(form))
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		var resp AuthResponse

		body, _ := io.ReadAll(w.Body)
		err = json.Unmarshal(body, &resp)
		assert.NoError(t, err)

		assert.NotEmpty(t, resp.Token)
		assert.Equal(t, "student", resp.Role)
		assert.Equal(t, http.StatusOK, w.Code)
	})

	t.Run("LoginHandler should not login with non existant email", func(t *testing.T) {
		form, _ := json.Marshal(map[string]string{"email": "nonexistant@example.com", "password": "12345678"})
		req, _ := http.NewRequest(http.MethodPost, "/v1/login", bytes.NewBuffer(form))
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		var resp ErrorResponse

		body, _ := io.ReadAll(w.Body)
		err = json.Unmarshal(body, &resp)
		assert.NoError(t, err)

		assert.Equal(t, "auth/user-not-found", resp.Code)
	})

	t.Run("LoginHandler should not login with invalid password", func(t *testing.T) {
		form, _ := json.Marshal(map[string]string{"email": "someone@example.com", "password": "invalid_password"})
		req, _ := http.NewRequest(http.MethodPost, "/v1/login", bytes.NewBuffer(form))
		w := httptest.NewRecorder()
		r.ServeHTTP(w, req)

		var resp ErrorResponse

		body, _ := io.ReadAll(w.Body)
		err = json.Unmarshal(body, &resp)
		assert.NoError(t, err)

		assert.Equal(t, "auth/invalid-password", resp.Code)
	})
}