def test_no_datawire_token(self):
"""
If DATAWIRE_TOKEN is not set neither the TracingClient nor the DiscoClient
are started.
"""
# Disable connecting to our Discovery server:
runtime = fakeRuntime()
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE", "synapse:path=" + mkdtemp())
# Start the MDK:
mdk = MDKImpl(runtime)
mdk.start()
# Do a bunch of logging:
session = mdk.session()
session.info("category", "hello!")
session.error("category", "ono")
session.warn("category", "gazoots")
session.critical("category", "aaaaaaa")
session.debug("category", "behold!")
# Time passes...
scheduleService = runtime.getScheduleService()
for i in range(10):
scheduleService.advance(1.0)
scheduleService.pump()
# No WebSocket connections made:
self.assertFalse(runtime.getWebSocketsService().fakeActors)
def test_no_datawire_token(self):
"""
If DATAWIRE_TOKEN is not set neither the TracingClient nor the DiscoClient
are started.
"""
# Disable connecting to our Discovery server:
runtime = fakeRuntime()
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE",
"synapse:path=" + mkdtemp())
# Start the MDK:
mdk = MDKImpl(runtime)
mdk.start()
# Do a bunch of logging:
session = mdk.session()
session.info("category", "hello!")
session.error("category", "ono")
session.warn("category", "gazoots")
session.critical("category", "aaaaaaa")
session.debug("category", "behold!")
# Time passes...
scheduleService = runtime.getScheduleService()
for i in range(10):
scheduleService.advance(1.0)
scheduleService.pump()
# No WebSocket connections made:
self.assertFalse(runtime.getWebSocketsService().fakeActors)
def setUp(self):
"""Initialize an empty environment."""
# Initialize runtime and MDK:
self.runtime = fakeRuntime()
self.runtime.getEnvVarsService().set("DATAWIRE_TOKEN", "something")
self.mdk = MDKImpl(self.runtime)
self.mdk.start()
self.session = self.mdk.session()
def fake_runtime():
"""
Create a fake runtime suitably configured for MDK usage.
"""
runtime = fakeRuntime()
runtime.dependencies.registerService("failurepolicy_factory",
CircuitBreakerFactory(runtime))
return runtime
def setUp(self):
"""Initialize an empty environment."""
# Initialize runtime and MDK:
self.runtime = fakeRuntime()
self.runtime.getEnvVarsService().set("DATAWIRE_TOKEN", "something")
self.mdk = MDKImpl(self.runtime)
self.mdk.start()
self.session = self.mdk.session()
def fake_runtime():
"""
Create a fake runtime suitably configured for MDK usage.
"""
runtime = fakeRuntime()
runtime.dependencies.registerService(
"failurepolicy_factory", CircuitBreakerFactory(runtime))
return runtime
def create_mdk_with_faketracer(environment="sandbox"):
"""Create an MDK with a FakeTracer.
Returns (mdk, fake_tracer).
"""
runtime = fakeRuntime()
tracer = FakeTracer()
runtime.dependencies.registerService("tracer", tracer)
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE", "static:nodes={}")
runtime.getEnvVarsService().set("MDK_ENVIRONMENT", environment)
mdk = MDKImpl(runtime)
mdk.start()
return mdk, tracer
def create_mdk_with_faketracer(environment="sandbox"):
"""Create an MDK with a FakeTracer.
Returns (mdk, fake_tracer).
"""
runtime = fakeRuntime()
tracer = FakeTracer()
runtime.dependencies.registerService("tracer", tracer)
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE", "static:nodes={}")
runtime.getEnvVarsService().set("MDK_ENVIRONMENT", environment)
mdk = MDKImpl(runtime)
mdk.start()
return mdk, tracer
def __init__(self, failurepolicy_factory=None, env={}, start=True):
self.runtime = fakeRuntime()
env_vars = self.runtime.getEnvVarsService()
for key, value in env.items():
env_vars.set(key, value)
env_vars.set("DATAWIRE_TOKEN", "xxx")
env_vars.set("MDK_SERVER_URL", self.URL)
if failurepolicy_factory is not None:
self.runtime.dependencies.registerService("failurepolicy_factory",
failurepolicy_factory)
self.mdk = MDKImpl(self.runtime)
if start:
self.mdk.start()
self.pump()
def __init__(self, failurepolicy_factory=None, env={}, start=True):
self.runtime = fakeRuntime()
env_vars = self.runtime.getEnvVarsService()
for key, value in env.items():
env_vars.set(key, value)
env_vars.set("DATAWIRE_TOKEN", "xxx");
env_vars.set("MDK_SERVER_URL", self.URL);
if failurepolicy_factory is not None:
self.runtime.dependencies.registerService(
"failurepolicy_factory", failurepolicy_factory)
self.mdk = MDKImpl(self.runtime)
if start:
self.mdk.start()
self.pump()
def test_no_tracer(self):
"""
If no tracer was setup, logging still returns a LoggedMessageId.
"""
runtime = fakeRuntime()
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE", "static:nodes={}")
mdk = MDKImpl(runtime)
mdk.start()
# No DATAWIRE_TOKEN, so no tracer:
self.assertEqual(mdk._tracer, None)
session = mdk.session()
session.info("cat", "message")
lmid = session.info("cat", "another message")
self.assertEqual((lmid.traceId, lmid.causalLevel),
(session._context.traceId, [2]))
def test_no_tracer(self):
"""
If no tracer was setup, logging still returns a LoggedMessageId.
"""
runtime = fakeRuntime()
runtime.getEnvVarsService().set("MDK_DISCOVERY_SOURCE",
"static:nodes={}")
mdk = MDKImpl(runtime)
mdk.start()
# No DATAWIRE_TOKEN, so no tracer:
self.assertEqual(mdk._tracer, None)
session = mdk.session()
session.info("cat", "message")
lmid = session.info("cat", "another message")
self.assertEqual((lmid.traceId, lmid.causalLevel),
(session._context.traceId, [2]))