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 /tun/helper_test.go | |
parent | d4034e5f8a298c263c820d13a7758436f5e1b89b (diff) |
device, tun: rearrange code and fix device tests
Signed-off-by: Matt Layher <mdlayher@gmail.com>
Diffstat (limited to 'tun/helper_test.go')
-rw-r--r-- | tun/helper_test.go | 93 |
1 files changed, 0 insertions, 93 deletions
diff --git a/tun/helper_test.go b/tun/helper_test.go deleted file mode 100644 index 4fa0357..0000000 --- a/tun/helper_test.go +++ /dev/null @@ -1,93 +0,0 @@ -/* SPDX-License-Identifier: MIT - * - * Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved. - */ - -package tun - -import ( - "bytes" - "errors" - "os" - "testing" - - "golang.zx2c4.com/wireguard/tun" -) - -/* Helpers for writing unit tests - */ - -type DummyTUN struct { - name string - mtu int - packets chan []byte - events chan tun.TUNEvent -} - -func (tun *DummyTUN) File() *os.File { - return nil -} - -func (tun *DummyTUN) Name() (string, error) { - return tun.name, nil -} - -func (tun *DummyTUN) MTU() (int, error) { - return tun.mtu, nil -} - -func (tun *DummyTUN) Write(d []byte, offset int) (int, error) { - tun.packets <- d[offset:] - return len(d), nil -} - -func (tun *DummyTUN) Close() error { - close(tun.events) - close(tun.packets) - return nil -} - -func (tun *DummyTUN) Events() chan tun.TUNEvent { - return tun.events -} - -func (tun *DummyTUN) Read(d []byte, offset int) (int, error) { - t, ok := <-tun.packets - if !ok { - return 0, errors.New("device closed") - } - copy(d[offset:], t) - return len(t), nil -} - -func CreateDummyTUN(name string) (tun.TUNDevice, error) { - var dummy DummyTUN - dummy.mtu = 0 - dummy.packets = make(chan []byte, 100) - dummy.events = make(chan tun.TUNEvent, 10) - return &dummy, nil -} - -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) - } -} - -func randDevice(t *testing.T) *Device { - sk, err := newPrivateKey() - if err != nil { - t.Fatal(err) - } - tun, _ := CreateDummyTUN("dummy") - logger := NewLogger(LogLevelError, "") - device := NewDevice(tun, logger) - device.SetPrivateKey(sk) - return device -} |