mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2025-10-08 17:50:15 +08:00
36 lines
657 B
Go
36 lines
657 B
Go
package hds
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEncryption(t *testing.T) {
|
|
key := []byte(core.RandString(16, 0))
|
|
salt := core.RandString(32, 0)
|
|
|
|
c, err := Client(nil, key, salt, true)
|
|
require.NoError(t, err)
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
c.wr = bufio.NewWriter(buf)
|
|
|
|
n, err := c.Write([]byte("test"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, 4, n)
|
|
|
|
c, err = Client(nil, key, salt, false)
|
|
c.rd = bufio.NewReader(buf)
|
|
require.NoError(t, err)
|
|
|
|
b := make([]byte, 32)
|
|
n, err = c.Read(b)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "test", string(b[:n]))
|
|
}
|