Merge pull request #400 from nak3/fingerprint-test

Update fingerprint test codes
hashicorp · Nov 11, 2015 · f28b9e7 · f28b9e7
2 parents fbabd12 + b66244b
commit f28b9e7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/client/fingerprint/cpu_test.go b/client/fingerprint/cpu_test.go
@@ -35,8 +35,8 @@ func TestCPUFingerprint(t *testing.T) {
 		t.Fatalf("Missing CPU Total Compute")
 	}
 
-	if node.Resources == nil {
-		t.Fatalf("Expected resources on node, but got nil")
+	if node.Resources == nil || node.Resources.CPU == 0 {
+		t.Fatalf("Expected to find CPU Resources")
 	}
 
 }
diff --git a/client/fingerprint/memory_test.go b/client/fingerprint/memory_test.go
@@ -25,5 +25,8 @@ func TestMemoryFingerprint(t *testing.T) {
 	if node.Resources == nil {
 		t.Fatalf("Node Resources was nil")
 	}
+	if node.Resources.MemoryMB == 0 {
+		t.Errorf("Expected node.Resources.MemoryMB to be non-zero")
+	}
 
 }
diff --git a/client/fingerprint/storage_test.go b/client/fingerprint/storage_test.go
@@ -29,9 +29,12 @@ func TestStorageFingerprint(t *testing.T) {
 	}
 
 	if free > total {
-		t.Errorf("storage.bytesfree %d is larger than storage.bytestotal %d", free, total)
+		t.Fatalf("storage.bytesfree %d is larger than storage.bytestotal %d", free, total)
 	}
 
+	if node.Resources == nil {
+		t.Fatalf("Node Resources was nil")
+	}
 	if node.Resources.DiskMB == 0 {
 		t.Errorf("Expected node.Resources.DiskMB to be non-zero")
 	}