2020-04-15 01:20:00 +02:00
package core
import (
2022-09-07 03:22:02 +02:00
"bytes"
"crypto/tls"
"crypto/x509"
2021-12-03 07:44:17 +01:00
"io"
2022-09-07 03:22:02 +02:00
"net/http"
2020-12-19 00:40:11 +01:00
"strings"
2020-04-15 01:20:00 +02:00
"testing"
2020-10-22 03:56:07 +02:00
"time"
2021-05-19 04:29:15 +02:00
2022-06-21 03:25:14 +02:00
"github.com/TwiN/gatus/v4/alerting/alert"
"github.com/TwiN/gatus/v4/client"
"github.com/TwiN/gatus/v4/core/ui"
2022-09-07 03:22:02 +02:00
"github.com/TwiN/gatus/v4/test"
2020-04-15 01:20:00 +02:00
)
2022-09-07 03:22:02 +02:00
func TestEndpoint ( t * testing . T ) {
defer client . InjectHTTPClient ( nil )
scenarios := [ ] struct {
Name string
Endpoint Endpoint
ExpectedResult * Result
MockRoundTripper test . MockRoundTripper
} {
{
Name : "success" ,
Endpoint : Endpoint {
Name : "website-health" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[STATUS] == 200" , "[BODY].status == UP" , "[CERTIFICATE_EXPIRATION] > 24h" } ,
} ,
ExpectedResult : & Result {
Success : true ,
Connected : true ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
{ Condition : "[STATUS] == 200" , Success : true } ,
{ Condition : "[BODY].status == UP" , Success : true } ,
{ Condition : "[CERTIFICATE_EXPIRATION] > 24h" , Success : true } ,
} ,
DomainExpiration : 0 , // Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
} ,
MockRoundTripper : test . MockRoundTripper ( func ( r * http . Request ) * http . Response {
return & http . Response {
StatusCode : http . StatusOK ,
Body : io . NopCloser ( bytes . NewBufferString ( ` { "status": "UP"} ` ) ) ,
TLS : & tls . ConnectionState { PeerCertificates : [ ] * x509 . Certificate { { NotAfter : time . Now ( ) . Add ( 9999 * time . Hour ) } } } ,
}
} ) ,
} ,
{
Name : "failed-body-condition" ,
Endpoint : Endpoint {
Name : "website-health" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[STATUS] == 200" , "[BODY].status == UP" } ,
} ,
ExpectedResult : & Result {
Success : false ,
Connected : true ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
{ Condition : "[STATUS] == 200" , Success : true } ,
{ Condition : "[BODY].status (DOWN) == UP" , Success : false } ,
} ,
DomainExpiration : 0 , // Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
} ,
MockRoundTripper : test . MockRoundTripper ( func ( r * http . Request ) * http . Response {
return & http . Response { StatusCode : http . StatusOK , Body : io . NopCloser ( bytes . NewBufferString ( ` { "status": "DOWN"} ` ) ) }
} ) ,
} ,
{
Name : "failed-status-condition" ,
Endpoint : Endpoint {
Name : "website-health" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[STATUS] == 200" } ,
} ,
ExpectedResult : & Result {
Success : false ,
Connected : true ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
{ Condition : "[STATUS] (502) == 200" , Success : false } ,
} ,
DomainExpiration : 0 , // Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
} ,
MockRoundTripper : test . MockRoundTripper ( func ( r * http . Request ) * http . Response {
return & http . Response { StatusCode : http . StatusBadGateway , Body : http . NoBody }
} ) ,
} ,
{
Name : "condition-with-failed-certificate-expiration" ,
Endpoint : Endpoint {
Name : "website-health" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[CERTIFICATE_EXPIRATION] > 100h" } ,
UIConfig : & ui . Config { DontResolveFailedConditions : true } ,
} ,
ExpectedResult : & Result {
Success : false ,
Connected : true ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
// Because UIConfig.DontResolveFailedConditions is true, the values in the condition should not be resolved
{ Condition : "[CERTIFICATE_EXPIRATION] > 100h" , Success : false } ,
} ,
DomainExpiration : 0 , // Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
} ,
MockRoundTripper : test . MockRoundTripper ( func ( r * http . Request ) * http . Response {
return & http . Response {
StatusCode : http . StatusOK ,
Body : http . NoBody ,
TLS : & tls . ConnectionState { PeerCertificates : [ ] * x509 . Certificate { { NotAfter : time . Now ( ) . Add ( 5 * time . Hour ) } } } ,
}
} ) ,
} ,
{
Name : "domain-expiration" ,
Endpoint : Endpoint {
Name : "website-health" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[DOMAIN_EXPIRATION] > 100h" } ,
} ,
ExpectedResult : & Result {
Success : true ,
Connected : true ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
{ Condition : "[DOMAIN_EXPIRATION] > 100h" , Success : true } ,
} ,
DomainExpiration : 999999 * time . Hour , // Note that this test only checks if it's non-zero.
} ,
MockRoundTripper : test . MockRoundTripper ( func ( r * http . Request ) * http . Response {
return & http . Response { StatusCode : http . StatusOK , Body : http . NoBody }
} ) ,
} ,
{
Name : "endpoint-that-will-time-out-and-hidden-hostname" ,
Endpoint : Endpoint {
Name : "endpoint-that-will-time-out" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[CONNECTED] == true" } ,
UIConfig : & ui . Config { HideHostname : true } ,
ClientConfig : & client . Config { Timeout : time . Millisecond } ,
} ,
ExpectedResult : & Result {
Success : false ,
Connected : false ,
Hostname : "" , // Because Endpoint.UIConfig.HideHostname is true, this should be empty.
ConditionResults : [ ] * ConditionResult {
{ Condition : "[CONNECTED] (false) == true" , Success : false } ,
} ,
// Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
DomainExpiration : 0 ,
// Because Endpoint.UIConfig.HideHostname is true, the hostname should be replaced by <redacted>.
Errors : [ ] string { ` Get "https://<redacted>/health": context deadline exceeded (Client.Timeout exceeded while awaiting headers) ` } ,
} ,
MockRoundTripper : nil ,
} ,
{
Name : "endpoint-that-will-time-out-and-hidden-url" ,
Endpoint : Endpoint {
Name : "endpoint-that-will-time-out" ,
URL : "https://twin.sh/health" ,
Conditions : [ ] Condition { "[CONNECTED] == true" } ,
UIConfig : & ui . Config { HideURL : true } ,
ClientConfig : & client . Config { Timeout : time . Millisecond } ,
} ,
ExpectedResult : & Result {
Success : false ,
Connected : false ,
Hostname : "twin.sh" ,
ConditionResults : [ ] * ConditionResult {
{ Condition : "[CONNECTED] (false) == true" , Success : false } ,
} ,
// Because there's no [DOMAIN_EXPIRATION] condition, this is not resolved, so it should be 0.
DomainExpiration : 0 ,
// Because Endpoint.UIConfig.HideURL is true, the URL should be replaced by <redacted>.
Errors : [ ] string { ` Get "<redacted>": context deadline exceeded (Client.Timeout exceeded while awaiting headers) ` } ,
} ,
MockRoundTripper : nil ,
} ,
}
for _ , scenario := range scenarios {
t . Run ( scenario . Name , func ( t * testing . T ) {
if scenario . MockRoundTripper != nil {
mockClient := & http . Client { Transport : scenario . MockRoundTripper }
if scenario . Endpoint . ClientConfig != nil && scenario . Endpoint . ClientConfig . Timeout > 0 {
mockClient . Timeout = scenario . Endpoint . ClientConfig . Timeout
}
client . InjectHTTPClient ( mockClient )
} else {
client . InjectHTTPClient ( nil )
}
scenario . Endpoint . ValidateAndSetDefaults ( )
result := scenario . Endpoint . EvaluateHealth ( )
if result . Success != scenario . ExpectedResult . Success {
t . Errorf ( "Expected success to be %v, got %v" , scenario . ExpectedResult . Success , result . Success )
}
if result . Connected != scenario . ExpectedResult . Connected {
t . Errorf ( "Expected connected to be %v, got %v" , scenario . ExpectedResult . Connected , result . Connected )
}
if result . Hostname != scenario . ExpectedResult . Hostname {
t . Errorf ( "Expected hostname to be %v, got %v" , scenario . ExpectedResult . Hostname , result . Hostname )
}
if len ( result . ConditionResults ) != len ( scenario . ExpectedResult . ConditionResults ) {
t . Errorf ( "Expected %v condition results, got %v" , len ( scenario . ExpectedResult . ConditionResults ) , len ( result . ConditionResults ) )
} else {
for i , conditionResult := range result . ConditionResults {
if conditionResult . Condition != scenario . ExpectedResult . ConditionResults [ i ] . Condition {
t . Errorf ( "Expected condition to be %v, got %v" , scenario . ExpectedResult . ConditionResults [ i ] . Condition , conditionResult . Condition )
}
if conditionResult . Success != scenario . ExpectedResult . ConditionResults [ i ] . Success {
t . Errorf ( "Expected success of condition '%s' to be %v, got %v" , conditionResult . Condition , scenario . ExpectedResult . ConditionResults [ i ] . Success , conditionResult . Success )
}
}
}
if len ( result . Errors ) != len ( scenario . ExpectedResult . Errors ) {
t . Errorf ( "Expected %v errors, got %v" , len ( scenario . ExpectedResult . Errors ) , len ( result . Errors ) )
} else {
for i , err := range result . Errors {
if err != scenario . ExpectedResult . Errors [ i ] {
t . Errorf ( "Expected error to be %v, got %v" , scenario . ExpectedResult . Errors [ i ] , err )
}
}
}
if result . DomainExpiration != scenario . ExpectedResult . DomainExpiration {
// Note that DomainExpiration is only resolved if there's a condition with the DomainExpirationPlaceholder in it.
// In other words, if there's no condition with [DOMAIN_EXPIRATION] in it, the DomainExpiration field will be 0.
// Because this is a live call, mocking it would be too much of a pain, so we're just going to check if
// the actual value is non-zero when the expected result is non-zero.
if scenario . ExpectedResult . DomainExpiration . Hours ( ) > 0 && ! ( result . DomainExpiration . Hours ( ) > 0 ) {
t . Errorf ( "Expected domain expiration to be non-zero, got %v" , result . DomainExpiration )
}
}
} )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_IsEnabled ( t * testing . T ) {
if ! ( Endpoint { Enabled : nil } ) . IsEnabled ( ) {
t . Error ( "endpoint.IsEnabled() should've returned true, because Enabled was set to nil" )
2021-09-18 17:52:11 +02:00
}
2021-10-23 22:47:12 +02:00
if value := false ; ( Endpoint { Enabled : & value } ) . IsEnabled ( ) {
t . Error ( "endpoint.IsEnabled() should've returned false, because Enabled was set to false" )
2021-09-18 17:52:11 +02:00
}
2021-10-23 22:47:12 +02:00
if value := true ; ! ( Endpoint { Enabled : & value } ) . IsEnabled ( ) {
t . Error ( "Endpoint.IsEnabled() should've returned true, because Enabled was set to true" )
2021-09-18 17:52:11 +02:00
}
}
2022-05-17 03:10:45 +02:00
func TestEndpoint_Type ( t * testing . T ) {
2022-09-02 03:09:36 +02:00
type args struct {
2022-05-17 03:10:45 +02:00
URL string
DNS * DNS
}
tests := [ ] struct {
2022-09-02 03:09:36 +02:00
args args
want EndpointType
} {
{
args : args {
2022-11-16 03:48:14 +01:00
URL : "8.8.8.8" ,
2022-09-02 03:09:36 +02:00
DNS : & DNS {
QueryType : "A" ,
QueryName : "example.com" ,
} ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
want : EndpointTypeDNS ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "tcp://127.0.0.1:6379" ,
} ,
want : EndpointTypeTCP ,
} ,
{
args : args {
URL : "icmp://example.com" ,
} ,
want : EndpointTypeICMP ,
} ,
2022-11-10 01:22:13 +01:00
{
args : args {
URL : "sctp://example.com" ,
} ,
want : EndpointTypeSCTP ,
} ,
{
args : args {
URL : "udp://example.com" ,
} ,
want : EndpointTypeUDP ,
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "starttls://smtp.gmail.com:587" ,
} ,
want : EndpointTypeSTARTTLS ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "tls://example.com:443" ,
} ,
want : EndpointTypeTLS ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "https://twin.sh/health" ,
} ,
want : EndpointTypeHTTP ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "invalid://example.org" ,
} ,
want : EndpointTypeUNKNOWN ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
{
args : args {
URL : "no-scheme" ,
} ,
want : EndpointTypeUNKNOWN ,
2022-05-17 03:10:45 +02:00
} ,
2022-09-02 03:09:36 +02:00
}
2022-05-17 03:10:45 +02:00
for _ , tt := range tests {
t . Run ( string ( tt . want ) , func ( t * testing . T ) {
endpoint := Endpoint {
2022-09-02 03:09:36 +02:00
URL : tt . args . URL ,
DNS : tt . args . DNS ,
2022-05-17 03:10:45 +02:00
}
if got := endpoint . Type ( ) ; got != tt . want {
t . Errorf ( "Endpoint.Type() = %v, want %v" , got , tt . want )
}
} )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_ValidateAndSetDefaults ( t * testing . T ) {
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { Condition ( "[STATUS] == 200" ) } ,
2021-05-19 04:29:15 +02:00
Alerts : [ ] * alert . Alert { { Type : alert . TypePagerDuty } } ,
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
if endpoint . ClientConfig == nil {
2021-07-29 03:41:26 +02:00
t . Error ( "client configuration should've been set to the default configuration" )
} else {
2021-10-23 22:47:12 +02:00
if endpoint . ClientConfig . Insecure != client . GetDefaultConfig ( ) . Insecure {
t . Errorf ( "Default client configuration should've set Insecure to %v, got %v" , client . GetDefaultConfig ( ) . Insecure , endpoint . ClientConfig . Insecure )
2021-07-29 03:41:26 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . ClientConfig . IgnoreRedirect != client . GetDefaultConfig ( ) . IgnoreRedirect {
t . Errorf ( "Default client configuration should've set IgnoreRedirect to %v, got %v" , client . GetDefaultConfig ( ) . IgnoreRedirect , endpoint . ClientConfig . IgnoreRedirect )
2021-07-29 03:41:26 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . ClientConfig . Timeout != client . GetDefaultConfig ( ) . Timeout {
t . Errorf ( "Default client configuration should've set Timeout to %v, got %v" , client . GetDefaultConfig ( ) . Timeout , endpoint . ClientConfig . Timeout )
2021-07-29 03:41:26 +02:00
}
}
2021-10-23 22:47:12 +02:00
if endpoint . Method != "GET" {
t . Error ( "Endpoint method should've defaulted to GET" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . Interval != time . Minute {
t . Error ( "Endpoint interval should've defaulted to 1 minute" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . Headers == nil {
t . Error ( "Endpoint headers should've defaulted to an empty map" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if len ( endpoint . Alerts ) != 1 {
t . Error ( "Endpoint should've had 1 alert" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . Alerts [ 0 ] . IsEnabled ( ) {
t . Error ( "Endpoint alert should've defaulted to disabled" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . Alerts [ 0 ] . SuccessThreshold != 2 {
t . Error ( "Endpoint alert should've defaulted to a success threshold of 2" )
2020-10-22 04:21:51 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . Alerts [ 0 ] . FailureThreshold != 3 {
t . Error ( "Endpoint alert should've defaulted to a failure threshold of 3" )
2020-10-22 04:21:51 +02:00
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_ValidateAndSetDefaultsWithClientConfig ( t * testing . T ) {
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2022-09-16 03:23:14 +02:00
Conditions : [ ] Condition { Condition ( "[STATUS] == 200" ) } ,
2021-07-29 03:41:26 +02:00
ClientConfig : & client . Config {
Insecure : true ,
IgnoreRedirect : true ,
Timeout : 0 ,
} ,
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
if endpoint . ClientConfig == nil {
2021-07-29 03:41:26 +02:00
t . Error ( "client configuration should've been set to the default configuration" )
} else {
2021-10-23 22:47:12 +02:00
if ! endpoint . ClientConfig . Insecure {
t . Error ( "endpoint.ClientConfig.Insecure should've been set to true" )
2021-07-29 03:41:26 +02:00
}
2021-10-23 22:47:12 +02:00
if ! endpoint . ClientConfig . IgnoreRedirect {
t . Error ( "endpoint.ClientConfig.IgnoreRedirect should've been set to true" )
2021-07-29 03:41:26 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . ClientConfig . Timeout != client . GetDefaultConfig ( ) . Timeout {
t . Error ( "endpoint.ClientConfig.Timeout should've been set to 10s, because the timeout value entered is not set or invalid" )
2021-07-29 03:41:26 +02:00
}
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_ValidateAndSetDefaultsWithDNS ( t * testing . T ) {
endpoint := & Endpoint {
2020-11-18 01:00:16 +01:00
Name : "dns-test" ,
2022-09-16 03:23:14 +02:00
URL : "https://example.com" ,
2020-11-18 00:55:31 +01:00
DNS : & DNS {
QueryType : "A" ,
QueryName : "example.com" ,
} ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { Condition ( "[DNS_RCODE] == NOERROR" ) } ,
2020-11-18 00:55:31 +01:00
}
2021-10-23 22:47:12 +02:00
err := endpoint . ValidateAndSetDefaults ( )
2021-05-19 05:27:43 +02:00
if err != nil {
2022-09-16 03:23:14 +02:00
t . Error ( "did not expect an error, got" , err )
2021-05-19 05:27:43 +02:00
}
2021-10-23 22:47:12 +02:00
if endpoint . DNS . QueryName != "example.com." {
t . Error ( "Endpoint.dns.query-name should be formatted with . suffix" )
2020-11-18 00:55:31 +01:00
}
}
2022-09-16 03:23:14 +02:00
func TestEndpoint_ValidateAndSetDefaultsWithSimpleErrors ( t * testing . T ) {
scenarios := [ ] struct {
endpoint * Endpoint
expectedErr error
} {
{
endpoint : & Endpoint {
Name : "" ,
URL : "https://example.com" ,
Conditions : [ ] Condition { Condition ( "[STATUS] == 200" ) } ,
} ,
expectedErr : ErrEndpointWithNoName ,
} ,
{
endpoint : & Endpoint {
Name : "endpoint-with-no-url" ,
URL : "" ,
Conditions : [ ] Condition { Condition ( "[STATUS] == 200" ) } ,
} ,
expectedErr : ErrEndpointWithNoURL ,
} ,
{
endpoint : & Endpoint {
Name : "endpoint-with-no-conditions" ,
URL : "https://example.com" ,
Conditions : nil ,
} ,
expectedErr : ErrEndpointWithNoCondition ,
} ,
{
endpoint : & Endpoint {
Name : "domain-expiration-with-bad-interval" ,
URL : "https://example.com" ,
Interval : time . Minute ,
Conditions : [ ] Condition { Condition ( "[DOMAIN_EXPIRATION] > 720h" ) } ,
} ,
expectedErr : ErrInvalidEndpointIntervalForDomainExpirationPlaceholder ,
} ,
{
endpoint : & Endpoint {
Name : "domain-expiration-with-good-interval" ,
URL : "https://example.com" ,
Interval : 5 * time . Minute ,
Conditions : [ ] Condition { Condition ( "[DOMAIN_EXPIRATION] > 720h" ) } ,
} ,
expectedErr : nil ,
} ,
}
for _ , scenario := range scenarios {
t . Run ( scenario . endpoint . Name , func ( t * testing . T ) {
if err := scenario . endpoint . ValidateAndSetDefaults ( ) ; err != scenario . expectedErr {
t . Errorf ( "Expected error %v, got %v" , scenario . expectedErr , err )
}
} )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_buildHTTPRequest ( t * testing . T ) {
2020-12-19 00:06:57 +01:00
condition := Condition ( "[STATUS] == 200" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition } ,
2020-12-19 00:06:57 +01:00
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
request := endpoint . buildHTTPRequest ( )
2020-12-19 00:40:11 +01:00
if request . Method != "GET" {
t . Error ( "request.Method should've been GET, but was" , request . Method )
}
2021-09-18 18:42:11 +02:00
if request . Host != "twin.sh" {
t . Error ( "request.Host should've been twin.sh, but was" , request . Host )
2020-12-19 00:06:57 +01:00
}
2020-12-31 01:56:12 +01:00
if userAgent := request . Header . Get ( "User-Agent" ) ; userAgent != GatusUserAgent {
t . Errorf ( "request.Header.Get(User-Agent) should've been %s, but was %s" , GatusUserAgent , userAgent )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_buildHTTPRequestWithCustomUserAgent ( t * testing . T ) {
2020-12-31 01:56:12 +01:00
condition := Condition ( "[STATUS] == 200" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition } ,
2020-12-31 01:56:12 +01:00
Headers : map [ string ] string {
"User-Agent" : "Test/2.0" ,
} ,
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
request := endpoint . buildHTTPRequest ( )
2020-12-31 01:56:12 +01:00
if request . Method != "GET" {
t . Error ( "request.Method should've been GET, but was" , request . Method )
}
2021-09-18 18:42:11 +02:00
if request . Host != "twin.sh" {
t . Error ( "request.Host should've been twin.sh, but was" , request . Host )
2020-12-31 01:56:12 +01:00
}
if userAgent := request . Header . Get ( "User-Agent" ) ; userAgent != "Test/2.0" {
t . Errorf ( "request.Header.Get(User-Agent) should've been %s, but was %s" , "Test/2.0" , userAgent )
}
2020-12-19 00:06:57 +01:00
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_buildHTTPRequestWithHostHeader ( t * testing . T ) {
2020-12-19 00:06:57 +01:00
condition := Condition ( "[STATUS] == 200" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2020-12-19 00:40:11 +01:00
Method : "POST" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition } ,
2020-12-19 00:06:57 +01:00
Headers : map [ string ] string {
"Host" : "example.com" ,
} ,
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
request := endpoint . buildHTTPRequest ( )
2020-12-19 00:40:11 +01:00
if request . Method != "POST" {
t . Error ( "request.Method should've been POST, but was" , request . Method )
}
2020-12-19 00:06:57 +01:00
if request . Host != "example.com" {
2020-12-19 00:40:11 +01:00
t . Error ( "request.Host should've been example.com, but was" , request . Host )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_buildHTTPRequestWithGraphQLEnabled ( t * testing . T ) {
2020-12-19 00:40:11 +01:00
condition := Condition ( "[STATUS] == 200" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-graphql" ,
URL : "https://twin.sh/graphql" ,
2020-12-19 00:40:11 +01:00
Method : "POST" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition } ,
2020-12-19 00:40:11 +01:00
GraphQL : true ,
Body : ` {
2021-03-06 02:33:06 +01:00
users ( gender : "female" ) {
2020-12-19 00:40:11 +01:00
id
name
gender
avatar
}
} ` ,
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
request := endpoint . buildHTTPRequest ( )
2020-12-19 00:40:11 +01:00
if request . Method != "POST" {
t . Error ( "request.Method should've been POST, but was" , request . Method )
}
if contentType := request . Header . Get ( ContentTypeHeader ) ; contentType != "application/json" {
t . Error ( "request.Header.Content-Type should've been application/json, but was" , contentType )
}
2021-12-03 07:44:17 +01:00
body , _ := io . ReadAll ( request . Body )
2020-12-19 00:40:11 +01:00
if ! strings . HasPrefix ( string ( body ) , "{\"query\":" ) {
2021-03-09 03:30:11 +01:00
t . Error ( "request.body should've started with '{\"query\":', but it didn't:" , string ( body ) )
2020-12-19 00:06:57 +01:00
}
}
2020-09-25 01:49:32 +02:00
func TestIntegrationEvaluateHealth ( t * testing . T ) {
2020-04-15 01:20:00 +02:00
condition := Condition ( "[STATUS] == 200" )
2021-03-09 03:30:11 +01:00
bodyCondition := Condition ( "[BODY].status == UP" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-09-18 18:42:11 +02:00
Name : "website-health" ,
URL : "https://twin.sh/health" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition , bodyCondition } ,
2020-04-15 01:20:00 +02:00
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
result := endpoint . EvaluateHealth ( )
2020-04-15 01:20:00 +02:00
if ! result . ConditionResults [ 0 ] . Success {
t . Errorf ( "Condition '%s' should have been a success" , condition )
}
2020-10-05 01:55:19 +02:00
if ! result . Connected {
t . Error ( "Because the connection has been established, result.Connected should've been true" )
}
2020-04-15 01:20:00 +02:00
if ! result . Success {
t . Error ( "Because all conditions passed, this should have been a success" )
}
2022-03-16 01:52:46 +01:00
if result . Hostname != "twin.sh" {
t . Error ( "result.Hostname should've been twin.sh, but was" , result . Hostname )
}
2020-04-15 01:20:00 +02:00
}
2022-03-16 01:52:46 +01:00
func TestIntegrationEvaluateHealthWithInvalidCondition ( t * testing . T ) {
condition := Condition ( "[STATUS] invalid 200" )
endpoint := Endpoint {
Name : "invalid-condition" ,
URL : "https://twin.sh/health" ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { condition } ,
2022-03-16 01:52:46 +01:00
}
if err := endpoint . ValidateAndSetDefaults ( ) ; err != nil {
// XXX: Should this really not return an error? After all, the condition is not valid and conditions are part of the endpoint...
t . Error ( "endpoint validation should've been successful, but wasn't" )
}
result := endpoint . EvaluateHealth ( )
if result . Success {
t . Error ( "Because one of the conditions was invalid, result.Success should have been false" )
}
if len ( result . Errors ) == 0 {
t . Error ( "There should've been an error" )
}
}
2022-06-16 23:53:03 +02:00
func TestIntegrationEvaluateHealthWithErrorAndHideURL ( t * testing . T ) {
endpoint := Endpoint {
Name : "invalid-url" ,
URL : "https://httpstat.us/200?sleep=100" ,
Conditions : [ ] Condition { Condition ( "[STATUS] == 200" ) } ,
ClientConfig : & client . Config {
Timeout : 1 * time . Millisecond ,
} ,
UIConfig : & ui . Config {
HideURL : true ,
} ,
}
endpoint . ValidateAndSetDefaults ( )
result := endpoint . EvaluateHealth ( )
if result . Success {
t . Error ( "Because one of the conditions was invalid, result.Success should have been false" )
}
if len ( result . Errors ) == 0 {
t . Error ( "There should've been an error" )
}
if ! strings . Contains ( result . Errors [ 0 ] , "<redacted>" ) || strings . Contains ( result . Errors [ 0 ] , endpoint . URL ) {
t . Error ( "result.Errors[0] should've had the URL redacted because ui.hide-url is set to true" )
}
}
2020-11-18 00:55:31 +01:00
func TestIntegrationEvaluateHealthForDNS ( t * testing . T ) {
conditionSuccess := Condition ( "[DNS_RCODE] == NOERROR" )
conditionBody := Condition ( "[BODY] == 93.184.216.34" )
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-03-09 03:30:11 +01:00
Name : "example" ,
2022-11-16 03:42:11 +01:00
URL : "8.8.8.8" ,
2020-11-18 00:55:31 +01:00
DNS : & DNS {
QueryType : "A" ,
QueryName : "example.com." ,
} ,
2022-06-14 01:15:30 +02:00
Conditions : [ ] Condition { conditionSuccess , conditionBody } ,
2020-11-18 00:55:31 +01:00
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
result := endpoint . EvaluateHealth ( )
2020-11-18 00:55:31 +01:00
if ! result . ConditionResults [ 0 ] . Success {
2022-09-07 03:22:02 +02:00
t . Errorf ( "Conditions '%s' and '%s' should have been a success" , conditionSuccess , conditionBody )
2020-11-18 00:55:31 +01:00
}
if ! result . Connected {
t . Error ( "Because the connection has been established, result.Connected should've been true" )
}
if ! result . Success {
t . Error ( "Because all conditions passed, this should have been a success" )
}
}
2020-12-25 06:07:18 +01:00
func TestIntegrationEvaluateHealthForICMP ( t * testing . T ) {
2021-10-23 22:47:12 +02:00
endpoint := Endpoint {
2021-03-09 03:30:11 +01:00
Name : "icmp-test" ,
2020-12-25 09:05:47 +01:00
URL : "icmp://127.0.0.1" ,
2022-09-07 03:22:02 +02:00
Conditions : [ ] Condition { "[CONNECTED] == true" } ,
2020-04-15 01:20:00 +02:00
}
2021-10-23 22:47:12 +02:00
endpoint . ValidateAndSetDefaults ( )
result := endpoint . EvaluateHealth ( )
2020-12-25 06:07:18 +01:00
if ! result . ConditionResults [ 0 ] . Success {
2022-09-07 03:22:02 +02:00
t . Errorf ( "Conditions '%s' should have been a success" , endpoint . Conditions [ 0 ] )
2020-04-15 01:20:00 +02:00
}
2020-10-05 01:55:19 +02:00
if ! result . Connected {
t . Error ( "Because the connection has been established, result.Connected should've been true" )
}
2020-12-25 06:07:18 +01:00
if ! result . Success {
t . Error ( "Because all conditions passed, this should have been a success" )
2020-04-15 01:20:00 +02:00
}
}
2020-12-29 23:27:58 +01:00
2022-10-10 03:34:36 +02:00
func TestEndpoint_DisplayName ( t * testing . T ) {
if endpoint := ( Endpoint { Name : "n" } ) ; endpoint . DisplayName ( ) != "n" {
t . Error ( "endpoint.DisplayName() should've been 'n', but was" , endpoint . DisplayName ( ) )
}
if endpoint := ( Endpoint { Group : "g" , Name : "n" } ) ; endpoint . DisplayName ( ) != "g/n" {
t . Error ( "endpoint.DisplayName() should've been 'g/n', but was" , endpoint . DisplayName ( ) )
}
}
2021-10-23 22:47:12 +02:00
func TestEndpoint_getIP ( t * testing . T ) {
endpoint := Endpoint {
2021-03-09 03:30:11 +01:00
Name : "invalid-url-test" ,
2020-12-29 23:27:58 +01:00
URL : "" ,
2022-09-07 03:22:02 +02:00
Conditions : [ ] Condition { "[CONNECTED] == true" } ,
2020-12-29 23:27:58 +01:00
}
result := & Result { }
2021-10-23 22:47:12 +02:00
endpoint . getIP ( result )
2020-12-29 23:27:58 +01:00
if len ( result . Errors ) == 0 {
2021-10-23 22:47:12 +02:00
t . Error ( "endpoint.getIP(result) should've thrown an error because the URL is invalid, thus cannot be parsed" )
2020-12-29 23:27:58 +01:00
}
}
2021-03-09 03:30:11 +01:00
2022-09-07 03:22:02 +02:00
func TestEndpoint_needsToReadBody ( t * testing . T ) {
2021-03-09 03:30:11 +01:00
statusCondition := Condition ( "[STATUS] == 200" )
bodyCondition := Condition ( "[BODY].status == UP" )
bodyConditionWithLength := Condition ( "len([BODY].tags) > 0" )
2022-06-14 01:15:30 +02:00
if ( & Endpoint { Conditions : [ ] Condition { statusCondition } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected false, got true" )
}
2022-06-14 01:15:30 +02:00
if ! ( & Endpoint { Conditions : [ ] Condition { bodyCondition } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected true, got false" )
}
2022-06-14 01:15:30 +02:00
if ! ( & Endpoint { Conditions : [ ] Condition { bodyConditionWithLength } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected true, got false" )
}
2022-06-14 01:15:30 +02:00
if ! ( & Endpoint { Conditions : [ ] Condition { statusCondition , bodyCondition } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected true, got false" )
}
2022-06-14 01:15:30 +02:00
if ! ( & Endpoint { Conditions : [ ] Condition { bodyCondition , statusCondition } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected true, got false" )
}
2022-06-14 01:15:30 +02:00
if ! ( & Endpoint { Conditions : [ ] Condition { bodyConditionWithLength , statusCondition } } ) . needsToReadBody ( ) {
2021-03-09 03:30:11 +01:00
t . Error ( "expected true, got false" )
}
}
2022-09-07 03:22:02 +02:00
func TestEndpoint_needsToRetrieveDomainExpiration ( t * testing . T ) {
if ( & Endpoint { Conditions : [ ] Condition { "[STATUS] == 200" } } ) . needsToRetrieveDomainExpiration ( ) {
t . Error ( "expected false, got true" )
}
if ! ( & Endpoint { Conditions : [ ] Condition { "[STATUS] == 200" , "[DOMAIN_EXPIRATION] < 720h" } } ) . needsToRetrieveDomainExpiration ( ) {
t . Error ( "expected true, got false" )
}
}
func TestEndpoint_needsToRetrieveIP ( t * testing . T ) {
if ( & Endpoint { Conditions : [ ] Condition { "[STATUS] == 200" } } ) . needsToRetrieveIP ( ) {
t . Error ( "expected false, got true" )
}
if ! ( & Endpoint { Conditions : [ ] Condition { "[STATUS] == 200" , "[IP] == 127.0.0.1" } } ) . needsToRetrieveIP ( ) {
t . Error ( "expected true, got false" )
}
}