def test_link_form_cast(self):
# Given
form = _LinkRequestForm()
items = RecordMap.create()
items.add(
Slot.create_slot(Text.create_from('node'),
Text.create_from('link_node')))
items.add(
Slot.create_slot(Text.create_from('lane'),
Text.create_from('link_lane')))
items.add(
Slot.create_slot(Text.create_from('prio'), Num.create_from(1)))
items.add(
Slot.create_slot(Text.create_from('rate'), Num.create_from(3)))
body = Attr.create_attr(Text.create_from('body'),
Text.create_from('link_body'))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('link'), items))
record_map.add(body)
# When
actual = form._cast(record_map)
# Then
self.assertIsInstance(actual, _LinkRequest)
self.assertEqual('link_node', actual._node_uri)
self.assertEqual('link_lane', actual._lane_uri)
self.assertEqual(1, actual._prio)
self.assertEqual(3, actual._rate)
self.assertEqual(body, actual._body.get_item(0))
def test_unlinked_form_cast(self):
# Given
form = _UnlinkedResponseForm()
items = RecordMap.create()
items.add(
Slot.create_slot(Text.create_from('node'),
Text.create_from('unlinked_node')))
items.add(
Slot.create_slot(Text.create_from('lane'),
Text.create_from('unlinked_lane')))
items.add(
Slot.create_slot(Text.create_from('prio'), Num.create_from(20)))
items.add(
Slot.create_slot(Text.create_from('rate'), Num.create_from(25)))
body = Attr.create_attr(Text.create_from('body'),
Text.create_from('unlinked_body'))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('unlinked'), items))
record_map.add(body)
# When
actual = form._cast(record_map)
# Then
self.assertIsInstance(actual, _UnlinkedResponse)
self.assertEqual('unlinked_node', actual._node_uri)
self.assertEqual('unlinked_lane', actual._lane_uri)
self.assertEqual(20, actual._prio)
self.assertEqual(25, actual._rate)
self.assertEqual(body, actual._body.get_item(0))
def test_link_addressed_cast_missing_headers(self):
# Given
form = SyncRequestForm()
items = RecordMap.create()
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('sync'), items))
# When
actual = form.cast(record_map)
# Then
self.assertEqual(None, actual)
def test_lane_addressed_form_cast_missing_headers(self):
# Given
form = _SyncedResponseForm()
items = RecordMap.create()
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('synced'), items))
# When
actual = form._cast(record_map)
# Then
self.assertEqual(None, actual)
def test_link_addressed_form_cast_missing_lane(self):
# Given
form = _SyncRequestForm()
items = RecordMap.create()
items.add(
Attr.create_attr(Text.create_from('node'),
Text.create_from('sync_foo_node')))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('sync'), items))
# When
actual = form._cast(record_map)
# Then
self.assertEqual(None, actual)
async def test_downlink_model_receive_message_unlinked(self, mock_warn):
# Given
with SwimClient(execute_on_exception=MockExecuteOnException.
get_mock_execute_on_exception()) as client:
downlink = EventDownlinkModel(client)
downlink.connection = MockConnection.get_mock_connection()
downlink.connection.owner = downlink
body = RecordMap.create()
body.add(Attr.create_attr('laneNotFound', 'foo'))
unlinked_message = UnlinkedResponse('unlinked_node',
'unlinked_lane',
body=body)
downlink.connection.messages_to_receive.append(unlinked_message)
# When
actual = downlink.open()
while not MockExecuteOnException.get_mock_execute_on_exception(
).called:
pass
# Then
self.assertEqual(downlink, actual)
self.assertEqual('Lane "None" was not found on the remote agent!',
mock_warn.call_args_list[0][0][0])
def get_key_item(self) -> 'Record':
key_slot = RecordMap.create()
key_slot.add(
Slot.create_slot(
Text.create_from('key'),
RecordConverter.get_converter().object_to_record(self.key)))
return key_slot
def test_event_form_cast(self):
# Given
form = EventMessageForm()
items = RecordMap.create()
items.add(Slot.create_slot(Text.create_from('node'), Text.create_from('event_node')))
items.add(Slot.create_slot(Text.create_from('lane'), Text.create_from('event_lane')))
body = Attr.create_attr(Text.create_from('body'), Text.create_from('event_body'))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('event'), items))
record_map.add(body)
# When
actual = form.cast(record_map)
# Then
self.assertIsInstance(actual, EventMessage)
self.assertEqual('event_node', actual.node_uri)
self.assertEqual('event_lane', actual.lane_uri)
self.assertEqual(body, actual.body.get_item(0))
def test_sync_form_cast(self):
# Given
form = SyncRequestForm()
items = RecordMap.create()
items.add(Slot.create_slot(Text.create_from('node'), Text.create_from('sync_node')))
items.add(Slot.create_slot(Text.create_from('lane'), Text.create_from('sync_lane')))
items.add(Slot.create_slot(Text.create_from('prio'), Num.create_from(33.12)))
items.add(Slot.create_slot(Text.create_from('rate'), Num.create_from(12.33)))
body = Attr.create_attr(Text.create_from('body'), Text.create_from('sync_body'))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('sync'), items))
record_map.add(body)
# When
actual = form.cast(record_map)
# Then
self.assertIsInstance(actual, SyncRequest)
self.assertEqual('sync_node', actual.node_uri)
self.assertEqual('sync_lane', actual.lane_uri)
self.assertEqual(33.12, actual.prio)
self.assertEqual(12.33, actual.rate)
self.assertEqual(body, actual.body.get_item(0))
def test_command_form_cast(self):
# Given
form = _CommandMessageForm()
items = RecordMap.create()
items.add(
Slot.create_slot(Text.create_from('node'),
Text.create_from('command_node')))
items.add(
Slot.create_slot(Text.create_from('lane'),
Text.create_from('command_lane')))
body = Attr.create_attr(Text.create_from('body'),
Text.create_from('command_body'))
record_map = RecordMap.create()
record_map.add(Attr.create_attr(Text.create_from('command'), items))
record_map.add(body)
# When
actual = form._cast(record_map)
# Then
self.assertIsInstance(actual, _CommandMessage)
self.assertEqual('command_node', actual._node_uri)
self.assertEqual('command_lane', actual._lane_uri)
self.assertEqual(body, actual._body.get_item(0))
def test_to_string(self):
# Given
value = RecordMap.create()
value.add(
Attr.create_attr(
Text.create_from('remove'),
RecordMap.create_record_map(
Slot.create_slot(Text.create_from('key'),
Text.create_from('foo')))))
# When
actual = Recon.to_string(value)
# Then
self.assertEqual('@remove(key:foo)', actual)
def get_key_item(self) -> '_Record':
"""
Convert the request key into an Item object.
:return: - Request key as an Item object.
"""
key_slot = RecordMap.create()
key_slot.add(
Slot.create_slot(
Text.create_from('key'),
RecordConverter.get_converter().object_to_record(self.key)))
return key_slot
def test_write_event_body_nested(self):
# Given
friend = RecordMap.create()
friend.add(
Attr.create_attr(Text.create_from('Person'),
_Extant._get_extant()))
friend.add(
Slot.create_slot(Text.create_from('name'),
Text.create_from('Sam/Spam')))
friend.add(
Slot.create_slot(Text.create_from('age'), Num.create_from(1)))
friend.add(
Slot.create_slot(Text.create_from('salary'), Num.create_from(22)))
body = RecordMap.create()
body.add(
Attr.create_attr(Text.create_from('Person'),
_Extant._get_extant()))
body.add(
Slot.create_slot(Text.create_from('name'),
Text.create_from('Par')))
body.add(Slot.create_slot(Text.create_from('age'),
Num.create_from(11)))
body.add(
Slot.create_slot(Text.create_from('salary'),
Num.create_from(-5.9)))
body.add(Slot.create_slot(Text.create_from('friend'), friend))
envelope = _EventMessage('/this/is/spam', 'hello', body=body)
person = '@Person{name:Par,age:11,salary:-5.9,friend:@Person{name:"Sam/Spam",age:1,salary:22}}'
expected = '@event(node:"/this/is/spam",lane:hello)' + person
# When
actual = envelope._to_recon()
# Then
self.assertEqual(expected, actual)
async def test_attr_expression_parser_parse_literal_existing_value(self):
# Given
message = await InputMessage.create('literal')
parser = ReconParser()
builder = RecordMap.create()
builder.add(Text.create_from('Moo'))
value = Text.create_from('Dog')
# When
actual = await parser.parse_attr_expression(message, builder=builder, value_output=value)
# Then
self.assertIsInstance(actual, RecordMap)
self.assertEqual(3, actual.size)
self.assertEqual('Moo', actual.get_item(0).value)
self.assertEqual('Dog', actual.get_item(1).value)
self.assertEqual('literal', actual.get_item(2).value)
async def test_attr_expression_parser_parse_attribute_existing_field(self):
# Given
message = await InputMessage.create('@test')
parser = ReconParser()
builder = RecordMap.create()
builder.add(Text.create_from('Moo'))
field = Text.create_from('Boo')
# When
actual = await parser.parse_attr_expression(message, builder=builder, field_output=field)
# Then
self.assertIsInstance(actual, RecordMap)
self.assertEqual(3, actual.size)
self.assertEqual('Moo', actual.get_item(0).value)
self.assertEqual('Boo', actual.get_item(1).value)
self.assertEqual('test', actual.get_item(2).key.value)
async def test_write_event_body_object(self):
# Given
body = RecordMap.create()
body.add(
Attr.create_attr(Text.create_from('Person'), Extant.get_extant()))
body.add(
Slot.create_slot(Text.create_from('name'),
Text.create_from('Bar')))
body.add(Slot.create_slot(Text.create_from('age'),
Num.create_from(14)))
body.add(
Slot.create_slot(Text.create_from('salary'),
Num.create_from(-5.9)))
envelope = EventMessage('/this/is/spam', 'hello', body=body)
expected = '@event(node:"/this/is/spam",lane:hello)@Person{name:Bar,age:14,salary:-5.9}'
# When
actual = await envelope.to_recon()
# Then
self.assertEqual(expected, actual)
