github
/
statping
mirror of https://github.com/statping/statping
1
0
Fork 0
Code Issues Releases Wiki Activity

Add more test cases for grpc checks 

Signed-off-by: thatInfrastructureGuy <thatInfrastructureGuy@gmail.com>
pull/806/head
thatInfrastructureGuy 2020-08-31 20:23:38 -07:00
parent 887f8cc9be
commit a6df4e09c9
No known key found for this signature in database
GPG Key ID: 3E9D4A7275BC5A6A
2 changed files with 49 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
types/services/routine.go
View File

@ -125,6 +125,9 @@ func CheckGrpc(s *Service, record bool) (*Service, error) {
if err != nil {
// Unable to parse.
log.Warnln(fmt.Sprintf("GRPC Service: '%s', Unable to parse URL: '%v'", s.Name, s.Domain))
if record {
RecordFailure(s, fmt.Sprintf("Unable to parse GRPC domain %v, %v", s.Domain, err), "parse_domain")
}
}
// Set domain as hostname without port number.

47
types/services/routine_test.go
View File

@ -97,6 +97,51 @@ var testdata = []struct {
VerifySSL: null.NewNullBool(true),
},
},
{
grpcService: func(port int) *grpc.Server {
return grpcServer(port, true)
},
clientChecker: &Service{
Name: "Check GRPC Server with http:// url",
Domain: "http://localhost",
Port: 50058,
Expected: null.NewNullString("status:SERVING"),
ExpectedStatus: 1,
Type: "grpc",
Timeout: 1,
VerifySSL: null.NewNullBool(false),
},
},
{
grpcService: func(port int) *grpc.Server {
return grpcServer(port, true)
},
clientChecker: &Service{
Name: "Unparseable Url Error",
Domain: "http://local//host",
Port: 50059,
Expected: null.NewNullString(""),
ExpectedStatus: 0,
Type: "grpc",
Timeout: 1,
VerifySSL: null.NewNullBool(false),
},
},
{
grpcService: func(port int) *grpc.Server {
return grpcServer(50060, true)
},
clientChecker: &Service{
Name: "Check GRPC on HTTP server",
Domain: "https://google.com",
Port: 443,
Expected: null.NewNullString(""),
ExpectedStatus: 0,
Type: "grpc",
Timeout: 1,
VerifySSL: null.NewNullBool(false),
},
},
}
// grpcServer creates grpc Service with optional parameters.
@ -128,7 +173,7 @@ func TestCheckGrpc(t *testing.T) {
defer server.Stop()
v.clientChecker.CheckService(false)
if v.clientChecker.LastStatusCode != v.clientChecker.ExpectedStatus || strings.TrimSpace(v.clientChecker.LastResponse) != v.clientChecker.Expected.String {
t.Errorf("Expected message: '%v', Got message: '%v' , Expected Status: '%v', Got Status: '%v'", v.clientChecker.Expected, v.clientChecker.LastResponse, v.clientChecker.ExpectedStatus, v.clientChecker.LastStatusCode)
t.Errorf("Expected message: '%v', Got message: '%v' , Expected Status: '%v', Got Status: '%v'", v.clientChecker.Expected.String, v.clientChecker.LastResponse, v.clientChecker.ExpectedStatus, v.clientChecker.LastStatusCode)
}
})
}