def test_single_component_registered(self):
test_component = "foo"
UserAgentManager.register_component(test_component)
self.assertEqual(
test_component, UserAgentManager.get_user_agent(),
"UserAgent Manager didn't register single component to "
"user agent")
def test_should_append_additional_user_agent_using_userAgentManager(self):
additional_user_agent = "test_string"
sdk_user_agent = user_agent_info(sdk_version=__version__)
mock_request_handler = mock.MagicMock(spec=AbstractRequestHandler)
mock_request_handler.can_handle.return_value = True
mock_response = Response()
mock_request_handler.handle.return_value = mock_response
self.sb.add_request_handler(request_handler=mock_request_handler)
mock_request_envelope_payload = {
"context": {
"System": {
"application": {
"applicationId": "test"
}
}
}
}
self.sb.skill_id = "test"
lambda_handler = self.sb.lambda_handler()
UserAgentManager.register_component(
component_name=additional_user_agent)
response_envelope = lambda_handler(event=mock_request_envelope_payload,
context=None)
self.assertEqual(
"{} {}".format(additional_user_agent,
sdk_user_agent), response_envelope["userAgent"],
"Response envelope doesn't have correct user agent when adding "
"using UserAgentManager")
def __init__(self, skill_configuration):
# type: (SkillConfiguration) -> None
"""Top level container for Request Dispatcher,
Persistence Adapter and Api Client.
:param skill_configuration: Configuration object that holds
information about different components needed to build the
skill object.
:type skill_configuration: SkillConfiguration
"""
self.persistence_adapter = skill_configuration.persistence_adapter
self.api_client = skill_configuration.api_client
self.serializer = DefaultSerializer()
self.skill_id = skill_configuration.skill_id
self.custom_user_agent = skill_configuration.custom_user_agent
self.loaders = skill_configuration.loaders
self.renderer = skill_configuration.renderer
self.request_dispatcher = GenericRequestDispatcher(
options=skill_configuration)
UserAgentManager.register_component(
user_agent_info(sdk_version=__version__))
if skill_configuration.custom_user_agent is not None:
UserAgentManager.register_component(
component_name=skill_configuration.custom_user_agent)
def test_multi_component_registered(self):
test_component_1 = "foo"
test_component_2 = "bar"
UserAgentManager.register_component(test_component_1)
UserAgentManager.register_component(test_component_2)
self.assertEqual(
'{} {}'.format(test_component_1, test_component_2),
UserAgentManager.get_user_agent(),
"UserAgent Manager didn't register single component to "
"user agent")
def add_custom_user_agent(self, user_agent):
# type: (str) -> None
"""Adds the user agent to the skill instance.
This method adds the passed in user_agent to the skill, which is
reflected in the skill's response envelope.
:param user_agent: Custom User Agent string provided by the developer.
:type user_agent: str
:rtype: None
"""
UserAgentManager.register_component(component_name=user_agent)
def test_components_user_agent_cleared(self):
UserAgentManager.register_component('foo')
UserAgentManager.register_component('bar')
UserAgentManager.clear()
self.assertEqual('', UserAgentManager.get_user_agent(),
"UserAgent Manager didn't clear user agent")