mirror of
https://github.com/netbirdio/netbird.git
synced 2025-08-19 11:20:18 +02:00
abd152ee5a
* Separate shared code dependencies * Fix import * Test respective shared code * Update openapi ref * Fix test * Fix test path
50 lines
773 B
Go
50 lines
773 B
Go
package store
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/netbirdio/netbird/shared/relay/messages"
|
|
)
|
|
|
|
type MocPeer struct {
|
|
id messages.PeerID
|
|
}
|
|
|
|
func (m *MocPeer) Close() {
|
|
|
|
}
|
|
|
|
func (m *MocPeer) ID() messages.PeerID {
|
|
return m.id
|
|
}
|
|
|
|
func TestStore_DeletePeer(t *testing.T) {
|
|
s := NewStore()
|
|
|
|
pID := messages.HashID("peer_one")
|
|
p := &MocPeer{id: pID}
|
|
s.AddPeer(p)
|
|
s.DeletePeer(p)
|
|
if _, ok := s.Peer(pID); ok {
|
|
t.Errorf("peer was not deleted")
|
|
}
|
|
}
|
|
|
|
func TestStore_DeleteDeprecatedPeer(t *testing.T) {
|
|
s := NewStore()
|
|
|
|
pID1 := messages.HashID("peer_one")
|
|
pID2 := messages.HashID("peer_one")
|
|
|
|
p1 := &MocPeer{id: pID1}
|
|
p2 := &MocPeer{id: pID2}
|
|
|
|
s.AddPeer(p1)
|
|
s.AddPeer(p2)
|
|
s.DeletePeer(p1)
|
|
|
|
if _, ok := s.Peer(pID2); !ok {
|
|
t.Errorf("second peer was deleted")
|
|
}
|
|
}
|