mirror of
https://github.com/slackhq/nebula.git
synced 2026-02-14 08:44:24 +01:00
Rework some things into packages (#489)
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/slackhq/nebula/firewall"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
@@ -50,7 +51,7 @@ func TestTimerWheel_findWheel(t *testing.T) {
|
||||
func TestTimerWheel_Add(t *testing.T) {
|
||||
tw := NewTimerWheel(time.Second, time.Second*10)
|
||||
|
||||
fp1 := FirewallPacket{}
|
||||
fp1 := firewall.Packet{}
|
||||
tw.Add(fp1, time.Second*1)
|
||||
|
||||
// Make sure we set head and tail properly
|
||||
@@ -61,7 +62,7 @@ func TestTimerWheel_Add(t *testing.T) {
|
||||
assert.Nil(t, tw.wheel[2].Tail.Next)
|
||||
|
||||
// Make sure we only modify head
|
||||
fp2 := FirewallPacket{}
|
||||
fp2 := firewall.Packet{}
|
||||
tw.Add(fp2, time.Second*1)
|
||||
assert.Equal(t, fp2, tw.wheel[2].Head.Packet)
|
||||
assert.Equal(t, fp1, tw.wheel[2].Head.Next.Packet)
|
||||
@@ -84,7 +85,7 @@ func TestTimerWheel_Purge(t *testing.T) {
|
||||
assert.NotNil(t, tw.lastTick)
|
||||
assert.Equal(t, 0, tw.current)
|
||||
|
||||
fps := []FirewallPacket{
|
||||
fps := []firewall.Packet{
|
||||
{LocalIP: 1},
|
||||
{LocalIP: 2},
|
||||
{LocalIP: 3},
|
||||
|
||||
Reference in New Issue
Block a user