summaryrefslogtreecommitdiffhomepage
path: root/tun
diff options
context:
space:
mode:
authorMatt Layher <mdlayher@gmail.com>2019-05-29 12:18:20 -0400
committerJason A. Donenfeld <Jason@zx2c4.com>2019-05-29 18:34:55 +0200
commit32912dc778a0b98637fbfe010b31bd0b746662e6 (patch)
tree7d9dfa99bcb6e0479f6b4e0358f5f6daa213929d /tun
parentd4034e5f8a298c263c820d13a7758436f5e1b89b (diff)
device, tun: rearrange code and fix device tests
Signed-off-by: Matt Layher <mdlayher@gmail.com>
Diffstat (limited to 'tun')
-rw-r--r--tun/helper_test.go93
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
-}