locks: fix testing stale locks created on other hosts

Closes #260
This commit is contained in:
Alexander Neumann 2015-08-16 16:24:04 +02:00
parent dc842ffa09
commit fbe746c261
2 changed files with 52 additions and 18 deletions

13
lock.go
View File

@ -206,6 +206,19 @@ func (l *Lock) Stale() bool {
return true return true
} }
hn, err := os.Hostname()
if err != nil {
debug.Log("Lock.Stale", "unable to find current hostnanme: %v", err)
// since we cannot find the current hostname, assume that the lock is
// not stale.
return false
}
if hn != l.Hostname {
// lock was created on a different host, assume the lock is not stale.
return false
}
proc, err := os.FindProcess(l.PID) proc, err := os.FindProcess(l.PID)
defer proc.Release() defer proc.Release()
if err != nil { if err != nil {

View File

@ -93,7 +93,12 @@ func TestExclusiveLockOnLockedRepo(t *testing.T) {
} }
func createFakeLock(repo *repository.Repository, t time.Time, pid int) (backend.ID, error) { func createFakeLock(repo *repository.Repository, t time.Time, pid int) (backend.ID, error) {
newLock := &restic.Lock{Time: t, PID: pid} hostname, err := os.Hostname()
if err != nil {
return backend.ID{}, err
}
newLock := &restic.Lock{Time: t, PID: pid, Hostname: hostname}
return repo.SaveJSONUnpacked(backend.Lock, &newLock) return repo.SaveJSONUnpacked(backend.Lock, &newLock)
} }
@ -102,42 +107,58 @@ func removeLock(repo *repository.Repository, id backend.ID) error {
} }
var staleLockTests = []struct { var staleLockTests = []struct {
timestamp time.Time timestamp time.Time
stale bool stale bool
pid int staleOnOtherHost bool
pid int
}{ }{
{ {
timestamp: time.Now(), timestamp: time.Now(),
stale: false, stale: false,
pid: os.Getpid(), staleOnOtherHost: false,
pid: os.Getpid(),
}, },
{ {
timestamp: time.Now().Add(-time.Hour), timestamp: time.Now().Add(-time.Hour),
stale: true, stale: true,
pid: os.Getpid(), staleOnOtherHost: true,
pid: os.Getpid(),
}, },
{ {
timestamp: time.Now().Add(3 * time.Minute), timestamp: time.Now().Add(3 * time.Minute),
stale: false, stale: false,
pid: os.Getpid(), staleOnOtherHost: false,
pid: os.Getpid(),
}, },
{ {
timestamp: time.Now(), timestamp: time.Now(),
stale: true, stale: true,
pid: os.Getpid() + 500, staleOnOtherHost: false,
pid: os.Getpid() + 500,
}, },
} }
func TestLockStale(t *testing.T) { func TestLockStale(t *testing.T) {
hostname, err := os.Hostname()
OK(t, err)
otherHostname := "other-" + hostname
for i, test := range staleLockTests { for i, test := range staleLockTests {
lock := restic.Lock{ lock := restic.Lock{
Time: test.timestamp, Time: test.timestamp,
PID: test.pid, PID: test.pid,
Hostname: hostname,
} }
Assert(t, lock.Stale() == test.stale, Assert(t, lock.Stale() == test.stale,
"TestStaleLock: test %d failed: expected stale: %v, got %v", "TestStaleLock: test %d failed: expected stale: %v, got %v",
i, test.stale, !test.stale) i, test.stale, !test.stale)
lock.Hostname = otherHostname
Assert(t, lock.Stale() == test.staleOnOtherHost,
"TestStaleLock: test %d failed: expected staleOnOtherHost: %v, got %v",
i, test.staleOnOtherHost, !test.staleOnOtherHost)
} }
} }