Robot
/
croc
mirror of https://github.com/schollz/croc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test sending 40mb

This commit is contained in:
Zack Scholl 2019-04-29 16:04:47 -06:00
parent 5b0c52a97b
commit 2414593c91
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/comm/comm_test.go
View File

@ -1,6 +1,7 @@
package comm package comm
import ( import (
"crypto/rand"
"net" "net"
"testing" "testing"
"time" "time"
@ -11,6 +12,8 @@ import (
func TestComm(t *testing.T) { func TestComm(t *testing.T) {
defer log.Flush() defer log.Flush()
token := make([]byte, 40000000)
rand.Read(token)
port := "8001" port := "8001"
go func() { go func() {
@ -37,6 +40,9 @@ func TestComm(t *testing.T) {
data, err = c.Receive() data, err = c.Receive()
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, []byte{'\x00'}, data) assert.Equal(t, []byte{'\x00'}, data)
data, err = c.Receive()
assert.Nil(t, err)
assert.Equal(t, token, data)
}(port, connection) }(port, connection)
} }
}() }()
@ -49,4 +55,7 @@ func TestComm(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
assert.Nil(t, a.Send([]byte("hello, computer"))) assert.Nil(t, a.Send([]byte("hello, computer")))
assert.Nil(t, a.Send([]byte{'\x00'})) assert.Nil(t, a.Send([]byte{'\x00'}))
assert.Nil(t, a.Send(token))
} }