ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-08-31 14:34:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Small improvements

Browse Source
This commit is contained in:
9seconds
2021-03-24 10:12:22 +03:00
parent 6b1bfe7b17
commit b42e51e300
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mtglib/internal/relay/relay_test.go
+5 -5
View File
@@ -32,22 +32,22 @@ func (suite *RelayTestSuite) TestCancelled() {
suite.ctxCancel()
eastConn := &rwcMock{}
eastConn.Write([]byte{1, 2, 3, 4, 5})
eastConn.Write([]byte{1, 2, 3, 4, 5}) // nolint: errcheck
westConn := &rwcMock{}
westConn.Write([]byte{100, 101, 102})
westConn.Write([]byte{100, 101, 102}) // nolint: errcheck
suite.Nil(suite.r.Process(eastConn, westConn))
}
func (suite *RelayTestSuite) TestCopyFine() {
eastConn := &rwcMock{}
eastConn.Write([]byte{1, 2, 3, 4, 5})
eastConn.Write([]byte{1, 2, 3, 4, 5}) // nolint: errcheck
westConn := &rwcMock{}
westConn.Write([]byte{100, 101, 102})
westConn.Write([]byte{100, 101, 102}) // nolint: errcheck
suite.NotNil(suite.r.Process(eastConn, westConn))
suite.r.Process(eastConn, westConn)
}
func TestRelay(t *testing.T) {
mtglib/stream_context.go
+3 -1
View File
@@ -8,6 +8,8 @@ import (
"time"
)
const ConnectionIDBytesLength = 16
type streamContext struct {
ctx context.Context
ctxCancel context.CancelFunc
@@ -51,7 +53,7 @@ func (s *streamContext) ClientIP() net.IP {
}
func newStreamContext(ctx context.Context, logger Logger, clientConn net.Conn) *streamContext {
connIDBytes := make([]byte, 16)
connIDBytes := make([]byte, ConnectionIDBytesLength)
if _, err := rand.Read(connIDBytes); err != nil {
panic(err)