tests: use testify/assert on context

Author: buchdag

internal/dockergen/context_test.go

@@ -5,6 +5,8 @@ import (
 	"io/ioutil"
 	"os"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 var (
@@ -86,16 +88,12 @@ func TestGetCurrentContainerID(t *testing.T) {
 	// Each time the HOSTNAME is set to a short form ID, GetCurrentContainerID() should match and return the corresponding full ID
 	for _, id := range ids {
 		os.Setenv("HOSTNAME", id[0:12])
-		if got, exp := GetCurrentContainerID(filepaths...), id; got != exp {
-			t.Fatalf("id mismatch with HOSTNAME %v: got %v, exp %v", id[0:12], got, exp)
-		}
+		assert.Equal(t, id, GetCurrentContainerID(filepaths...), "id mismatch with default HOSTNAME")
 	}
 
 	// If the Hostname isn't a short form ID, we should match the first valid ID (64 character hex string) instead
 	os.Setenv("HOSTNAME", "customhostname")
-	if got, exp := GetCurrentContainerID(filepaths...), ids[0]; got != exp {
-		t.Fatalf("id mismatch with custom HOSTNAME: got %v, exp %v", got, exp)
-	}
+	assert.Equal(t, ids[0], GetCurrentContainerID(filepaths...), "id mismatch with custom HOSTNAME")
 }
 
 func TestGetCurrentContainerIDMountInfo(t *testing.T) {
@@ -128,7 +126,5 @@ func TestGetCurrentContainerIDMountInfo(t *testing.T) {
 	}
 
 	// We should match the correct 64 characters long ID in mountinfo, not the first encountered
-	if got, exp := GetCurrentContainerID(filepaths...), id; got != exp {
-		t.Fatalf("id mismatch on mountinfo: got %v, exp %v", got, exp)
-	}
+	assert.Equal(t, id, GetCurrentContainerID(filepaths...), "id mismatch on mountinfo")
 }