diff options
author | Matt Layher <mdlayher@gmail.com> | 2019-05-29 12:18:20 -0400 |
---|---|---|
committer | Jason A. Donenfeld <Jason@zx2c4.com> | 2019-05-29 18:34:55 +0200 |
commit | 32912dc778a0b98637fbfe010b31bd0b746662e6 (patch) | |
tree | 7d9dfa99bcb6e0479f6b4e0358f5f6daa213929d /device | |
parent | d4034e5f8a298c263c820d13a7758436f5e1b89b (diff) |
device, tun: rearrange code and fix device tests
Signed-off-by: Matt Layher <mdlayher@gmail.com>
Diffstat (limited to 'device')
-rw-r--r-- | device/device_test.go | 40 | ||||
-rw-r--r-- | device/tun_test.go | 56 |
2 files changed, 86 insertions, 10 deletions
diff --git a/device/device_test.go b/device/device_test.go index db5a3c0..b6212b5 100644 --- a/device/device_test.go +++ b/device/device_test.go @@ -9,21 +9,17 @@ package device * without network dependencies */ -import "testing" +import ( + "bytes" + "testing" +) func TestDevice(t *testing.T) { // prepare tun devices for generating traffic - tun1, err := CreateDummyTUN("tun1") - if err != nil { - t.Error("failed to create tun:", err.Error()) - } - - tun2, err := CreateDummyTUN("tun2") - if err != nil { - t.Error("failed to create tun:", err.Error()) - } + tun1 := newDummyTUN("tun1") + tun2 := newDummyTUN("tun2") _ = tun1 _ = tun2 @@ -46,3 +42,27 @@ func TestDevice(t *testing.T) { // create binds } + +func randDevice(t *testing.T) *Device { + sk, err := newPrivateKey() + if err != nil { + t.Fatal(err) + } + tun := newDummyTUN("dummy") + logger := NewLogger(LogLevelError, "") + device := NewDevice(tun, logger) + device.SetPrivateKey(sk) + return device +} + +func assertNil(t *testing.T, err error) { + if err != nil { + t.Fatal(err) + } +} + +func assertEqual(t *testing.T, a []byte, b []byte) { + if bytes.Compare(a, b) != 0 { + t.Fatal(a, "!=", b) + } +} diff --git a/device/tun_test.go b/device/tun_test.go new file mode 100644 index 0000000..fbe4c1d --- /dev/null +++ b/device/tun_test.go @@ -0,0 +1,56 @@ +/* SPDX-License-Identifier: MIT + * + * Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved. + */ + +package device + +import ( + "errors" + "os" + + "golang.zx2c4.com/wireguard/tun" +) + +// newDummyTUN creates a dummy TUN device with the specified name. +func newDummyTUN(name string) tun.TUNDevice { + return &dummyTUN{ + name: name, + packets: make(chan []byte, 100), + events: make(chan tun.TUNEvent, 10), + } +} + +// A dummyTUN is a tun.TUNDevice which is used in unit tests. +type dummyTUN struct { + name string + mtu int + packets chan []byte + events chan tun.TUNEvent +} + +func (d *dummyTUN) Events() chan tun.TUNEvent { return d.events } +func (*dummyTUN) File() *os.File { return nil } +func (*dummyTUN) Flush() error { return nil } +func (d *dummyTUN) MTU() (int, error) { return d.mtu, nil } +func (d *dummyTUN) Name() (string, error) { return d.name, nil } + +func (d *dummyTUN) Close() error { + close(d.events) + close(d.packets) + return nil +} + +func (d *dummyTUN) Read(b []byte, offset int) (int, error) { + buf, ok := <-d.packets + if !ok { + return 0, errors.New("device closed") + } + copy(b[offset:], buf) + return len(buf), nil +} + +func (d *dummyTUN) Write(b []byte, offset int) (int, error) { + d.packets <- b[offset:] + return len(b), nil +} |