Skip to content

Commit bc92603

Browse files
committed
fixed formatting and testing coverage
Signed-off-by: SoumyaRaikwar <[email protected]>
1 parent aa25cf0 commit bc92603

File tree

2 files changed

+26
-17
lines changed

2 files changed

+26
-17
lines changed

internal/storage/integration/integration.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -425,13 +425,12 @@ func (s *StorageIntegration) writeTrace(t *testing.T, traces ptrace.Traces) {
425425
}
426426

427427
func (s *StorageIntegration) loadParseAndWriteExampleTrace(t *testing.T) *model.Trace {
428-
trace := s.getTraceFixture(t, "example_trace")
429-
otelTraces := v1adapter.V1TraceToOtelTrace(trace)
430-
s.writeTrace(t, otelTraces)
431-
return trace
428+
trace := s.getTraceFixture(t, "example_trace")
429+
otelTraces := v1adapter.V1TraceToOtelTrace(trace)
430+
s.writeTrace(t, otelTraces)
431+
return trace
432432
}
433433

434-
435434
func (s *StorageIntegration) writeLargeTraceWithDuplicateSpanIds(
436435
t *testing.T,
437436
totalCount int,

internal/storage/v2/v1adapter/translator_test.go

Lines changed: 22 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -255,34 +255,44 @@ func TestV1TracesFromSeq2(t *testing.T) {
255255
}
256256
}
257257

258-
func TestV1TraceToOtelTrace_ReturnsExptectedOtelTrace(t *testing.T) {
258+
func TestV1TraceFromOtelTrace_ReturnsExpectedModelTrace(t *testing.T) {
259259
jTrace := &model.Trace{
260260
Spans: []*model.Span{
261261
{
262262
TraceID: model.NewTraceID(2, 3),
263263
SpanID: model.NewSpanID(1),
264264
Process: model.NewProcess("Service1", nil),
265-
OperationName: "two-resources-1",
266-
}, {
265+
OperationName: "test-operation-1",
266+
},
267+
{
267268
TraceID: model.NewTraceID(2, 3),
268269
SpanID: model.NewSpanID(2),
269-
Process: model.NewProcess("service2", nil),
270-
OperationName: "two-resources-2",
270+
Process: model.NewProcess("Service1", nil),
271+
OperationName: "test-operation-2",
271272
},
272273
},
273274
}
274-
actualTrace := V1TraceToOtelTrace(jTrace)
275+
276+
otelTraces := V1TraceToOtelTrace(jTrace)
277+
278+
actualTrace := V1TraceFromOtelTrace(otelTraces)
275279

276280
require.NotEmpty(t, actualTrace)
277-
require.Equal(t, 2, actualTrace.ResourceSpans().Len())
281+
require.Len(t, actualTrace.Spans, 2)
282+
assert.Equal(t, model.NewTraceID(2, 3), actualTrace.Spans[0].TraceID)
283+
assert.Equal(t, model.NewSpanID(1), actualTrace.Spans[0].SpanID)
284+
assert.Equal(t, "test-operation-1", actualTrace.Spans[0].OperationName)
285+
assert.Equal(t, "Service1", actualTrace.Spans[0].Process.ServiceName)
286+
assert.Equal(t, model.NewSpanID(2), actualTrace.Spans[1].SpanID)
287+
assert.Equal(t, "test-operation-2", actualTrace.Spans[1].OperationName)
278288
}
279289

280-
func TestV1TraceToOtelTrace_ReturnEmptyOtelTrace(t *testing.T) {
281-
jTrace := &model.Trace{}
282-
eTrace := ptrace.NewTraces()
283-
aTrace := V1TraceToOtelTrace(jTrace)
290+
func TestV1TraceFromOtelTrace_ReturnEmptyModelTrace(t *testing.T) {
291+
otelTraces := ptrace.NewTraces()
292+
actualTrace := V1TraceFromOtelTrace(otelTraces)
284293

285-
require.Equal(t, eTrace.SpanCount(), aTrace.SpanCount(), 0)
294+
require.NotEmpty(t, actualTrace)
295+
require.Len(t, actualTrace.Spans, 0)
286296
}
287297

288298
func TestV1TraceIDsFromSeq2(t *testing.T) {

0 commit comments

Comments
 (0)