remove wrapToParse dependence

master
bel 2023-03-25 22:54:23 -06:00
parent 373d8be1a0
commit 802266e500
1 changed files with 14 additions and 7 deletions

View File

@ -2,7 +2,8 @@ package wrap
import ( import (
"context" "context"
"mayhem-party/src/device/input/parse" "mayhem-party/src/device/input/button"
"mayhem-party/src/device/input/raw"
"os" "os"
"syscall" "syscall"
"testing" "testing"
@ -50,7 +51,7 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
newParsers := 0 newParsers := 0
newParser := func() Wrap { newParser := func() Wrap {
newParsers += 1 newParsers += 1
return parse.NewPlaintext(src) return src
} }
ctx, can := context.WithCancel(context.Background()) ctx, can := context.WithCancel(context.Background())
defer can() defer can()
@ -72,7 +73,7 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
ChSigUsr1 <- syscall.SIGINT ChSigUsr1 <- syscall.SIGINT
} }
time.Sleep(time.Millisecond * 250) time.Sleep(time.Millisecond * 250)
if want := (telemetrySrc{reads: 5}); *src != want { if want := (telemetrySrc{reads: 5, closeWraps: 5}); *src != want {
t.Errorf("want %+v, got %+v", want, *src) t.Errorf("want %+v, got %+v", want, *src)
} else if newParsers != 6 { } else if newParsers != 6 {
t.Error(newParsers) t.Error(newParsers)
@ -80,15 +81,21 @@ func TestRefreshDoesntCloseSources(t *testing.T) {
} }
type telemetrySrc struct { type telemetrySrc struct {
closeWraps int
closes int closes int
reads int reads int
} }
func (src *telemetrySrc) CloseWrap() raw.Raw {
src.closeWraps += 1
return nil
}
func (src *telemetrySrc) Close() { func (src *telemetrySrc) Close() {
src.closes += 1 src.closes += 1
} }
func (src *telemetrySrc) Read() []byte { func (src *telemetrySrc) Read() []button.Button {
src.reads += 1 src.reads += 1
return []byte("foo") return nil
} }