request_test.go

package grid

import (
	"errors"
	"strings"
	"testing"
)

func TestRespondWithAlreadyResponded(t *testing.T) {
	t.Parallel()
	req := &request{finished: true}
	err := req.Respond("some-msg")
	if err != ErrAlreadyResponded {
		t.Fatal("expected error")
	}
}

func TestResponedWithError(t *testing.T) {
	t.Parallel()
	expected := errors.New("expected-error")

	req := &request{failure: make(chan error, 1)}
	err := req.Respond(expected)
	if err != nil {
		t.Fatal(err)
	}

	select {
	case err := <-req.failure:
		if err != expected {
			t.Fatal(err)
		}
	default:
		t.Fatal("expected error")
	}
}

type unregisteredMsg struct{}

func TestUnregisteredMessageError(t *testing.T) {
	t.Parallel()
	req := &request{}
	err := req.Respond(&unregisteredMsg{})
	if err == nil {
		t.Fatal("expected error")
	}
	if !strings.Contains(err.Error(), "unregistered message type") {
		t.Fatal("expected specific error")
	}
}