def test_patch_traces(self):
from google.cloud.trace_v1.gapic import trace_service_client
from google.cloud.trace_v1.proto.trace_pb2 import (TraceSpan, Trace,
Traces)
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
from google.cloud._helpers import _datetime_to_pb_timestamp
trace_id = 'test_trace_id'
span_id = 1234
span_name = 'test_span_name'
start_time = datetime.datetime.utcnow()
end_time = datetime.datetime.utcnow()
traces = {
'traces': [
{
'projectId':
self.project,
'traceId':
trace_id,
'spans': [
{
'spanId': span_id,
'name': span_name,
'startTime': start_time.isoformat() + 'Z',
'endTime': end_time.isoformat() + 'Z',
},
],
},
],
}
traces_pb = _traces_mapping_to_pb(traces)
gapic_api = mock.Mock(spec=trace_service_client.TraceServiceClient)
_, api = self._make_one(gapic_api, None)
api.patch_traces(project_id=self.project, traces=traces)
gapic_api.patch_traces.assert_called_with(self.project, traces_pb)
call_args = gapic_api.patch_traces.call_args[0]
self.assertEqual(len(call_args), 2)
traces_called = call_args[1]
self.assertEqual(len(traces_called.traces), 1)
trace = traces_called.traces[0]
self.assertEqual(len(trace.spans), 1)
span = trace.spans[0]
self.assertIsInstance(traces_called, Traces)
self.assertEqual(trace.project_id, self.project)
self.assertEqual(trace.trace_id, trace_id)
self.assertIsInstance(trace, Trace)
self.assertEqual(span.span_id, span_id)
self.assertEqual(span.name, span_name)
self.assertEqual(span.start_time,
_datetime_to_pb_timestamp(start_time))
self.assertEqual(span.end_time, _datetime_to_pb_timestamp(end_time))
self.assertIsInstance(span, TraceSpan)
def test_patch_traces(self):
from google.cloud.trace_v1.gapic import trace_service_client
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
trace_id = "test_trace_id"
span_id = 1234
span_name = "test_span_name"
start_time = datetime.datetime.utcnow()
end_time = datetime.datetime.utcnow()
traces = {
"traces": [
{
"projectId": self.project,
"traceId": trace_id,
"spans": [
{
"spanId": span_id,
"name": span_name,
"startTime": start_time.isoformat() + "Z",
"endTime": end_time.isoformat() + "Z",
}
],
}
]
}
traces_pb = _traces_mapping_to_pb(traces)
gapic_api = mock.create_autospec(trace_service_client.TraceServiceClient)
api = self._make_one(gapic_api, None)
api.patch_traces(project_id=self.project, traces=traces)
gapic_api.patch_traces.assert_called_once_with(self.project, traces_pb)
def _make_trace_pb(self, project, trace_id, span_id, span_name, start_time,
end_time, parent_span_id, labels):
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
span_kind = 2
traces = {
'traces': [
{
'projectId':
project,
'traceId':
trace_id,
'spans': [
{
'spanId': span_id,
'name': span_name,
'startTime': start_time,
'endTime': end_time,
'kind': span_kind,
'parentSpanId': parent_span_id,
'labels': labels,
},
],
},
],
}
traces_pb = _traces_mapping_to_pb(traces)
trace_pb = traces_pb.traces
return trace_pb
def test_patch_traces(self):
from google.cloud.trace_v1.gapic import trace_service_client
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
trace_id = "test_trace_id"
span_id = 1234
span_name = "test_span_name"
start_time = datetime.datetime.utcnow()
end_time = datetime.datetime.utcnow()
traces = {
"traces": [{
"projectId":
self.project,
"traceId":
trace_id,
"spans": [{
"spanId": span_id,
"name": span_name,
"startTime": start_time.isoformat() + "Z",
"endTime": end_time.isoformat() + "Z",
}],
}]
}
traces_pb = _traces_mapping_to_pb(traces)
gapic_api = mock.create_autospec(
trace_service_client.TraceServiceClient)
api = self._make_one(gapic_api, None)
api.patch_traces(project_id=self.project, traces=traces)
gapic_api.patch_traces.assert_called_once_with(self.project, traces_pb)
def test_patch_traces(self):
from google.cloud.trace_v1.gapic import trace_service_client
from google.cloud.trace_v1.proto.trace_pb2 import TraceSpan, Trace, Traces
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
from google.cloud._helpers import _datetime_to_pb_timestamp
trace_id = "test_trace_id"
span_id = 1234
span_name = "test_span_name"
start_time = datetime.datetime.utcnow()
end_time = datetime.datetime.utcnow()
traces = {
"traces": [
{
"projectId": self.project,
"traceId": trace_id,
"spans": [
{
"spanId": span_id,
"name": span_name,
"startTime": start_time.isoformat() + "Z",
"endTime": end_time.isoformat() + "Z",
}
],
}
]
}
traces_pb = _traces_mapping_to_pb(traces)
gapic_api = mock.Mock(spec=trace_service_client.TraceServiceClient)
_, api = self._make_one(gapic_api, None)
api.patch_traces(project_id=self.project, traces=traces)
gapic_api.patch_traces.assert_called_with(self.project, traces_pb)
call_args = gapic_api.patch_traces.call_args[0]
self.assertEqual(len(call_args), 2)
traces_called = call_args[1]
self.assertEqual(len(traces_called.traces), 1)
trace = traces_called.traces[0]
self.assertEqual(len(trace.spans), 1)
span = trace.spans[0]
self.assertIsInstance(traces_called, Traces)
self.assertEqual(trace.project_id, self.project)
self.assertEqual(trace.trace_id, trace_id)
self.assertIsInstance(trace, Trace)
self.assertEqual(span.span_id, span_id)
self.assertEqual(span.name, span_name)
self.assertEqual(span.start_time, _datetime_to_pb_timestamp(start_time))
self.assertEqual(span.end_time, _datetime_to_pb_timestamp(end_time))
self.assertIsInstance(span, TraceSpan)
def _make_trace_pb(
self,
project,
trace_id,
span_id,
span_name,
start_time,
end_time,
parent_span_id,
labels,
):
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
span_kind = 2
traces = {
"traces": [
{
"projectId": project,
"traceId": trace_id,
"spans": [
{
"spanId": span_id,
"name": span_name,
"startTime": start_time,
"endTime": end_time,
"kind": span_kind,
"parentSpanId": parent_span_id,
"labels": labels,
}
],
}
]
}
traces_pb = _traces_mapping_to_pb(traces)
trace_pb = traces_pb.traces
return trace_pb
def _make_trace_pb(
self,
project,
trace_id,
span_id,
span_name,
start_time,
end_time,
parent_span_id,
labels):
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
span_kind = 2
traces = {
'traces': [
{
'projectId': project,
'traceId': trace_id,
'spans': [
{
'spanId': span_id,
'name': span_name,
'startTime': start_time,
'endTime': end_time,
'kind': span_kind,
'parentSpanId': parent_span_id,
'labels': labels,
},
],
},
],
}
traces_pb = _traces_mapping_to_pb(traces)
trace_pb = traces_pb.traces
return trace_pb
def _make_trace_pb(
self,
project,
trace_id,
span_id,
span_name,
start_time,
end_time,
parent_span_id,
labels,
):
from google.cloud.trace.v1._gapic import _traces_mapping_to_pb
span_kind = 2
traces = {
"traces": [{
"projectId":
project,
"traceId":
trace_id,
"spans": [{
"spanId": span_id,
"name": span_name,
"startTime": start_time,
"endTime": end_time,
"kind": span_kind,
"parentSpanId": parent_span_id,
"labels": labels,
}],
}]
}
traces_pb = _traces_mapping_to_pb(traces)
trace_pb = traces_pb.traces
return trace_pb