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