def test_modifying_correlations(self): ctx = cctx.set_correlation("test", "value") self.assertEqual(cctx.get_correlation("test", context=ctx), "value") correlations = cctx.get_correlations(context=ctx) with self.assertRaises(TypeError): correlations["test"] = "mess-this-up" self.assertEqual(cctx.get_correlation("test", context=ctx), "value")
def test_remove_correlations(self): self.assertEqual({}, cctx.get_correlations()) ctx = cctx.set_correlation("test", "value") ctx = cctx.set_correlation("test2", "value2", context=ctx) ctx = cctx.remove_correlation("test", context=ctx) self.assertEqual(cctx.get_correlation("test", context=ctx), None) self.assertEqual(cctx.get_correlation("test2", context=ctx), "value2")
def test_set_multiple_correlations(self): ctx = cctx.set_correlation("test", "value") ctx = cctx.set_correlation("test2", "value2", context=ctx) self.assertEqual(cctx.get_correlation("test", context=ctx), "value") self.assertEqual(cctx.get_correlation("test2", context=ctx), "value2") self.assertEqual( cctx.get_correlations(context=ctx), {"test": "value", "test2": "value2"}, )
def test_clear_correlations(self): self.assertEqual({}, cctx.get_correlations()) ctx = cctx.set_correlation("test", "value") self.assertEqual(cctx.get_correlation("test", context=ctx), "value") ctx = cctx.clear_correlations(context=ctx) self.assertEqual(cctx.get_correlations(context=ctx), {})
def get_baggage_item(self, key: str) -> Optional[object]: """Retrieves value of the baggage item with the given key. Args: key: A tag key. Returns: Returns this :class:`SpanShim` instance to allow call chaining. """ # pylint: disable=protected-access return get_correlation(key, context=self._context._baggage)
def test_correlations_current_context(self): token = context.attach(cctx.set_correlation("test", "value")) self.assertEqual(cctx.get_correlation("test"), "value") context.detach(token) self.assertEqual(cctx.get_correlation("test"), None)
def get_baggage_item(self, key): """Implements the ``get_baggage_item`` method from the base class.""" # pylint: disable=protected-access return get_correlation(key, context=self._context._baggage)