Я покрываю проект тестами, и для этого мне нужен фиктивный TCP-сервер, который мог бы принимать соединение, записывать / читать данные в / из него, закрывать его и т. Д ... Я нашел этот вопрос при переполнении стека, охватывающем фиктивное соединение, но оно не покрывает то, что мне действительно нужно проверить.
Моя идея основана на этой статье в качестве отправной точки, но когда я начал реализовывать канал, чтобы позволить серверу записывать некоторые данные во вновь открытое соединение, я застрял с неразрушаемой тупиковой ситуацией при записи в канал.
Чего я хочу добиться , так это записать некоторые данные в канал сервера, скажем, sendingQueue chan *[]byte, поэтому позже соответствующий []byte будет отправлен на вновь установленное соединение.
Во время этих небольших исследований я пытался отлаживать и распечатывать сообщения до / после отправки данных на канал и пытаться отправлять / читать данные с канала в разных местах программы.
Что я узнал :
Моя идея работает, если я добавляю данные непосредственно в handleConnection с
go func() {
f := []byte("test.")
t.sendingQueue <- &f
}()
Моя идея не сработает, если я отправлю данные на канал из TestUtils_TestingTCPServer_WritesRequest в любой форме, либо с func (t *TCPServer) Put(data *[]byte) (err error), либо напрямую с:
go func(queue chan *[]byte, data *[]byte) {
queue <- data
}(t.sendingQueue, &payload)
- Неважно, буферизован канал или нет.
Так что, очевидно, что-то не так с тем, как я отлаживаю свой код (я не углублялся в cli dlv, используя только IDE-отладчик), или с чем-то, что мне совершенно не хватает в работе с каналами go, goroutines или net Модуль .Conn.
Для удобства Публичная суть с полным кодом доступен. Примечание. В TestUtils_TestingTCPServer_WritesRequest есть часть // INIT, необходимая для запуска / отладки одиночного теста. Это должно быть закомментировано при запуске go test в каталоге.
utils.go:
// NewServer creates a new Server using given protocol
// and addr.
func NewTestingTCPServer(protocol, addr string) (*TCPServer, error) {
switch strings.ToLower(protocol) {
case "tcp":
return &TCPServer{
addr: addr,
sendingQueue: make(chan *[]byte, 10),
}, nil
case "udp":
}
return nil, errors.New("invalid protocol given")
}
// TCPServer holds the structure of our TCP
// implementation.
type TCPServer struct {
addr string
server net.Listener
sendingQueue chan *[]byte
}
func (t *TCPServer) Run() (err error) {}
func (t *TCPServer) Close() (err error) {}
func (t *TCPServer) Put(data *[]byte) (err error) {}
func (t *TCPServer) handleConnection(conn net.Conn){
// <...>
// Putting data here successfully sends it via freshly established
// Connection:
// go func() {
// f := []byte("test.")
// t.sendingQueue <- &f
// }()
for {
fmt.Printf("Started for loop\n")
select {
case data := <-readerChannel:
fmt.Printf("Read written data\n")
writeBuffer.Write(*data)
writeBuffer.Flush()
case data := <-t.sendingQueue:
fmt.Printf("Read pushed data\n")
writeBuffer.Write(*data)
writeBuffer.Flush()
case <-ticker:
fmt.Printf("Tick\n")
return
}
fmt.Printf("Finished for loop\n")
}
}
utils_test.go
func TestUtils_TestingTCPServer_WritesRequest(t *testing.T) {
payload := []byte("hello world\n")
// <...> In gist here is placed INIT piece, which
// is required to debug single test
fmt.Printf("Putting payload into queue\n")
// This doesn't affect channel
err = utilTestingSrv.Put(&payload)
assert.Nil(t, err)
// This doesn't work either
//go func(queue chan *[]byte, data *[]byte) {
// queue <- data
//}(utilTestingSrv.sendingQueue, &payload)
conn, err := net.Dial("tcp", ":41123")
if !assert.Nil(t, err) {
t.Error("could not connect to server: ", err)
}
defer conn.Close()
out := make([]byte, 1024)
if _, err := conn.Read(out); assert.Nil(t, err) {
// Need to remove trailing byte 0xa from bytes array to make sure bytes array are equal.
if out[len(payload)] == 0xa {
out[len(payload)] = 0x0
}
assert.Equal(t, payload, bytes.Trim(out, "\x00"))
} else {
t.Error("could not read from connection")
}
}