mirror of
https://github.com/restic/restic.git
synced 2025-10-27 03:49:13 +00:00
Upadte vendored dependencies
This commit is contained in:
44
vendor/cloud.google.com/go/profiler/profiler_test.go
generated
vendored
44
vendor/cloud.google.com/go/profiler/profiler_test.go
generated
vendored
@@ -52,7 +52,6 @@ const (
|
||||
testSvcVersion = "test-service-version"
|
||||
testProfileDuration = time.Second * 10
|
||||
testServerTimeout = time.Second * 15
|
||||
wantFunctionName = "profilee"
|
||||
)
|
||||
|
||||
func createTestDeployment() *pb.Deployment {
|
||||
@@ -569,10 +568,9 @@ func TestInitializeConfig(t *testing.T) {
|
||||
|
||||
type fakeProfilerServer struct {
|
||||
pb.ProfilerServiceServer
|
||||
count int
|
||||
gotCPUProfile []byte
|
||||
gotHeapProfile []byte
|
||||
done chan bool
|
||||
count int
|
||||
gotProfiles map[string][]byte
|
||||
done chan bool
|
||||
}
|
||||
|
||||
func (fs *fakeProfilerServer) CreateProfile(ctx context.Context, in *pb.CreateProfileRequest) (*pb.Profile, error) {
|
||||
@@ -590,9 +588,9 @@ func (fs *fakeProfilerServer) CreateProfile(ctx context.Context, in *pb.CreatePr
|
||||
func (fs *fakeProfilerServer) UpdateProfile(ctx context.Context, in *pb.UpdateProfileRequest) (*pb.Profile, error) {
|
||||
switch in.Profile.ProfileType {
|
||||
case pb.ProfileType_CPU:
|
||||
fs.gotCPUProfile = in.Profile.ProfileBytes
|
||||
fs.gotProfiles["CPU"] = in.Profile.ProfileBytes
|
||||
case pb.ProfileType_HEAP:
|
||||
fs.gotHeapProfile = in.Profile.ProfileBytes
|
||||
fs.gotProfiles["HEAP"] = in.Profile.ProfileBytes
|
||||
fs.done <- true
|
||||
}
|
||||
|
||||
@@ -629,16 +627,7 @@ func profileeWork() {
|
||||
}
|
||||
}
|
||||
|
||||
func checkSymbolization(p *profile.Profile) error {
|
||||
for _, l := range p.Location {
|
||||
if len(l.Line) > 0 && l.Line[0].Function != nil && strings.Contains(l.Line[0].Function.Name, wantFunctionName) {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return fmt.Errorf("want function name %v not found in profile", wantFunctionName)
|
||||
}
|
||||
|
||||
func validateProfile(rawData []byte) error {
|
||||
func validateProfile(rawData []byte, wantFunctionName string) error {
|
||||
p, err := profile.ParseData(rawData)
|
||||
if err != nil {
|
||||
return fmt.Errorf("ParseData failed: %v", err)
|
||||
@@ -656,10 +645,12 @@ func validateProfile(rawData []byte) error {
|
||||
return fmt.Errorf("profile contains zero functions: %v", p)
|
||||
}
|
||||
|
||||
if err := checkSymbolization(p); err != nil {
|
||||
return fmt.Errorf("checkSymbolization failed: %v for %v", err, p)
|
||||
for _, l := range p.Location {
|
||||
if len(l.Line) > 0 && l.Line[0].Function != nil && strings.Contains(l.Line[0].Function.Name, wantFunctionName) {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return nil
|
||||
return fmt.Errorf("wanted function name %s not found in the profile", wantFunctionName)
|
||||
}
|
||||
|
||||
func TestAgentWithServer(t *testing.T) {
|
||||
@@ -672,7 +663,7 @@ func TestAgentWithServer(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatalf("testutil.NewServer(): %v", err)
|
||||
}
|
||||
fakeServer := &fakeProfilerServer{done: make(chan bool)}
|
||||
fakeServer := &fakeProfilerServer{gotProfiles: map[string][]byte{}, done: make(chan bool)}
|
||||
pb.RegisterProfilerServiceServer(srv.Gsrv, fakeServer)
|
||||
|
||||
srv.Start()
|
||||
@@ -698,10 +689,11 @@ func TestAgentWithServer(t *testing.T) {
|
||||
}
|
||||
quitProfilee <- true
|
||||
|
||||
if err := validateProfile(fakeServer.gotCPUProfile); err != nil {
|
||||
t.Errorf("validateProfile(gotCPUProfile): %v", err)
|
||||
}
|
||||
if err := validateProfile(fakeServer.gotHeapProfile); err != nil {
|
||||
t.Errorf("validateProfile(gotHeapProfile): %v", err)
|
||||
for _, pType := range []string{"CPU", "HEAP"} {
|
||||
if profile, ok := fakeServer.gotProfiles[pType]; !ok {
|
||||
t.Errorf("fakeServer.gotProfiles[%s] got no profile, want profile", pType)
|
||||
} else if err := validateProfile(profile, "profilee"); err != nil {
|
||||
t.Errorf("validateProfile(%s) got error: %v", pType, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user