/
test__logging.py
87 lines (67 loc) · 2.75 KB
/
test__logging.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# Copyright 2017 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import unittest
import mock
def _make_credentials():
import google.auth.credentials
return mock.Mock(spec=google.auth.credentials.Credentials)
class Test_ErrorReportingLoggingAPI(unittest.TestCase):
PROJECT = "PROJECT"
def _make_one(self, project, credentials, **kw):
from google.cloud.error_reporting._logging import _ErrorReportingLoggingAPI
return _ErrorReportingLoggingAPI(project, credentials, **kw)
@mock.patch("google.cloud.logging.Client")
def test_ctor_defaults(self, mocked_cls):
credentials = _make_credentials()
logging_api = self._make_one(self.PROJECT, credentials)
self.assertIs(logging_api.logging_client, mocked_cls.return_value)
mocked_cls.assert_called_once_with(
project=self.PROJECT,
credentials=credentials,
_http=None,
client_info=None,
client_options=None,
)
@mock.patch("google.cloud.logging.Client")
def test_ctor_explicit(self, mocked_cls):
credentials = _make_credentials()
http = mock.Mock()
client_info = mock.Mock()
client_options = mock.Mock()
logging_api = self._make_one(
self.PROJECT,
credentials,
_http=http,
client_info=client_info,
client_options=client_options,
)
self.assertIs(logging_api.logging_client, mocked_cls.return_value)
mocked_cls.assert_called_once_with(
project=self.PROJECT,
credentials=credentials,
_http=http,
client_info=client_info,
client_options=client_options,
)
@mock.patch("google.cloud.logging.Client")
def test_report_error_event(self, mocked_cls):
credentials = _make_credentials()
logging_api = self._make_one(self.PROJECT, credentials)
logger = mock.Mock(spec=["log_struct"])
logging_api.logging_client.logger.return_value = logger
# Actually make the API call.
error_report = {"message": "The cabs are here."}
logging_api.report_error_event(error_report)
# Check the mocks.
logger.log_struct.assert_called_once_with(error_report)