package api import ( "context" "testing" "buf.build/go/protovalidate" "github.com/google/uuid" "github.com/stretchr/testify/assert" auditV1 "dev.azure.com/schwarzit/schwarzit.stackit-public/audit-go.git/gen/go/audit/v1" internalAuditApi "dev.azure.com/schwarzit/schwarzit.stackit-public/audit-go.git/internal/audit/api" pkgAuditCommon "dev.azure.com/schwarzit/schwarzit.stackit-public/audit-go.git/pkg/audit/common" pkgAuditUtils "dev.azure.com/schwarzit/schwarzit.stackit-public/audit-go.git/pkg/audit/utils" ) func Test_LogEvent(t *testing.T) { api, _ := NewMockAuditApi() sequenceNumberGenerator := pkgAuditUtils.NewDefaultSequenceNumberGenerator() t.Run("new format", func(t *testing.T) { eventBuilder := NewAuditEventBuilder(api, sequenceNumberGenerator, "demo-service", uuid.NewString(), "eu01") cloudEvent, _, err := eventBuilder. WithRequiredApiRequest(pkgAuditCommon.ApiRequest{ Body: nil, Header: internalAuditApi.TestHeaders, Host: "localhost", Method: "GET", Scheme: "https", Proto: "HTTP/1.1", URL: pkgAuditCommon.RequestUrl{ Path: "/", RawQuery: nil, }, }). WithRequiredObjectId(uuid.NewString()). WithRequiredObjectType(pkgAuditCommon.ObjectTypeProject). WithRequiredOperation("stackit.demo-service.v1.project.update"). WithRequiredRequestClientIp("0.0.0.0"). Build(context.Background(), eventBuilder.NextSequenceNumber()) assert.NoError(t, err) assert.NoError(t, LogEvent(cloudEvent)) }) t.Run("legacy format", func(t *testing.T) { objectId := uuid.NewString() entry, err := NewAuditLogEntryBuilder(). WithRequiredApiRequest(pkgAuditCommon.ApiRequest{ Body: nil, Header: internalAuditApi.TestHeaders, Host: "localhost", Method: "GET", Scheme: "https", Proto: "HTTP/1.1", URL: pkgAuditCommon.RequestUrl{ Path: "/", RawQuery: nil, }, }). WithRequiredLocation("eu01"). WithRequiredObjectId(objectId). WithRequiredObjectType(pkgAuditCommon.ObjectTypeProject). WithRequiredOperation("stackit.demo-service.v1.project.update"). WithRequiredRequestClientIp("0.0.0.0"). WithRequiredServiceName("demo-service"). WithRequiredWorkerId(uuid.NewString()). Build(context.Background(), SequenceNumber(1)) assert.NoError(t, err) validator, err := protovalidate.New() assert.NoError(t, err) var protoValidator pkgAuditCommon.ProtobufValidator = validator routableIdentifier := pkgAuditCommon.RoutableIdentifier{ Identifier: objectId, Type: pkgAuditCommon.ObjectTypeProject, } routableEvent, err := internalAuditApi.ValidateAndSerializePartially(protoValidator, entry, auditV1.Visibility_VISIBILITY_PUBLIC, &routableIdentifier) assert.NoError(t, err) legacyBytes, err := internalAuditApi.ConvertAndSerializeIntoLegacyFormat(entry, routableEvent) assert.NoError(t, err) cloudEvent := pkgAuditCommon.CloudEvent{ SpecVersion: "1.0", Source: entry.ProtoPayload.ServiceName, Id: entry.InsertId, Time: entry.ProtoPayload.RequestMetadata.RequestAttributes.Time.AsTime(), DataContentType: pkgAuditCommon.ContentTypeCloudEventsJson, DataType: DataTypeLegacyAuditEventV1, Subject: entry.ProtoPayload.ResourceName, Data: legacyBytes, TraceParent: nil, TraceState: nil, } assert.NoError(t, LogEvent(&cloudEvent)) }) }