forked from perlin-network/noise
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
msg, init, empty, opcode, peer: split up msg.go into separate files examples/e2e: increase num messages for each node
- Loading branch information
1 parent
8286d6b
commit 34f73a4
Showing
9 changed files
with
234 additions
and
219 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
package noise | ||
|
||
import "github.com/perlin-network/noise/payload" | ||
|
||
var _ Message = (*EmptyMessage)(nil) | ||
|
||
type EmptyMessage struct{} | ||
|
||
func (EmptyMessage) Read(reader payload.Reader) (Message, error) { | ||
return EmptyMessage{}, nil | ||
} | ||
|
||
func (EmptyMessage) Write() []byte { | ||
return nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -3,5 +3,7 @@ package noise | |
import "go.dedis.ch/kyber/v3/suites" | ||
|
||
func init() { | ||
resetOpcodes() | ||
|
||
suites.RequireConstantTime() | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,77 +1 @@ | ||
package noise | ||
|
||
import ( | ||
"github.com/perlin-network/noise/payload" | ||
"github.com/pkg/errors" | ||
"github.com/stretchr/testify/assert" | ||
"testing" | ||
) | ||
|
||
type testMsg struct { | ||
text string | ||
} | ||
|
||
func (testMsg) Read(reader payload.Reader) (Message, error) { | ||
text, err := reader.ReadString() | ||
if err != nil { | ||
return nil, errors.Wrap(err, "failed to read chat msg") | ||
} | ||
|
||
return testMsg{text: text}, nil | ||
} | ||
|
||
func (m testMsg) Write() []byte { | ||
return payload.NewWriter(nil).WriteString(m.text).Bytes() | ||
} | ||
|
||
func TestNextAvailableOpcode(t *testing.T) { | ||
resetOpcodes() | ||
|
||
// opcode 0 should be an empty message | ||
msg, err := MessageFromOpcode(Opcode(0)) | ||
assert.Nil(t, err) | ||
assert.Equal(t, EmptyMessage{}, msg) | ||
|
||
// an unset opcode should be an error | ||
_, err = MessageFromOpcode(Opcode(1)) | ||
assert.NotNil(t, err) | ||
|
||
RegisterMessage(Opcode(1), (*testMsg)(nil)) | ||
|
||
// an opcode should still exist after the loop | ||
msg, err = MessageFromOpcode(Opcode(1)) | ||
assert.Nil(t, err) | ||
assert.Equal(t, testMsg{}, msg) | ||
} | ||
|
||
func TestEncodeMessage(t *testing.T) { | ||
resetOpcodes() | ||
o := RegisterMessage(Opcode(123), (*testMsg)(nil)) | ||
|
||
msg := testMsg{ | ||
text: "hello", | ||
} | ||
|
||
p := newPeer(nil, nil) | ||
|
||
bytes, err := p.EncodeMessage(msg) | ||
assert.Nil(t, err) | ||
assert.Equal(t, append([]byte{byte(o)}, msg.Write()...), bytes) | ||
} | ||
|
||
func TestDecodeMessage(t *testing.T) { | ||
resetOpcodes() | ||
o := RegisterMessage(Opcode(45), (*testMsg)(nil)) | ||
|
||
msg := testMsg{ | ||
text: "world", | ||
} | ||
assert.Equal(t, o, RegisterMessage(o, (*testMsg)(nil))) | ||
|
||
p := newPeer(nil, nil) | ||
|
||
resultO, resultM, err := p.DecodeMessage(append([]byte{byte(o)}, msg.Write()...)) | ||
assert.Nil(t, err) | ||
assert.Equal(t, o, resultO) | ||
assert.Equal(t, msg, resultM) | ||
} |
Oops, something went wrong.