Browse Source

Code review fixes

Willem van Bergen 10 years ago
parent
commit
3dc39f7c7c
1 changed files with 6 additions and 9 deletions
  1. 6 9
      functional_test.go

+ 6 - 9
functional_test.go

@@ -18,29 +18,26 @@ var (
 )
 
 func init() {
-	env := func(key, defaultValue string) string {
-		if value := os.Getenv(key); value != "" {
-			return value
-		}
-		return defaultValue
+	kafkaAddr = os.Getenv("KAFKA_ADDR")
+	if kafkaAddr == "" {
+		kafkaAddr = "localhost:9092"
 	}
 
-	kafkaAddr := env("KAFKA_ADDR", "localhost:9092")
 	c, err := net.Dial("tcp", kafkaAddr)
 	if err == nil {
 		kafkaIsAvailable = true
 		c.Close()
 	}
 
-	kafkaShouldBeAvailable = (env("CI", "") != "")
+	kafkaShouldBeAvailable = os.Getenv("CI") != ""
 }
 
 func checkKafkaAvailability(t *testing.T) {
 	if !kafkaIsAvailable {
 		if kafkaShouldBeAvailable {
-			t.Fatal("Kafka broker is not available")
+			t.Fatalf("Kafka broker is not available on %s. Set KAFKA_ADDR to connect to Kafka on a different location.", kafkaAddr)
 		} else {
-			t.Skip("Kafka broker is not available")
+			t.Skipf("Kafka broker is not available on %s. Set KAFKA_ADDR to connect to Kafka on a different location.", kafkaAddr)
 		}
 	}
 }