onix/pkg/model/error_test.go

package model

import (
	"errors"
	"fmt"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
	"gopkg.in/yaml.v2"
)

// NewSignValidationErrf creates a new SignValidationErr with a formatted error message.
func NewSignValidationErrf(format string, a ...any) *SignValidationErr {
	return &SignValidationErr{fmt.Errorf(format, a...)}
}

// NewNotFoundErrf creates a new NotFoundErr with a formatted error message.
func NewNotFoundErrf(format string, a ...any) *NotFoundErr {
	return &NotFoundErr{fmt.Errorf(format, a...)}
}

// NewBadReqErrf creates a new BadReqErr with a formatted error message.
func NewBadReqErrf(format string, a ...any) *BadReqErr {
	return &BadReqErr{fmt.Errorf(format, a...)}
}

func TestError_Error(t *testing.T) {
	err := &Error{
		Code:    "404",
		Paths:   "/api/v1/user",
		Message: "User not found",
	}

	expected := "Error: Code=404, Path=/api/v1/user, Message=User not found"
	actual := err.Error()

	if actual != expected {
		t.Errorf("err.Error() = %s, want %s",
			actual, expected)
	}

}

func TestSchemaValidationErr_Error(t *testing.T) {
	schemaErr := &SchemaValidationErr{
		Errors: []Error{
			{Paths: "/user", Message: "Field required"},
			{Paths: "/email", Message: "Invalid format"},
		},
	}

	expected := "/user: Field required; /email: Invalid format"
	actual := schemaErr.Error()

	if actual != expected {
		t.Errorf("err.Error() = %s, want %s",
			actual, expected)
	}
}

func TestSchemaValidationErr_BecknError(t *testing.T) {
	schemaErr := &SchemaValidationErr{
		Errors: []Error{
			{Paths: "/user", Message: "Field required"},
		},
	}

	beErr := schemaErr.BecknError()
	expected := "Bad Request"
	if beErr.Code != expected {
		t.Errorf("err.Error() = %s, want %s",
			beErr.Code, expected)
	}
}

func TestSignValidationErr_BecknError(t *testing.T) {
	signErr := NewSignValidationErr(errors.New("signature failed"))
	beErr := signErr.BecknError()

	expectedMsg := "Signature Validation Error: signature failed"
	if beErr.Message != expectedMsg {
		t.Errorf("err.Error() = %s, want %s",
			beErr.Message, expectedMsg)
	}

}

func TestNewSignValidationErrf(t *testing.T) {
	signErr := NewSignValidationErrf("error %s", "signature failed")
	expected := "error signature failed"
	if signErr.Error() != expected {
		t.Errorf("err.Error() = %s, want %s",
			signErr.Error(), expected)
	}
}

func TestNewSignValidationErr(t *testing.T) {
	err := errors.New("signature error")
	signErr := NewSignValidationErr(err)

	if signErr.Error() != err.Error() {
		t.Errorf("err.Error() = %s, want %s", err.Error(),
			signErr.Error())
	}
}

func TestBadReqErr_BecknError(t *testing.T) {
	badReqErr := NewBadReqErr(errors.New("invalid input"))
	beErr := badReqErr.BecknError()

	expectedMsg := "BAD Request: invalid input"
	if beErr.Message != expectedMsg {
		t.Errorf("err.Error() = %s, want %s",
			beErr.Message, expectedMsg)
	}
}

func TestNewBadReqErrf(t *testing.T) {
	badReqErr := NewBadReqErrf("invalid field %s", "name")
	expected := "invalid field name"
	if badReqErr.Error() != expected {
		t.Errorf("err.Error() = %s, want %s",
			badReqErr, expected)
	}
}

func TestNewBadReqErr(t *testing.T) {
	err := errors.New("bad request")
	badReqErr := NewBadReqErr(err)

	if badReqErr.Error() != err.Error() {
		t.Errorf("err.Error() = %s, want %s",
			badReqErr.Error(), err.Error())
	}

}

func TestNotFoundErr_BecknError(t *testing.T) {
	notFoundErr := NewNotFoundErr(errors.New("resource not found"))
	beErr := notFoundErr.BecknError()

	expectedMsg := "Endpoint not found: resource not found"
	if beErr.Message != expectedMsg {
		t.Errorf("err.Error() = %s, want %s",
			beErr.Message, expectedMsg)
	}
}

func TestNewNotFoundErrf(t *testing.T) {
	notFoundErr := NewNotFoundErrf("resource %s not found", "user")
	expected := "resource user not found"
	if notFoundErr.Error() != expected {
		t.Errorf("err.Error() = %s, want %s",
			notFoundErr.Error(), expected)
	}
}

func TestNewNotFoundErr(t *testing.T) {
	err := errors.New("not found")
	notFoundErr := NewNotFoundErr(err)

	if notFoundErr.Error() != err.Error() {
		t.Errorf("err.Error() = %s, want %s",
			notFoundErr.Error(), err.Error())
	}
}

func TestRole_UnmarshalYAML_ValidRole(t *testing.T) {
	var role Role
	yamlData := []byte("bap")

	err := yaml.Unmarshal(yamlData, &role)
	assert.NoError(t, err) //TODO: should replace assert here
	assert.Equal(t, RoleBAP, role)
}

func TestRole_UnmarshalYAML_InvalidRole(t *testing.T) {
	var role Role
	yamlData := []byte("invalid")

	err := yaml.Unmarshal(yamlData, &role)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "invalid Role")
}

func TestSchemaValidationErr_BecknError_NoErrors(t *testing.T) {
	schemaValidationErr := &SchemaValidationErr{Errors: nil}
	beErr := schemaValidationErr.BecknError()

	expectedMsg := "Schema validation error."
	expectedCode := http.StatusText(http.StatusBadRequest)

	if beErr.Message != expectedMsg {
		t.Errorf("beErr.Message = %s, want %s", beErr.Message, expectedMsg)
	}
	if beErr.Code != expectedCode {
		t.Errorf("beErr.Code = %s, want %s", beErr.Code, expectedCode)
	}
}

func TestParseContextKey_ValidKeys(t *testing.T) {
	tests := []struct {
		input    string
		expected ContextKey
	}{
		{"message_id", ContextKeyMsgID},
		{"subscriber_id", ContextKeySubscriberID},
		{"model_id", ContextKeyModelID},
	}

	for _, tt := range tests {
		key, err := ParseContextKey(tt.input)
		if err != nil {
			t.Errorf("unexpected error for input %s: %v", tt.input, err)
		}
		if key != tt.expected {
			t.Errorf("expected %s, got %s", tt.expected, key)
		}
	}
}

func TestParseContextKey_InvalidKey(t *testing.T) {
	_, err := ParseContextKey("invalid_key")
	if err == nil {
		t.Error("expected error for invalid context key, got nil")
	}
}

func TestContextKey_UnmarshalYAML_Valid(t *testing.T) {
	yamlData := []byte("message_id")
	var key ContextKey

	err := yaml.Unmarshal(yamlData, &key)
	if err != nil {
		t.Errorf("unexpected error: %v", err)
	}
	if key != ContextKeyMsgID {
		t.Errorf("expected %s, got %s", ContextKeyMsgID, key)
	}
}

func TestContextKey_UnmarshalYAML_Invalid(t *testing.T) {
	yamlData := []byte("invalid_key")
	var key ContextKey

	err := yaml.Unmarshal(yamlData, &key)
	if err == nil {
		t.Error("expected error for invalid context key, got nil")
	}
}