| // Copyright 2017 The Go Authors. All rights reserved. | |
| // Use of this source code is governed by a BSD-style | |
| // license that can be found in the LICENSE file. | |
| package main | |
| import ( | |
| "log" | |
| "runtime" | |
| "./mysync" | |
| ) | |
| func main() { | |
| var wg mysync.WaitGroup | |
| wg.Done() | |
| ci := runtime.CallersFrames(wg.Callers) | |
| frames := make([]runtime.Frame, 0, 4) | |
| for { | |
| frame, more := ci.Next() | |
| frames = append(frames, frame) | |
| if !more { | |
| break | |
| } | |
| } | |
| expecting := []string{ | |
| "test/mysync.(*WaitGroup).Add", | |
| "test/mysync.(*WaitGroup).Done", | |
| } | |
| for i := 0; i < 2; i++ { | |
| if frames[i].Function != expecting[i] { | |
| log.Fatalf("frame %d: got %s, want %s", i, frames[i].Function, expecting[i]) | |
| } | |
| } | |
| } |