mirror of https://github.com/XTLS/Xray-core
51 lines
1008 B
Go
51 lines
1008 B
Go
package buf_test
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"io"
|
|
"testing"
|
|
|
|
. "github.com/xtls/xray-core/common/buf"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/testing/servers/tcp"
|
|
)
|
|
|
|
func TestWriterCreation(t *testing.T) {
|
|
tcpServer := tcp.Server{}
|
|
dest, err := tcpServer.Start()
|
|
if err != nil {
|
|
t.Fatal("failed to start tcp server: ", err)
|
|
}
|
|
defer tcpServer.Close()
|
|
|
|
conn, err := net.Dial("tcp", dest.NetAddr())
|
|
if err != nil {
|
|
t.Fatal("failed to dial a TCP connection: ", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
{
|
|
writer := NewWriter(conn)
|
|
if _, ok := writer.(*BufferToBytesWriter); !ok {
|
|
t.Fatal("writer is not a BufferToBytesWriter")
|
|
}
|
|
|
|
writer2 := NewWriter(writer.(io.Writer))
|
|
if writer2 != writer {
|
|
t.Fatal("writer is not reused")
|
|
}
|
|
}
|
|
|
|
tlsConn := tls.Client(conn, &tls.Config{
|
|
InsecureSkipVerify: true,
|
|
})
|
|
defer tlsConn.Close()
|
|
|
|
{
|
|
writer := NewWriter(tlsConn)
|
|
if _, ok := writer.(*SequentialWriter); !ok {
|
|
t.Fatal("writer is not a SequentialWriter")
|
|
}
|
|
}
|
|
}
|