Defer unlock for testing daemon

Make sure we use defer when unlocking the daemon during testing. Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
containerd · Aug 16, 2017 · bfd62ce · bfd62ce
1 parent 73ac66e
commit bfd62ce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/daemon_test.go b/daemon_test.go
@@ -60,7 +60,7 @@ func (d *daemon) waitForStart(ctx context.Context) (*Client, error) {
 
 func (d *daemon) Stop() error {
 	d.Lock()
-	d.Unlock()
+	defer d.Unlock()
 	if d.cmd == nil {
 		return errors.New("daemon is not running")
 	}
@@ -69,7 +69,7 @@ func (d *daemon) Stop() error {
 
 func (d *daemon) Kill() error {
 	d.Lock()
-	d.Unlock()
+	defer d.Unlock()
 	if d.cmd == nil {
 		return errors.New("daemon is not running")
 	}
@@ -78,7 +78,7 @@ func (d *daemon) Kill() error {
 
 func (d *daemon) Wait() error {
 	d.Lock()
-	d.Unlock()
+	defer d.Unlock()
 	if d.cmd == nil {
 		return errors.New("daemon is not running")
 	}
@@ -87,7 +87,7 @@ func (d *daemon) Wait() error {
 
 func (d *daemon) Restart() error {
 	d.Lock()
-	d.Unlock()
+	defer d.Unlock()
 	if d.cmd == nil {
 		return errors.New("daemon is not running")
 	}