mirror of
https://github.com/bettercap/bettercap
synced 2025-07-06 04:52:10 -07:00
add some queue tests
This commit is contained in:
parent
8b74453842
commit
d6564a5b4d
1 changed files with 65 additions and 0 deletions
65
packets/queue_test.go
Normal file
65
packets/queue_test.go
Normal file
|
@ -0,0 +1,65 @@
|
|||
package packets
|
||||
|
||||
import (
|
||||
"net"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestQueueActivity(t *testing.T) {
|
||||
i := net.IP{}
|
||||
h := net.HardwareAddr{}
|
||||
a := Activity{
|
||||
IP: i,
|
||||
MAC: h,
|
||||
}
|
||||
var units = []struct {
|
||||
got interface{}
|
||||
exp interface{}
|
||||
}{
|
||||
{a.IP, i},
|
||||
{a.MAC, h},
|
||||
{a.Source, false},
|
||||
}
|
||||
for _, u := range units {
|
||||
if !reflect.DeepEqual(u.exp, u.got) {
|
||||
t.Fatalf("expected '%v', got '%v'", u.exp, u.got)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueTraffic(t *testing.T) {
|
||||
tr := Traffic{}
|
||||
var units = []struct {
|
||||
got interface{}
|
||||
exp interface{}
|
||||
}{
|
||||
{tr.Sent, uint64(0)},
|
||||
{tr.Received, uint64(0)},
|
||||
}
|
||||
for _, u := range units {
|
||||
if !reflect.DeepEqual(u.exp, u.got) {
|
||||
t.Fatalf("expected '%v', got '%v'", u.exp, u.got)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestQueueStats(t *testing.T) {
|
||||
s := Stats{}
|
||||
var units = []struct {
|
||||
got interface{}
|
||||
exp interface{}
|
||||
}{
|
||||
{s.Sent, uint64(0)},
|
||||
{s.Received, uint64(0)},
|
||||
{s.PktReceived, uint64(0)},
|
||||
{s.Errors, uint64(0)},
|
||||
}
|
||||
for _, u := range units {
|
||||
if !reflect.DeepEqual(u.exp, u.got) {
|
||||
t.Fatalf("expected '%v', got '%v'", u.exp, u.got)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: add tests for the rest of queue.go
|
Loading…
Add table
Add a link
Reference in a new issue