Skip to content

Commit d7f736c

Browse files
committed
test: probe config
1 parent e3a5420 commit d7f736c

File tree

3 files changed

+38
-0
lines changed

3 files changed

+38
-0
lines changed

tests/probe_config/__init__.py

Whitespace-only changes.
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
spp:
2+
platform_host: "spp-platform"
3+
platform_port: 12800
4+
ssl_enabled: false
5+
skywalking:
6+
logging:
7+
level: "WARN"
8+
agent:
9+
service_name: "tutorial-jvm"

tests/probe_config/tests.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
import unittest
2+
from unittest import mock
3+
4+
from sourceplusplus.SourcePlusPlus import SourcePlusPlus
5+
6+
7+
class TestSum(unittest.TestCase):
8+
9+
def test_default_config(self):
10+
spp = SourcePlusPlus()
11+
self.assertEqual("localhost", spp.probe_config["spp"]["platform_host"])
12+
self.assertEqual(12800, spp.probe_config["spp"]["platform_port"])
13+
self.assertEqual(True, spp.probe_config["spp"]["verify_host"])
14+
self.assertEqual(True, spp.probe_config["spp"]["ssl_enabled"])
15+
self.assertIsNotNone(spp.probe_config["spp"]["probe_id"])
16+
self.assertEqual("localhost:11800", spp.probe_config["skywalking"]["collector"]["backend_service"])
17+
self.assertEqual("spp", spp.probe_config["skywalking"]["agent"]["service_name"])
18+
19+
@mock.patch.dict('os.environ', {'SPP_PROBE_CONFIG_FILE': 'resources/base-config.yml'})
20+
def test_config_from_env(self):
21+
spp = SourcePlusPlus()
22+
self.assertEqual("spp-platform", spp.probe_config["spp"]["platform_host"])
23+
self.assertEqual(12800, spp.probe_config["spp"]["platform_port"])
24+
self.assertEqual(True, spp.probe_config["spp"]["verify_host"])
25+
self.assertEqual(False, spp.probe_config["spp"]["ssl_enabled"])
26+
self.assertEqual("spp-platform:11800", spp.probe_config["skywalking"]["collector"]["backend_service"])
27+
self.assertEqual("tutorial-jvm", spp.probe_config["skywalking"]["agent"]["service_name"])
28+
self.assertEqual("WARN", spp.probe_config["skywalking"]["logging"]["level"])
29+

0 commit comments

Comments
 (0)