Use pool for frames

This commit is contained in:
9seconds
2018-06-20 09:19:27 +03:00
parent 9ddf3e147c
commit c76d8307c2
8 changed files with 66 additions and 30 deletions
+5 -4
View File
@@ -1,6 +1,7 @@
package obfuscated2
import (
"bytes"
"testing"
"github.com/stretchr/testify/assert"
@@ -47,21 +48,21 @@ func TestFrameValid(t *testing.T) {
func TestFrameDoubleInvert(t *testing.T) {
frame := makeFrame()
assert.Equal(t, frame, frame.Invert().Invert())
assert.True(t, bytes.Equal(frame, *frame.Invert().Invert()))
}
func TestFrameInvert(t *testing.T) {
frame := makeFrame()
reversed := frame.Invert()
assert.Exactly(t, frame[:8], reversed[:8])
assert.Exactly(t, frame[56:], reversed[56:])
assert.Exactly(t, frame[:8], (*reversed)[:8])
assert.Exactly(t, frame[56:], (*reversed)[56:])
toCompare := make([]byte, 48)
for i := 0; i < 48; i++ {
toCompare[i] = frame[55-i]
}
assert.Equal(t, []byte(reversed[8:56]), toCompare)
assert.Equal(t, []byte((*reversed)[8:56]), toCompare)
}
func TestFrameGenerateValid(t *testing.T) {