121 lines
2.5 KiB
Go
121 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
"testing"
|
|
|
|
pb "github.com/google/protobuf/examples/tutorial"
|
|
)
|
|
|
|
func TestWritePersonWritesPerson(t *testing.T) {
|
|
buf := new(bytes.Buffer)
|
|
// [START populate_proto]
|
|
p := pb.Person{
|
|
Id: 1234,
|
|
Name: "John Doe",
|
|
Email: "jdoe@example.com",
|
|
Phones: []*pb.Person_PhoneNumber{
|
|
{Number: "555-4321", Type: pb.Person_HOME},
|
|
},
|
|
}
|
|
// [END populate_proto]
|
|
writePerson(buf, &p)
|
|
got := buf.String()
|
|
want := `Person ID: 1234
|
|
Name: John Doe
|
|
E-mail address: jdoe@example.com
|
|
Home phone #: 555-4321
|
|
`
|
|
if got != want {
|
|
t.Errorf("writePerson(%s) =>\n\t%q, want %q", p.String(), got, want)
|
|
}
|
|
}
|
|
|
|
func TestListPeopleWritesList(t *testing.T) {
|
|
buf := new(bytes.Buffer)
|
|
in := pb.AddressBook{[]*pb.Person{
|
|
{
|
|
Name: "John Doe",
|
|
Id: 101,
|
|
Email: "john@example.com",
|
|
},
|
|
{
|
|
Name: "Jane Doe",
|
|
Id: 102,
|
|
},
|
|
{
|
|
Name: "Jack Doe",
|
|
Id: 201,
|
|
Email: "jack@example.com",
|
|
Phones: []*pb.Person_PhoneNumber{
|
|
{Number: "555-555-5555", Type: pb.Person_WORK},
|
|
},
|
|
},
|
|
{
|
|
Name: "Jack Buck",
|
|
Id: 301,
|
|
Email: "buck@example.com",
|
|
Phones: []*pb.Person_PhoneNumber{
|
|
{Number: "555-555-0000", Type: pb.Person_HOME},
|
|
{Number: "555-555-0001", Type: pb.Person_MOBILE},
|
|
{Number: "555-555-0002", Type: pb.Person_WORK},
|
|
},
|
|
},
|
|
{
|
|
Name: "Janet Doe",
|
|
Id: 1001,
|
|
Email: "janet@example.com",
|
|
Phones: []*pb.Person_PhoneNumber{
|
|
{Number: "555-777-0000"},
|
|
{Number: "555-777-0001", Type: pb.Person_HOME},
|
|
},
|
|
},
|
|
}}
|
|
listPeople(buf, &in)
|
|
want := strings.Split(`Person ID: 101
|
|
Name: John Doe
|
|
E-mail address: john@example.com
|
|
Person ID: 102
|
|
Name: Jane Doe
|
|
Person ID: 201
|
|
Name: Jack Doe
|
|
E-mail address: jack@example.com
|
|
Work phone #: 555-555-5555
|
|
Person ID: 301
|
|
Name: Jack Buck
|
|
E-mail address: buck@example.com
|
|
Home phone #: 555-555-0000
|
|
Mobile phone #: 555-555-0001
|
|
Work phone #: 555-555-0002
|
|
Person ID: 1001
|
|
Name: Janet Doe
|
|
E-mail address: janet@example.com
|
|
Mobile phone #: 555-777-0000
|
|
Home phone #: 555-777-0001
|
|
`, "\n")
|
|
got := strings.Split(buf.String(), "\n")
|
|
if len(got) != len(want) {
|
|
t.Errorf(
|
|
"listPeople(%s) =>\n\t%q has %d lines, want %d",
|
|
in.String(),
|
|
buf.String(),
|
|
len(got),
|
|
len(want))
|
|
}
|
|
lines := len(got)
|
|
if lines > len(want) {
|
|
lines = len(want)
|
|
}
|
|
for i := 0; i < lines; i++ {
|
|
if got[i] != want[i] {
|
|
t.Errorf(
|
|
"listPeople(%s) =>\n\tline %d %q, want %q",
|
|
in.String(),
|
|
i,
|
|
got[i],
|
|
want[i])
|
|
}
|
|
}
|
|
}
|