🧪 add test for delete geopoint

Haransis · Nov 6, 2022 · e40b27b · e40b27b
1 parent 7b59a61
commit e40b27b
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/controller/controller_test.go b/controller/controller_test.go
@@ -2,6 +2,7 @@ package controller
 
 import (
 	"bytes"
+	"database/sql"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -410,6 +411,38 @@ func TestGetRestrictedGeoPoint(t *testing.T) {
 	}
 }
 
+func TestDeleteGeoPoint(t *testing.T) {
+	tests := []struct {
+		IdToDelete int
+		JWT        string
+		StatusCode int
+	}{
+		{geoIdEnabled, "", http.StatusUnauthorized},
+		{12000, validTokens[0], http.StatusNotFound},
+		{geoIdEnabled, validTokens[0], http.StatusOK},
+	}
+
+	for _, test := range tests {
+		w := httptest.NewRecorder()
+
+		req, _ := http.NewRequest(http.MethodDelete, fmt.Sprintf("/api/v1/restricted/geopoint/%d", test.IdToDelete), nil)
+
+		if test.JWT != "" {
+			// Write authorization token in header
+			req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", test.JWT))
+		}
+
+		r.ServeHTTP(w, req)
+		assert.Equal(t, test.StatusCode, w.Code)
+
+		if test.StatusCode == http.StatusOK {
+			var geoPoint geopoint.DbGeoPoint
+			err := c.Db.Get(&geoPoint, "SELECT * FROM geopoints WHERE id = $1", test.IdToDelete)
+			assert.Equal(t, err, sql.ErrNoRows)
+		}
+	}
+}
+
 func TestMakeAdmin(t *testing.T) {
 	tests := []struct {
 		Id         int