nhyatt/go-onerng
1
0
Fork 0
mirror of https://github.com/hairyhenderson/go-onerng.git synced 2025-07-05 01:52:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fixing some linting errors and refactoring

Browse Source 
Signed-off-by: Dave Henderson <dhenderson@gmail.com>
This commit is contained in:
Dave Henderson
2020-11-02 22:22:19 -05:00
parent 1e19f9f3ec
commit fc519b98e2
18 changed files with 587 additions and 532 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
onerng_test.go
View File

@ -8,31 +8,25 @@ import (
"github.com/stretchr/testify/assert"
)
// func TestNoiseCommand(t *testing.T) {
// fmt.Println(DisableWhitener)
// fmt.Println(EnableRF)
// fmt.Println(DisableAvalanche)
// fmt.Println(Default)
// fmt.Println(Silent)
// testdata := []struct {
// flags NoiseMode
// cmd string
// }{
// {Default, "cmd0\n"},
// {DisableWhitener, "cmd1\n"},
// {EnableRF, "cmd2\n"},
// {EnableRF | DisableWhitener, "cmd3\n"},
// {DisableAvalanche, "cmd4\n"},
// {DisableAvalanche | DisableWhitener, "cmd5\n"},
// {DisableAvalanche | EnableRF, "cmd6\n"},
// {DisableAvalanche | EnableRF | DisableWhitener, "cmd7\n"},
// }
// for _, d := range testdata {
// assert.Equal(t, d.cmd, noiseCommand(d.flags), d.cmd, d.flags)
// }
// }
func TestNoiseCommand(t *testing.T) {
testdata := []struct {
flags NoiseMode
cmd string
}{
{Default, "cmd0\n"},
{DisableWhitener, "cmd1\n"},
{EnableRF, "cmd2\n"},
{EnableRF | DisableWhitener, "cmd3\n"},
{Silent, "cmd4\n"},
{DisableAvalanche, "cmd4\n"},
{DisableAvalanche | DisableWhitener, "cmd5\n"},
{DisableAvalanche | EnableRF, "cmd6\n"},
{DisableAvalanche | EnableRF | DisableWhitener, "cmd7\n"},
}
for _, d := range testdata {
assert.Equal(t, d.cmd, noiseCommand(d.flags), d.cmd, d.flags)
}
}
type fakeDev struct {
closed bool
@ -48,6 +42,7 @@ func (d *fakeDev) reset() {
func (d *fakeDev) Close() error {
d.closed = true
return nil
}
@ -101,3 +96,19 @@ func TestVersion(t *testing.T) {
assert.Equal(t, "cmdo\ncmd4\ncmdv\ncmdO\ncmdo\n", d.wbuf.String())
assert.Equal(t, 3, v)
}
func TestIdentify(t *testing.T) {
d := &fakeDev{
wbuf: &bytes.Buffer{},
rbuf: bytes.NewBufferString("dfoawiuhf98h9inf2oifoi2jr\n" +
"dfkjawflihjwfoiuh2rliu13he487631487645t98y23rtoqu3rbno9q34htgfv\n" +
"\r\nVersion 3\r\nas;dlfjaw;oihf2ih2o3iuf2ofnlo2jnlfuhf2iou\n\n" +
"dlfkhadslfihwaflkhjw\n___lskdjfalsdkjflsd___\n\n"),
}
o := &OneRNG{Path: "/dev/null", device: d}
ctx := context.Background()
id, err := o.Identify(ctx)
assert.NoError(t, err)
assert.Equal(t, "cmd4\ncmdI\ncmdO\ncmdo\n", d.wbuf.String())
assert.Equal(t, "___lskdjfalsdkjflsd___", id)
}