Testing your code during development can expose bugs that find their way in as you make changes. In this topic, you add a test for the Hello
function.
Go's built-in support for unit testing makes it easier to test as you go. Specifically, using naming conventions, Go's testing
package, and the go test
command, you can quickly write and execute tests.
Ending a file's name with _test.go tells the go test
command that this file contains test functions.
package greetings
import (
"regexp"
"testing"
)
// TestHelloName calls greetings.Hello with a name, checking for a valid return value.
func TestHelloName(t *testing.T) {
name := "Gladys"
want := regexp.MustCompile(`\b` + name + `\b`)
msg, err := Hello("Gladys")
if !want.MatchString(msg) || err != nil {
t.Fatalf(`Hello("Gladys")= %q, %v, want match for %#q, nil`, msg, err, want)
}
}
// TestHelloEmpty calls greetings.Hello with an empty string, checking for an error
func TestHelloEmpty(t *testing.T) {
msg, err := Hello("")
if msg != "" || err == nil {
t.Fatalf(`Hello("") = %q, %v, want "", error`, msg, err)
}
}
greetings.Hello
function. Test function names have the form TestName
, where Name says something about the specific test. Also, test functions take a pointer to the testing
package's testing.T type as a parameter. You use this parameter's methods for reporting and logging from your test.TestHelloName
calls the Hello
function, passing a name
value with which the function should be able to return a valid response message. If the call returns an error or an unexpected response message (one that doesn't include the name you passed in), you use the t
parameter's Fatalf method to print a message to the console and end execution.TestHelloEmpty
calls the Hello
function with an empty string. This test is designed to confirm that your error handling works. If the call returns a non-empty string or no error, you use the t
parameter's Fatalf
method to print a message to the console and end execution.The go test
command executes test functions (whose names begin with Test
) in test files (whose names end with _test.go). You can add the -v
flag to get verbose output that lists all of the tests and their results.
package greetings
import (
"errors"
"fmt"
"math/rand"
)
// Hello returns a greeting for the named person.
func Hello(name string) (string, error) {
// If no name was given, return an error with a message
if name == "" {
return name, errors.New("empty name")
}
// Create a message using a random format.
message := fmt.Sprintf(randomFormat(), name)
//message := fmt.Sprint(randomFormat())
return message, nil
}
// Hellos returns a map that associates each of the named people with a greeting message.
func Hellos(names []string) (map[string]string, error) {
// A map to associate names with messages.
messages := make(map[string]string)
// Loop through the received slice of names, calling the Hello function to get a message for each name.
for _, name := range names {
message, err := Hello(name)
if err != nil {
return nil, err
}
// In the map, associate the retrieved message with the name.
messages[name] = message
}
return messages, nil
}
// randomFormat returns one of a set of greeting messages. The returned message is selected at random.
func randomFormat() string {
// A slice of message formats.
formats := []string{
"Hi, %v. Welcome!",
"Great to see you, %v!",
"Hail, %v! Well met!",
}
// Return one of the message formats selected at random.
return formats[rand.Intn(len(formats))]
}
package main
import (
"fmt"
"log"
"example.com/greetings"
)
func main() {
// Set properties of the predefined Logger, including
// the log entry prefix and a flag to disable printing
// the time, source file, and line number.
log.SetPrefix("greetings: ")
log.SetFlags(0)
// A slice of names.
names := []string{"Gladys", "Samantha", "Darrin"}
// Request greeting messages for the names.
messages, err := greetings.Hellos(names)
if err != nil {
log.Fatal(err)
}
// If no error was returned, print the returned map of
// messages to the console.
fmt.Println(messages)
}
The tests should pass.
greetings.Hello
function to view a failing test.The TestHelloName
test function checks the return value for the name you specified as a Hello
function parameter. To view a failing test result, change the greetings.Hello
function so that it no longer includes the name.
In greetings/greetings.go, paste the following code in place of the Hello
function. Note that the highlighted lines change the value that the function returns, as if the name
argument had been accidentally removed.
// Hello returns a greeting for the named person.
func Hello(name string) (string, error) {
// If no name was given, return an error with a message
if name == "" {
return name, errors.New("empty name")
}
// Create a message using a random format.
//message := fmt.Sprintf(randomFormat(), name)
message := fmt.Sprint(randomFormat())
return message, nil
}
go test
to execute the test.This time, run go test
without the -v
flag. The output will include results for only the tests that failed, which can be useful when you have a lot of tests. The TestHelloName
test should fail -- TestHelloEmpty
still passes.