2022-03-15 17:57:46 +00:00
|
|
|
package aghnet
|
|
|
|
|
|
|
|
import (
|
|
|
|
"net"
|
2022-10-25 13:08:12 +01:00
|
|
|
"net/netip"
|
2022-03-15 17:57:46 +00:00
|
|
|
"sync"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/AdguardTeam/golibs/errors"
|
|
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
2022-03-23 17:47:45 +00:00
|
|
|
func TestNewARPDB(t *testing.T) {
|
|
|
|
var a ARPDB
|
2022-03-30 13:11:57 +01:00
|
|
|
require.NotPanics(t, func() { a = NewARPDB() })
|
2022-03-23 17:47:45 +00:00
|
|
|
|
|
|
|
assert.NotNil(t, a)
|
|
|
|
}
|
|
|
|
|
2022-03-15 17:57:46 +00:00
|
|
|
// TestARPDB is the mock implementation of ARPDB to use in tests.
|
|
|
|
type TestARPDB struct {
|
|
|
|
OnRefresh func() (err error)
|
|
|
|
OnNeighbors func() (ns []Neighbor)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Refresh implements the ARPDB interface for *TestARPDB.
|
|
|
|
func (arp *TestARPDB) Refresh() (err error) {
|
|
|
|
return arp.OnRefresh()
|
|
|
|
}
|
|
|
|
|
|
|
|
// Neighbors implements the ARPDB interface for *TestARPDB.
|
|
|
|
func (arp *TestARPDB) Neighbors() (ns []Neighbor) {
|
|
|
|
return arp.OnNeighbors()
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestARPDBS(t *testing.T) {
|
2022-10-25 13:08:12 +01:00
|
|
|
knownIP := netip.MustParseAddr("1.2.3.4")
|
2022-03-15 17:57:46 +00:00
|
|
|
knownMAC := net.HardwareAddr{0xAB, 0xCD, 0xEF, 0xAB, 0xCD, 0xEF}
|
|
|
|
|
|
|
|
succRefrCount, failRefrCount := 0, 0
|
|
|
|
clnp := func() {
|
|
|
|
succRefrCount, failRefrCount = 0, 0
|
|
|
|
}
|
|
|
|
|
|
|
|
succDB := &TestARPDB{
|
|
|
|
OnRefresh: func() (err error) { succRefrCount++; return nil },
|
|
|
|
OnNeighbors: func() (ns []Neighbor) {
|
|
|
|
return []Neighbor{{Name: "abc", IP: knownIP, MAC: knownMAC}}
|
|
|
|
},
|
|
|
|
}
|
|
|
|
failDB := &TestARPDB{
|
|
|
|
OnRefresh: func() (err error) { failRefrCount++; return errors.Error("refresh failed") },
|
|
|
|
OnNeighbors: func() (ns []Neighbor) { return nil },
|
|
|
|
}
|
|
|
|
|
|
|
|
t.Run("begin_with_success", func(t *testing.T) {
|
|
|
|
t.Cleanup(clnp)
|
|
|
|
|
|
|
|
a := newARPDBs(succDB, failDB)
|
|
|
|
err := a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Equal(t, 1, succRefrCount)
|
|
|
|
assert.Zero(t, failRefrCount)
|
|
|
|
assert.NotEmpty(t, a.Neighbors())
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("begin_with_fail", func(t *testing.T) {
|
|
|
|
t.Cleanup(clnp)
|
|
|
|
|
|
|
|
a := newARPDBs(failDB, succDB)
|
|
|
|
err := a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Equal(t, 1, succRefrCount)
|
|
|
|
assert.Equal(t, 1, failRefrCount)
|
|
|
|
assert.NotEmpty(t, a.Neighbors())
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("fail_only", func(t *testing.T) {
|
|
|
|
t.Cleanup(clnp)
|
|
|
|
|
|
|
|
wantMsg := `each arpdb failed: 2 errors: "refresh failed", "refresh failed"`
|
|
|
|
|
|
|
|
a := newARPDBs(failDB, failDB)
|
|
|
|
err := a.Refresh()
|
|
|
|
require.Error(t, err)
|
|
|
|
|
|
|
|
testutil.AssertErrorMsg(t, wantMsg, err)
|
|
|
|
|
|
|
|
assert.Equal(t, 2, failRefrCount)
|
|
|
|
assert.Empty(t, a.Neighbors())
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("fail_after_success", func(t *testing.T) {
|
|
|
|
t.Cleanup(clnp)
|
|
|
|
|
|
|
|
shouldFail := false
|
|
|
|
unstableDB := &TestARPDB{
|
|
|
|
OnRefresh: func() (err error) {
|
|
|
|
if shouldFail {
|
|
|
|
err = errors.Error("unstable failed")
|
|
|
|
}
|
|
|
|
shouldFail = !shouldFail
|
|
|
|
|
|
|
|
return err
|
|
|
|
},
|
|
|
|
OnNeighbors: func() (ns []Neighbor) {
|
|
|
|
if !shouldFail {
|
|
|
|
return failDB.OnNeighbors()
|
|
|
|
}
|
|
|
|
|
|
|
|
return succDB.OnNeighbors()
|
|
|
|
},
|
|
|
|
}
|
|
|
|
a := newARPDBs(unstableDB, succDB)
|
|
|
|
|
|
|
|
// Unstable ARPDB should refresh successfully.
|
|
|
|
err := a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Zero(t, succRefrCount)
|
|
|
|
assert.NotEmpty(t, a.Neighbors())
|
|
|
|
|
|
|
|
// Unstable ARPDB should fail and the succDB should be used.
|
|
|
|
err = a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Equal(t, 1, succRefrCount)
|
|
|
|
assert.NotEmpty(t, a.Neighbors())
|
|
|
|
|
2022-03-30 13:11:57 +01:00
|
|
|
// Unstable ARPDB should refresh successfully again.
|
2022-03-15 17:57:46 +00:00
|
|
|
err = a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
2022-03-30 13:11:57 +01:00
|
|
|
assert.Equal(t, 1, succRefrCount)
|
2022-03-15 17:57:46 +00:00
|
|
|
assert.NotEmpty(t, a.Neighbors())
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("empty", func(t *testing.T) {
|
|
|
|
a := newARPDBs()
|
|
|
|
require.NoError(t, a.Refresh())
|
|
|
|
|
|
|
|
assert.Empty(t, a.Neighbors())
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestCmdARPDB_arpa(t *testing.T) {
|
|
|
|
a := &cmdARPDB{
|
2022-03-30 13:11:57 +01:00
|
|
|
cmd: "cmd",
|
2022-03-15 17:57:46 +00:00
|
|
|
parse: parseArpA,
|
|
|
|
ns: &neighs{
|
|
|
|
mu: &sync.RWMutex{},
|
|
|
|
ns: make([]Neighbor, 0),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
t.Run("arp_a", func(t *testing.T) {
|
2022-03-30 13:11:57 +01:00
|
|
|
sh := theOnlyCmd("cmd", 0, arpAOutput, nil)
|
|
|
|
substShell(t, sh.RunCmd)
|
2022-03-15 17:57:46 +00:00
|
|
|
|
|
|
|
err := a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Equal(t, wantNeighs, a.Neighbors())
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("runcmd_error", func(t *testing.T) {
|
2022-03-30 13:11:57 +01:00
|
|
|
sh := theOnlyCmd("cmd", 0, "", errors.Error("can't run"))
|
|
|
|
substShell(t, sh.RunCmd)
|
2022-03-15 17:57:46 +00:00
|
|
|
|
|
|
|
err := a.Refresh()
|
|
|
|
testutil.AssertErrorMsg(t, "cmd arpdb: running command: can't run", err)
|
|
|
|
})
|
2022-03-23 17:47:45 +00:00
|
|
|
|
2022-03-30 13:11:57 +01:00
|
|
|
t.Run("bad_code", func(t *testing.T) {
|
|
|
|
sh := theOnlyCmd("cmd", 1, "", nil)
|
|
|
|
substShell(t, sh.RunCmd)
|
2022-03-23 17:47:45 +00:00
|
|
|
|
2022-03-30 13:11:57 +01:00
|
|
|
err := a.Refresh()
|
|
|
|
testutil.AssertErrorMsg(t, "cmd arpdb: running command: unexpected exit code 1", err)
|
2022-03-23 17:47:45 +00:00
|
|
|
})
|
2022-03-31 17:56:50 +01:00
|
|
|
|
|
|
|
t.Run("empty", func(t *testing.T) {
|
|
|
|
sh := theOnlyCmd("cmd", 0, "", nil)
|
|
|
|
substShell(t, sh.RunCmd)
|
|
|
|
|
|
|
|
err := a.Refresh()
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
assert.Empty(t, a.Neighbors())
|
|
|
|
})
|
2022-03-23 17:47:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestEmptyARPDB(t *testing.T) {
|
|
|
|
a := EmptyARPDB{}
|
|
|
|
|
|
|
|
t.Run("refresh", func(t *testing.T) {
|
|
|
|
var err error
|
|
|
|
require.NotPanics(t, func() {
|
|
|
|
err = a.Refresh()
|
|
|
|
})
|
|
|
|
|
|
|
|
assert.NoError(t, err)
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("neighbors", func(t *testing.T) {
|
|
|
|
var ns []Neighbor
|
|
|
|
require.NotPanics(t, func() {
|
|
|
|
ns = a.Neighbors()
|
|
|
|
})
|
|
|
|
|
|
|
|
assert.Empty(t, ns)
|
|
|
|
})
|
|
|
|
}
|