135 lines
2.7 KiB
Go
135 lines
2.7 KiB
Go
package lease
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/tblyler/sheepmq/shepard"
|
|
)
|
|
|
|
func TestManagerAddCheckLease(t *testing.T) {
|
|
manager := NewManager()
|
|
|
|
info := &shepard.GetInfo{}
|
|
|
|
err := manager.AddLease(123, info)
|
|
if err != ErrNoLeaser {
|
|
t.Error("Failed to get ErrNoLeaser on add lease got", err)
|
|
}
|
|
|
|
info.PidLease = &shepard.PidLease{
|
|
// use a bad PID
|
|
Pid: 0,
|
|
}
|
|
|
|
err = manager.AddLease(123, info)
|
|
if err != nil {
|
|
t.Error("Failed to add a crappy PID leaser", err)
|
|
}
|
|
|
|
err = manager.AddLease(123, info)
|
|
if err != nil {
|
|
t.Error("Failed to add a crappy PID leaser ontop of another", err)
|
|
}
|
|
|
|
info.PidLease = nil
|
|
|
|
info.HeartLease = &shepard.HeartbeatLease{
|
|
Ttl: int64(time.Second / 2),
|
|
}
|
|
|
|
err = manager.AddLease(2, info)
|
|
if err != nil {
|
|
t.Error("Failed to add a valid heartbeat lease", err)
|
|
}
|
|
|
|
if !manager.CheckLease(2) {
|
|
t.Error("Failed to check for valid heartbeat lease")
|
|
}
|
|
|
|
info.HeartLease = nil
|
|
info.TimeoutLease = &shepard.TimeLease{
|
|
Ttl: int64(time.Second),
|
|
}
|
|
|
|
err = manager.AddLease(123, info)
|
|
if err != nil {
|
|
t.Error("Failed to add a good timeout lease of 1 second")
|
|
}
|
|
|
|
err = manager.AddLease(123, info)
|
|
if err == nil {
|
|
t.Error("Should not be able to add a lease ontop of another valid one")
|
|
}
|
|
|
|
err = manager.AddLease(124, info)
|
|
if err != nil {
|
|
t.Error("Failed to add a valid lease against a different id", err)
|
|
}
|
|
|
|
if !manager.CheckLease(123) {
|
|
t.Error("Failed to make sure valid lease was valid")
|
|
}
|
|
|
|
if !manager.CheckLease(124) {
|
|
t.Error("Failed to make sure valid lease was valid")
|
|
}
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
if manager.CheckLease(123) {
|
|
t.Error("failed to make sure invalid lease was invalid")
|
|
}
|
|
|
|
if manager.CheckLease(124) {
|
|
t.Error("failed to make sure invalid lease was invalid")
|
|
}
|
|
}
|
|
|
|
func TestManagerPruneLeases(t *testing.T) {
|
|
manager := NewManager()
|
|
|
|
info := &shepard.GetInfo{}
|
|
info.PidLease = &shepard.PidLease{
|
|
Pid: 0,
|
|
}
|
|
|
|
err := manager.AddLease(5, info)
|
|
if err != nil {
|
|
t.Error("Failed to add crappy PID lease", err)
|
|
}
|
|
|
|
info.PidLease = nil
|
|
info.TimeoutLease = &shepard.TimeLease{
|
|
Ttl: int64(time.Second),
|
|
}
|
|
|
|
err = manager.AddLease(2, info)
|
|
if err != nil {
|
|
t.Error("Failed to add valid timeout lease", err)
|
|
}
|
|
|
|
info.TimeoutLease = nil
|
|
info.HeartLease = &shepard.HeartbeatLease{
|
|
Ttl: int64(time.Second),
|
|
}
|
|
|
|
err = manager.AddLease(1, info)
|
|
if err != nil {
|
|
t.Error("Failed to add valid heartbeat lease", err)
|
|
}
|
|
|
|
manager.PruneLeases()
|
|
|
|
if len(manager.leases) != 2 {
|
|
t.Errorf("Should have 2 leases left, have %d", len(manager.leases))
|
|
}
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
manager.PruneLeases()
|
|
if len(manager.leases) != 0 {
|
|
t.Errorf("Should have 0 leases left, have %d", len(manager.leases))
|
|
}
|
|
}
|