gitlab.com/apertussolutions/u-root@v7.0.0+incompatible/pkg/lockfile/lockfile_test.go (about) 1 // Copyright 2018 the u-root Authors. All rights reserved 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package lockfile 6 7 import ( 8 "io/ioutil" 9 "os" 10 "os/exec" 11 "path/filepath" 12 "testing" 13 ) 14 15 func testProcess(t *testing.T) *os.Process { 16 p := exec.Command("sleep", "1000") 17 if err := p.Start(); err != nil { 18 t.Fatal(err) 19 } 20 21 go p.Wait() 22 return p.Process 23 } 24 25 func TestTryLock(t *testing.T) { 26 p1 := testProcess(t) 27 defer p1.Kill() 28 p2 := testProcess(t) 29 defer p2.Kill() 30 31 dir, err := ioutil.TempDir("", "lockfile-") 32 if err != nil { 33 t.Fatal(err) 34 } 35 defer os.RemoveAll(dir) 36 37 l := &Lockfile{ 38 path: filepath.Join(dir, "test1"), 39 pid: p1.Pid, 40 } 41 if err := l.TryLock(); err != nil { 42 t.Fatalf("l(%d).TryLock() = %v, want nil", l.pid, err) 43 } 44 45 l2 := &Lockfile{ 46 path: l.path, 47 pid: p2.Pid, 48 } 49 if err := l2.TryLock(); err != ErrBusy { 50 t.Fatalf("l(%d).TryLock() = %v, want ErrBusy", l2.pid, err) 51 } 52 53 if err := l.Unlock(); err != nil { 54 t.Fatal(err) 55 } 56 57 if err := l.Unlock(); err != ErrRogueDeletion { 58 t.Fatalf("2nd Unlock() = %v, want ErrRogueDeletion", err) 59 } 60 61 if err := l2.TryLock(); err != nil { 62 t.Fatalf("l(%d).TryLock() = %v, want nil", l2.pid, err) 63 } 64 } 65 66 func TestLockFileRemoval(t *testing.T) { 67 p := testProcess(t) 68 defer p.Kill() 69 70 dir, err := ioutil.TempDir("", "lockfile-") 71 if err != nil { 72 t.Fatal(err) 73 } 74 defer os.RemoveAll(dir) 75 76 l := &Lockfile{ 77 path: filepath.Join(dir, "test2"), 78 pid: p.Pid, 79 } 80 if err := l.TryLock(); err != nil { 81 t.Fatalf("l(%d).TryLock() = %v, want nil", l.pid, err) 82 } 83 84 // Evil actor deletes the lockfile. 85 if err := os.Remove(l.path); err != nil { 86 t.Fatalf("remove(%v) = %v, want nil", l.path, err) 87 } 88 89 if err := l.Unlock(); err != ErrRogueDeletion { 90 t.Fatalf("l(%d).Unlock() = %v, want ErrRogueDeletion", l.pid, err) 91 } 92 93 if err := l.TryLock(); err != nil { 94 t.Fatalf("l(%d).TryLock() = %v, want nil", l.pid, err) 95 } 96 } 97 98 func TestDeadProcess(t *testing.T) { 99 p1 := testProcess(t) 100 defer p1.Kill() 101 p2 := testProcess(t) 102 defer p2.Kill() 103 104 dir, err := ioutil.TempDir("", "lockfile-") 105 if err != nil { 106 t.Fatal(err) 107 } 108 defer os.RemoveAll(dir) 109 110 l := &Lockfile{ 111 path: filepath.Join(dir, "test3"), 112 pid: p1.Pid, 113 } 114 if err := l.TryLock(); err != nil { 115 t.Fatalf("l(%d).TryLock() = %v, want nil", l.pid, err) 116 } 117 if err := p1.Kill(); err != nil { 118 t.Fatalf("Kill() = %v, want nil", err) 119 } 120 p1.Wait() 121 122 l2 := &Lockfile{ 123 path: l.path, 124 pid: p2.Pid, 125 } 126 if err := l2.TryLock(); err != nil { 127 t.Fatalf("l(%d).TryLock() = %v, want nil", l2.pid, err) 128 } 129 if err := l2.Unlock(); err != nil { 130 t.Fatalf("l(%d).Unlock() = %v, want nil", l2.pid, err) 131 } 132 }