aboutsummaryrefslogtreecommitdiffstats
path: root/worker/lib/parse_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'worker/lib/parse_test.go')
-rw-r--r--worker/lib/parse_test.go18
1 files changed, 6 insertions, 12 deletions
diff --git a/worker/lib/parse_test.go b/worker/lib/parse_test.go
index 5c0a1b17..12190981 100644
--- a/worker/lib/parse_test.go
+++ b/worker/lib/parse_test.go
@@ -1,9 +1,9 @@
package lib
import (
- "bytes"
"io"
"io/ioutil"
+ "os"
"path/filepath"
"testing"
@@ -36,23 +36,17 @@ func TestMessageInfoHandledError(t *testing.T) {
}
type mockRawMessage struct {
- body []byte
+ path string
}
-func newMockRawMessage(body []byte) *mockRawMessage {
+func newMockRawMessageFromPath(p string) *mockRawMessage {
return &mockRawMessage{
- body: body,
+ path: p,
}
}
-func newMockRawMessageFromPath(p string) *mockRawMessage {
- b, err := ioutil.ReadFile(p)
- die(err)
- return newMockRawMessage(b)
-}
-
-func (m *mockRawMessage) NewReader() (io.Reader, error) {
- return bytes.NewReader(m.body), nil
+func (m *mockRawMessage) NewReader() (io.ReadCloser, error) {
+ return os.Open(m.path)
}
func (m *mockRawMessage) ModelFlags() ([]models.Flag, error) { return nil, nil }
func (m *mockRawMessage) Labels() ([]string, error) { return nil, nil }