851ad300cb
* ci for acquisition and cwapi * update README Co-authored-by: AlteredCoder <AlteredCoder>
130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package cwapi
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const apiVersion = "v1"
|
|
const apiURL = "https://my_test_endpoint"
|
|
|
|
var apiBaseURL = fmt.Sprintf("%s/%s/", apiURL, apiVersion)
|
|
|
|
var pullResponse = `{
|
|
"statusCode" : 200,
|
|
"message":
|
|
[
|
|
{
|
|
"range_ip" : "1.2.3.4",
|
|
"country" : "FR",
|
|
"as_org" : "Test",
|
|
"as_num" : "1234",
|
|
"action" : "ban",
|
|
"reason": "crowdsec/test"
|
|
},
|
|
{
|
|
"range_ip" : "1.2.3.5",
|
|
"country" : "FR",
|
|
"as_org" : "Test",
|
|
"as_num" : "1235",
|
|
"action" : "ban",
|
|
"reason": "crowdsec/test"
|
|
},
|
|
{
|
|
"range_ip" : "1.2.3.6",
|
|
"country" : "FR",
|
|
"as_org" : "Test",
|
|
"as_num" : "1236",
|
|
"action" : "ban",
|
|
"reason": "crowdsec/test"
|
|
}
|
|
]
|
|
}`
|
|
|
|
type mockTransport struct {
|
|
nbTryPushTokenOK int // to test token expiration
|
|
nbTryPushTokenNOK int
|
|
}
|
|
|
|
func newMockTransport() http.RoundTripper {
|
|
return &mockTransport{}
|
|
}
|
|
|
|
func newMockClient() *http.Client {
|
|
return &http.Client{
|
|
Transport: newMockTransport(),
|
|
Timeout: time.Second * 20,
|
|
}
|
|
}
|
|
|
|
// Implement http.RoundTripper
|
|
func (t *mockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
var responseBody string
|
|
var statusCode int
|
|
// Create mocked http.Response
|
|
response := &http.Response{
|
|
Header: make(http.Header),
|
|
Request: req,
|
|
}
|
|
response.Header.Set("Content-Type", "application/json")
|
|
switch req.URL.Path {
|
|
case "/v1/signin":
|
|
responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
|
|
statusCode = 200
|
|
case "/v1/register":
|
|
responseBody = `{"statusCode": 200, "message": "OK"}`
|
|
statusCode = 200
|
|
case "/v1/signals":
|
|
responseBody = `{"statusCode": 200, "message": "OK"}`
|
|
statusCode = 200
|
|
case "/v1/pull":
|
|
responseBody = pullResponse
|
|
statusCode = 200
|
|
case "/v1/signals_token_expired":
|
|
if t.nbTryPushTokenOK == 0 {
|
|
responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
|
|
statusCode = 401
|
|
t.nbTryPushTokenOK++
|
|
} else {
|
|
responseBody = `{"statusCode": 200, "message": "OK"}`
|
|
statusCode = 200
|
|
}
|
|
case "/v1/signals_token_renew_fail":
|
|
if t.nbTryPushTokenNOK == 0 {
|
|
responseBody = `{"statusCode": 200, "message": "crowdsec_api_token"}`
|
|
statusCode = 401
|
|
t.nbTryPushTokenNOK++
|
|
} else {
|
|
responseBody = `{"statusCode": 500, "message": "token expired"}`
|
|
statusCode = 500
|
|
}
|
|
case "/v1/signals_bad_response_code":
|
|
responseBody = `{"statusCode": 200, "message": "OK"}`
|
|
statusCode = 500
|
|
case "/v1/enroll":
|
|
responseBody = `{"statusCode": 200, "message": "OK"}`
|
|
statusCode = 200
|
|
case "/v1/resetpassword":
|
|
responseBody = `{"statusCode": 200, "message": "password updated successfully"}`
|
|
statusCode = 200
|
|
case "/v1/resetpassword_unknown_user":
|
|
responseBody = `{"statusCode": 500, "message": "User not found"}`
|
|
statusCode = 200
|
|
case "/v1/unknown_path":
|
|
statusCode = 404
|
|
responseBody = `{"error": "unknown URI"}`
|
|
case "/v1/malformed_response":
|
|
statusCode = 200
|
|
responseBody = `{"statusCode" : 200, "msg" : "api_token"`
|
|
case "/v1/bad_response":
|
|
statusCode = 200
|
|
responseBody = `{"statusCode" : 200, "msg" : "api_token"}`
|
|
}
|
|
response.StatusCode = statusCode
|
|
response.Body = ioutil.NopCloser(strings.NewReader(responseBody))
|
|
return response, nil
|
|
}
|