-
Notifications
You must be signed in to change notification settings - Fork 49
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Showing
25 changed files
with
1,323 additions
and
103 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
93 changes: 93 additions & 0 deletions
93
pkg/inst-api-semconv/instrumenter/db/db_client_extractor_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,93 @@ | ||
| package db | ||
|
|
||
| import ( | ||
| "context" | ||
| "github.com/alibaba/opentelemetry-go-auto-instrumentation/pkg/inst-api/utils" | ||
| "go.opentelemetry.io/otel/attribute" | ||
| semconv "go.opentelemetry.io/otel/semconv/v1.19.0" | ||
| "log" | ||
| "testing" | ||
| ) | ||
|
|
||
| type testRequest struct { | ||
| Name string | ||
| Operation string | ||
| } | ||
|
|
||
| type testResponse struct { | ||
| } | ||
|
|
||
| type mongoAttrsGetter struct { | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetSystem(request testRequest) string { | ||
| return "test" | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetUser(request testRequest) string { | ||
| return "test" | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetName(request testRequest) string { | ||
| if request.Name != "" { | ||
| return request.Name | ||
| } | ||
| return "" | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetConnectionString(request testRequest) string { | ||
| return "test" | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetStatement(request testRequest) string { | ||
| return "test" | ||
| } | ||
|
|
||
| func (m mongoAttrsGetter) GetOperation(request testRequest) string { | ||
| if request.Operation != "" { | ||
| return request.Operation | ||
| } | ||
| return "" | ||
| } | ||
|
|
||
| func TestGetSpanKey(t *testing.T) { | ||
| dbExtractor := &DbClientAttrsExtractor[testRequest, any, mongoAttrsGetter]{} | ||
| if dbExtractor.GetSpanKey() != utils.DB_CLIENT_KEY { | ||
| t.Fatalf("Should have returned DB_CLIENT_KEY") | ||
| } | ||
| } | ||
|
|
||
| func TestDbClientExtractorStart(t *testing.T) { | ||
| dbExtractor := DbClientAttrsExtractor[testRequest, testResponse, mongoAttrsGetter]{} | ||
| attrs := make([]attribute.KeyValue, 0) | ||
| parentContext := context.Background() | ||
| attrs = dbExtractor.OnStart(attrs, parentContext, testRequest{Name: "test"}) | ||
| if attrs[0].Key != semconv.DBNameKey || attrs[0].Value.AsString() != "test" { | ||
| t.Fatalf("db name should be test") | ||
| } | ||
| if attrs[1].Key != semconv.DBSystemKey || attrs[1].Value.AsString() != "test" { | ||
| t.Fatalf("db system should be test") | ||
| } | ||
| if attrs[2].Key != semconv.DBUserKey || attrs[2].Value.AsString() != "test" { | ||
| t.Fatalf("db user should be test") | ||
| } | ||
| if attrs[3].Key != semconv.DBConnectionStringKey || attrs[3].Value.AsString() != "test" { | ||
| t.Fatalf("db connection key should be test") | ||
| } | ||
| if attrs[4].Key != semconv.DBStatementKey || attrs[4].Value.AsString() != "test" { | ||
| t.Fatalf("db statement key should be test") | ||
| } | ||
| if attrs[5].Key != semconv.DBOperationKey || attrs[5].Value.AsString() != "" { | ||
| t.Fatalf("db operation key should be empty") | ||
| } | ||
| } | ||
|
|
||
| func TestDbClientExtractorEnd(t *testing.T) { | ||
| dbExtractor := DbClientAttrsExtractor[testRequest, testResponse, mongoAttrsGetter]{} | ||
| attrs := make([]attribute.KeyValue, 0) | ||
| parentContext := context.Background() | ||
| attrs = dbExtractor.OnEnd(attrs, parentContext, testRequest{Name: "test"}, testResponse{}, nil) | ||
| if len(attrs) != 0 { | ||
| log.Fatal("attrs should be empty") | ||
| } | ||
| } |
18 changes: 18 additions & 0 deletions
18
pkg/inst-api-semconv/instrumenter/db/db_span_name_extractor_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
| package db | ||
|
|
||
| import "testing" | ||
|
|
||
| func TestDbNameExtractor(t *testing.T) { | ||
| dbSpanNameExtractor := DBSpanNameExtractor[testRequest]{ | ||
| getter: mongoAttrsGetter{}, | ||
| } | ||
| if dbSpanNameExtractor.Extract(testRequest{}) != "DB Query" { | ||
| t.Fatalf("Should have returned DB_QUERY") | ||
| } | ||
| if dbSpanNameExtractor.Extract(testRequest{Name: "test"}) != "test" { | ||
| t.Fatalf("Should have returned test") | ||
| } | ||
| if dbSpanNameExtractor.Extract(testRequest{Operation: "op_test"}) != "op_test" { | ||
| t.Fatalf("Should have returned op_test") | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.