def test_delattr(self):
ctx = Context('foo')
ctx_manager = ContextManager([ctx])
del ctx_manager.foo
assert 'foo' not in ctx_manager
with pytest.raises(AttributeError):
del ctx_manager.bar
def test_collection_iter(self):
ctx = Context('foo')
ctx_manager = ContextManager([ctx])
ctx_iter = iter(ctx_manager)
assert isinstance(ctx_iter, Iterator)
assert next(ctx_iter) == ctx
with pytest.raises(StopIteration):
next(ctx_iter)
def test_delete_keep_in_registry(self):
ctx_manager = ContextManager([Context('foo')])
ctx_manager._context_registry.register('foo', keep_around=True)
assert ctx_manager._context_registry.should_keep_around('foo')
ctx_manager.delete('foo', keep_in_registry=True)
assert 'foo' not in ctx_manager
assert 'foo' in ctx_manager._marked_for_deletion
assert ctx_manager._context_registry.should_keep_around('foo')
def test_delete_dont_keep_in_registry(self):
ctx_manager = ContextManager([Context('foo')])
ctx_manager._context_registry.register('foo', keep_around=True)
assert ctx_manager._context_registry.should_keep_around('foo')
ctx_manager.delete('foo', keep_in_registry=False)
assert 'foo' not in ctx_manager
assert 'foo' in ctx_manager._marked_for_deletion
with pytest.raises(ContextNotRegisteredError):
assert ctx_manager._context_registry.should_keep_around('foo')
def test_as_list_with_deleted_ctx(self):
contexts = [
Context('foo', lifespan_count=5),
Context('bar', lifespan_count=5)
]
ctx_manager = ContextManager(contexts)
ctx_manager.delete('bar')
as_list = ctx_manager.as_list()
assert len(as_list) == 2
assert as_list[0].lifespan_count == 5
assert as_list[1].lifespan_count == 0
def __init__(
self,
webhook_request: Optional[_df.WebhookRequest] = None,
context_manager: Optional['ContextManager'] = None,
integration_convs: Optional[Mapping[
str, AbstractIntegrationConversation]] = None,
) -> None:
"""Initialize a conversation.
This is not supposed to be done by the user, several steps usually
preceed the initialization to set the conversation up correctly.
Args:
webhook_request: The :class:`.WebhookRequest` that this
conversation represents.
context_manager: The :class:`.ContextManager` that handles the
contexts.
integration_convs: The mapping of integration conversations.
"""
if webhook_request is None:
odir = self._df.OriginalDetectIntentRequest()
webhook_request = self._df.WebhookRequest(
query_result=self._df.QueryResult(intent=self._df.Intent()),
original_detect_intent_request=odir,
)
self._webhook_request = webhook_request
self._contexts = context_manager or ContextManager(contexts=[
Context('_session_context', parameters=SessionContext())
])
self._integration_convs: DefaultDict[
str, AbstractIntegrationConversation] = defaultdict(
GenericIntegrationConversation)
if integration_convs:
self._integration_convs.update(integration_convs)
self._session_ctx = self.contexts.get('_session_context').parameters
self._webhook_response = self._df.WebhookResponse()
if self.webhook_request.query_result.intent.is_fallback:
self._session_ctx.fallback_level += 1
else:
self._session_ctx.fallback_level = 0
def _initialize_conversation(
self, webhook_request: WebhookRequest) -> DialogflowConversation:
"""Initialize the conversation object from a WebhookRequest."""
contexts = [
Context.from_context(ctx)
for ctx in webhook_request.query_result.output_contexts
]
self._deserialize_context_params(contexts)
self._initialize_default_contexts(contexts, webhook_request.session)
self._reset_lifespan_of_keep_around_contexts(contexts)
context_manager = ContextManager(
contexts=contexts,
session=webhook_request.session,
context_registry=self._context_registry)
integration_convs = self._integration_registry.init_integration_convs(
webhook_request.original_detect_intent_request)
return self._conv_cls(
webhook_request,
context_manager=context_manager,
integration_convs=integration_convs,
)
def test_has(self):
ctx_manager = ContextManager([Context('foo')])
assert ctx_manager.has('foo')
assert not ctx_manager.has('bar')
def test_as_list(self):
contexts = [Context('foo'), Context('bar')]
ctx_manager = ContextManager(contexts)
assert ctx_manager.as_list() == contexts
def test_delattr_to_ctx_param(self):
ctx = Context('foo', parameters={'bar': 'baz'})
ctx_manager = ContextManager([ctx])
del ctx_manager.foo['bar']
assert ctx.parameters == {}
def ctx_manager(self):
return ContextManager(session='foo/bar')
def test_getattr_to_ctx_param(self):
ctx = Context('foo', parameters={'bar': 'baz'})
ctx_manager = ContextManager([ctx])
assert ctx_manager.foo['bar'] == 'baz'
def test_getattr(self):
ctx = Context('foo')
ctx_manager = ContextManager([ctx])
assert ctx_manager.foo == ctx
with pytest.raises(AttributeError):
_ = ctx_manager.bar
def test_init(self):
assert isinstance(ContextManager([Context()]), ContextManager)
def test_collection_len(self):
assert len(ContextManager()) == 0
def test_collection_contains(self):
ctx_manager = ContextManager([Context('foo')])
assert 'foo' in ctx_manager
assert 'bar' not in ctx_manager
def test_in_operator(self):
ctx_manager = ContextManager([Context('foo')])
assert 'foo' in ctx_manager
assert 'bar ' not in ctx_manager
def test_delete_as_attribute(self):
ctx_manager = ContextManager([Context('foo')])
del ctx_manager.foo
assert 'foo' not in ctx_manager
assert 'foo' in ctx_manager._marked_for_deletion
def test_emtpy_init(self):
assert isinstance(ContextManager(), ContextManager)
def test_get(self):
ctx = Context('foo')
ctx_manager = ContextManager([ctx])
assert ctx_manager.get('foo') == ctx
def test_delete(self):
ctx_manager = ContextManager([Context('foo')])
ctx_manager.delete('foo')
assert 'foo' not in ctx_manager
assert 'foo' in ctx_manager._marked_for_deletion
def test_get_as_attribute(self):
ctx = Context('foo')
ctx_manager = ContextManager([ctx])
assert ctx_manager.foo == ctx
