diff --git a/testing/BUILD.bazel b/testing/BUILD.bazel index 3071b82f935eeb1d0f3f722214b282c98763245d..02411722b3d58ed2be6fcd2e4c950da21807f8cf 100644 --- a/testing/BUILD.bazel +++ b/testing/BUILD.bazel @@ -1,4 +1,4 @@ -load("@io_bazel_rules_go//go:def.bzl", "go_library") +load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test") go_library( name = "go_default_library", @@ -6,3 +6,10 @@ go_library( importpath = "github.com/bio-routing/bio-rd/testing", visibility = ["//visibility:public"], ) + +go_test( + name = "go_default_test", + srcs = ["conn_mock_test.go"], + embed = [":go_default_library"], + deps = ["//vendor/github.com/stretchr/testify/assert:go_default_library"], +) diff --git a/testing/conn_mock_test.go b/testing/conn_mock_test.go new file mode 100644 index 0000000000000000000000000000000000000000..9d964ffa9fecef3f577dd02c498ed9dc110bcf1b --- /dev/null +++ b/testing/conn_mock_test.go @@ -0,0 +1,28 @@ +package testing + +import ( + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestWrite(t *testing.T) { + m := &MockConn{} + + payload := []byte{1, 2, 3} + m.Write(payload) + + assert.Equal(t, payload, m.Bytes) +} + +func TestRead(t *testing.T) { + m := &MockConn{} + + payload := []byte{1, 2, 3} + m.Bytes = payload + + buffer := make([]byte, 4) + n, _ := m.Read(buffer) + + assert.Equal(t, payload, buffer[:n]) +}