-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathtest_base.py
More file actions
84 lines (60 loc) · 3.04 KB
/
test_base.py
File metadata and controls
84 lines (60 loc) · 3.04 KB
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
import logging
from tests.bases import ClearEnvTest
from stackify.protos import stackify_agent_pb2
from stackify.transport.base import BaseTransport
from stackify.transport.base import AgentBaseTransport
from stackify.transport.application import EnvironmentDetail
from stackify.transport.application import get_configuration
CONFIG = {
'application': 'test_appname',
'environment': 'test_environment',
'api_key': 'test_apikey',
'api_url': 'test_apiurl',
}
class BaseTransportTest(ClearEnvTest):
def test_init(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
base_transport = BaseTransport(api_config, env_details)
assert base_transport._api_config == api_config
assert base_transport._env_details == env_details
def test_create_message(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
base_transport = BaseTransport(api_config, env_details)
self.assertRaises(NotImplementedError, base_transport.create_message, 'test_record')
def test_create_group_message(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
base_transport = BaseTransport(api_config, env_details)
self.assertRaises(NotImplementedError, base_transport.create_group_message, 'test_messages')
def test_send(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
base_transport = BaseTransport(api_config, env_details)
self.assertRaises(NotImplementedError, base_transport.send, 'test_group_message')
class AgentBaseTransportTest(ClearEnvTest):
def test_init(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
agent_base_transport = AgentBaseTransport(api_config, env_details)
assert agent_base_transport._api_config == api_config
assert agent_base_transport._env_details == env_details
def test_create_message(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
agent_base_transport = AgentBaseTransport(api_config, env_details)
message = agent_base_transport.create_message(logging.makeLogRecord({'mgs': 'message', 'funcName': 'foo'}))
assert isinstance(message, stackify_agent_pb2.LogGroup.Log)
def test_create_group_message(self):
api_config = get_configuration(**CONFIG)
env_details = EnvironmentDetail(api_config)
agent_base_transport = AgentBaseTransport(api_config, env_details)
message = agent_base_transport.create_message(logging.makeLogRecord({'mgs': 'message', 'funcName': 'foo'}))
group_message = agent_base_transport.create_group_message([message])
assert isinstance(group_message, stackify_agent_pb2.LogGroup)
def test_send(self):
api_config = 'test_api_config'
env_details = 'test_env_details'
agent_base_transport = AgentBaseTransport(api_config, env_details)
self.assertRaises(AttributeError, agent_base_transport.send, 'test_group_message')