106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package websocket
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/httpclient"
|
|
"github.com/fiskerinc/cloud-services/pkg/httpclient/mock"
|
|
"github.com/fiskerinc/cloud-services/pkg/testhelper"
|
|
)
|
|
|
|
func TestVerifyTokenAuthorized(t *testing.T) {
|
|
c := mock.Client{
|
|
DoFunc: func(*http.Request) (*http.Response, error) {
|
|
return &http.Response{StatusCode: 200}, nil
|
|
},
|
|
}
|
|
httpclient.Client = &c
|
|
|
|
ad := AuthData{
|
|
Token: "validtoken",
|
|
}
|
|
|
|
ok, err := verifyToken(ad)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestVerifyTokenAuthorized", nil, err)
|
|
}
|
|
if !ok {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestVerifyTokenAuthorized", true, ok)
|
|
}
|
|
}
|
|
|
|
func TestVerifyTokenUnauthorized(t *testing.T) {
|
|
c := mock.Client{
|
|
DoFunc: func(*http.Request) (*http.Response, error) {
|
|
return &http.Response{StatusCode: 401}, nil
|
|
},
|
|
}
|
|
httpclient.Client = &c
|
|
|
|
ad := AuthData{
|
|
Token: "invalidtoken",
|
|
}
|
|
|
|
ok, err := verifyToken(ad)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestVerifyTokenUnauthorized", nil, err)
|
|
}
|
|
if ok {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestVerifyTokenUnauthorized", false, ok)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticateRequest(t *testing.T) {
|
|
c := mock.Client{
|
|
DoFunc: func(*http.Request) (*http.Response, error) {
|
|
return &http.Response{StatusCode: 200}, nil
|
|
},
|
|
}
|
|
httpclient.Client = &c
|
|
|
|
ae := AuthEvent{
|
|
Topic: "auth_service",
|
|
Key: "FISKER123",
|
|
Payload: AuthPayload{
|
|
Handler: "verify",
|
|
Data: AuthData{
|
|
Token: "validtoken",
|
|
},
|
|
},
|
|
}
|
|
|
|
ok, err := AuthenticateRequest(ae)
|
|
if err != nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestAuthenticateRequest", nil, err)
|
|
}
|
|
if !ok {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestAuthenticateRequest", true, ok)
|
|
}
|
|
}
|
|
|
|
func TestAuthenticateRequestInvalid(t *testing.T) {
|
|
c := mock.Client{
|
|
DoFunc: func(*http.Request) (*http.Response, error) {
|
|
return &http.Response{StatusCode: 401}, nil
|
|
},
|
|
}
|
|
httpclient.Client = &c
|
|
|
|
ae := AuthEvent{
|
|
Topic: "invalid_topic",
|
|
Key: "FISKER123",
|
|
Payload: AuthPayload{
|
|
Handler: "verify",
|
|
Data: AuthData{
|
|
Token: "validtoken",
|
|
},
|
|
},
|
|
}
|
|
|
|
_, err := AuthenticateRequest(ae)
|
|
if err == nil {
|
|
t.Errorf(testhelper.TestErrorTemplate, "TestAuthenticateRequestInvalid", "error", nil)
|
|
}
|
|
}
|