Jeremy Tregunna
a0a1c0512f
All checks were successful
Go Tests / Run Tests (1.24.2) (push) Successful in 9m50s
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package transport
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
func TestBasicRequest(t *testing.T) {
|
|
// Test creating a request
|
|
payload := []byte("test payload")
|
|
req := NewRequest(TypeGet, payload)
|
|
|
|
// Test Type method
|
|
if req.Type() != TypeGet {
|
|
t.Errorf("Expected type %s, got %s", TypeGet, req.Type())
|
|
}
|
|
|
|
// Test Payload method
|
|
if string(req.Payload()) != string(payload) {
|
|
t.Errorf("Expected payload %s, got %s", string(payload), string(req.Payload()))
|
|
}
|
|
}
|
|
|
|
func TestBasicResponse(t *testing.T) {
|
|
// Test creating a response with no error
|
|
payload := []byte("test response")
|
|
resp := NewResponse(TypeGet, payload, nil)
|
|
|
|
// Test Type method
|
|
if resp.Type() != TypeGet {
|
|
t.Errorf("Expected type %s, got %s", TypeGet, resp.Type())
|
|
}
|
|
|
|
// Test Payload method
|
|
if string(resp.Payload()) != string(payload) {
|
|
t.Errorf("Expected payload %s, got %s", string(payload), string(resp.Payload()))
|
|
}
|
|
|
|
// Test Error method
|
|
if resp.Error() != nil {
|
|
t.Errorf("Expected nil error, got %v", resp.Error())
|
|
}
|
|
|
|
// Test creating a response with an error
|
|
testErr := errors.New("test error")
|
|
resp = NewResponse(TypeGet, payload, testErr)
|
|
|
|
if resp.Error() != testErr {
|
|
t.Errorf("Expected error %v, got %v", testErr, resp.Error())
|
|
}
|
|
}
|
|
|
|
func TestNewErrorResponse(t *testing.T) {
|
|
// Test creating an error response
|
|
testErr := errors.New("test error")
|
|
resp := NewErrorResponse(testErr)
|
|
|
|
// Test Type method
|
|
if resp.Type() != TypeError {
|
|
t.Errorf("Expected type %s, got %s", TypeError, resp.Type())
|
|
}
|
|
|
|
// Test Payload method - should contain error message
|
|
if string(resp.Payload()) != testErr.Error() {
|
|
t.Errorf("Expected payload %s, got %s", testErr.Error(), string(resp.Payload()))
|
|
}
|
|
|
|
// Test Error method
|
|
if resp.Error() != testErr {
|
|
t.Errorf("Expected error %v, got %v", testErr, resp.Error())
|
|
}
|
|
|
|
// Test with nil error
|
|
resp = NewErrorResponse(nil)
|
|
|
|
if resp.Type() != TypeError {
|
|
t.Errorf("Expected type %s, got %s", TypeError, resp.Type())
|
|
}
|
|
|
|
if len(resp.Payload()) != 0 {
|
|
t.Errorf("Expected empty payload, got %s", string(resp.Payload()))
|
|
}
|
|
|
|
if resp.Error() != nil {
|
|
t.Errorf("Expected nil error, got %v", resp.Error())
|
|
}
|
|
}
|