mirrors_public/schollz_croc
0
0
Fork 0
mirror of https://github.com/schollz/croc.git synced 2025-10-11 13:21:00 +02:00
Code Activity

fix tests

Browse source
This commit is contained in:
Zack Scholl 2019-11-18 13:25:51 -08:00
parent 6b26af4dca
commit af95cddd8d
2 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/message/message_test.go
View file

@ -15,9 +15,9 @@ import (
func TestMessage(t *testing.T) { func TestMessage(t *testing.T) {
m := Message{Type: "message", Message: "hello, world"} m := Message{Type: "message", Message: "hello, world"}
e, err := crypt.New(nil, nil) e, salt, err := crypt.New([]byte("pass"), nil)
assert.Nil(t, err) assert.Nil(t, err)
fmt.Println(e.Salt()) fmt.Println(salt)
b, err := Encode(e, m) b, err := Encode(e, m)
assert.Nil(t, err) assert.Nil(t, err)
fmt.Printf("%x\n", b) fmt.Printf("%x\n", b)
@ -68,7 +68,8 @@ func TestSend(t *testing.T) {
a, err := comm.NewConnection("localhost:"+port, 10*time.Minute) a, err := comm.NewConnection("localhost:"+port, 10*time.Minute)
assert.Nil(t, err) assert.Nil(t, err)
m := Message{Type: "message", Message: "hello, world"} m := Message{Type: "message", Message: "hello, world"}
e, err := crypt.New(nil, nil) e, salt, err := crypt.New([]byte("pass"), nil)
log.Debug(salt)
assert.Nil(t, err) assert.Nil(t, err)
assert.Nil(t, Send(a, e, m)) assert.Nil(t, Send(a, e, m))

16
src/tcp/tcp_test.go
View file

@ -1,6 +1,7 @@
package tcp package tcp
import ( import (
"bytes"
"testing" "testing"
"time" "time"
@ -23,12 +24,23 @@ func TestTCP(t *testing.T) {
// try sending data // try sending data
assert.Nil(t, c1.Send([]byte("hello, c2"))) assert.Nil(t, c1.Send([]byte("hello, c2")))
data, err := c2.Receive() var data []byte
for {
data, err = c2.Receive()
if bytes.Equal(data, []byte{1}) {
continue
}
}
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, []byte("hello, c2"), data) assert.Equal(t, []byte("hello, c2"), data)
assert.Nil(t, c2.Send([]byte("hello, c1"))) assert.Nil(t, c2.Send([]byte("hello, c1")))
data, err = c1.Receive() for {
data, err = c1.Receive()
if bytes.Equal(data, []byte{1}) {
continue
}
}
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, []byte("hello, c1"), data) assert.Equal(t, []byte("hello, c1"), data)