def test_resource_can_be_highlighted(self):
# GIVEN the following mesh
#
# _____________ _____________
# | A | | B |
# |-------------| |-------------|
# | | n1 |-------->| p1 | n2 |--------->[ Resource X ]
# |______|______| |______|______|
#
#
m = Mesh()
m.add_component('A', needs_ports=['n1'])
m.add_component('B', needs_ports=['n2'], provides_ports=['p1'])
m.add_resource('Resource X')
m.add_connection('A', 'n1', 'B', 'p1')
m.add_connection_to_resource('B', 'n2', 'Resource X')
# WHEN Resource X is highlighted
m.highlight_resource('Resource X')
# THEN this is reflected in the dict representation
self.assertEqual({
'components': [
{'name': 'A', 'needs_ports': ['n1'], 'provides_ports': []},
{'name': 'B', 'needs_ports': ['n2'], 'provides_ports': ['p1']}
],
'resources': ['Resource X'],
'highlighted_resources': ['Resource X'],
'connections': [
{
'consumer_component': 'A',
'consumer_port': 'n1',
'producer_component': 'B',
'producer_port': 'p1',
},
{
'consumer_component': 'B',
'consumer_port': 'n2',
'resource': 'Resource X',
},
],
}, m.as_dict())
def test_InvalidResource_exception_raised_when_highlighting_unknown_resource(self):
# GIVEN a mesh with the following component and resource
#
# _____________ _____________
# | A | | B |
# |-------------| |-------------|
# | | n1 |-------->| p1 | n2 |--------->[ Resource X ]
# |______|______| |______|______|
#
#
m = Mesh()
m.add_component('A', needs_ports=['n1'])
m.add_component('B', needs_ports=['n2'], provides_ports=['p1'])
m.add_resource('Resource X')
m.add_connection('A', 'n1', 'B', 'p1')
m.add_connection_to_resource('B', 'n2', 'Resource X')
# WHEN unknown resource is highlighted
# THEN an InvalidResource exception is raised
self.assertRaises(InvalidResource, m.highlight_resource, 'Resource K')
def test_DuplicateEntry_exception_raised_when_adding_connection_to_port_already_connected_to_resource(self):
# GIVEN the following mesh
#
# _____________
# | A |
# |-------------|
# | | n1 |-------->[ Resource X ]
# |______|______|
#
# _____________
# | D |
# |-------------|
# | pX | |
# |______|______|
#
#
m = Mesh()
m.add_component('A', needs_ports=['n1'])
m.add_component('D', provides_ports=['pX'])
m.add_resource('Resource X')
m.add_connection_to_resource('A', 'n1', 'Resource X')
# WHEN attempting to create the following invalid connections
# _____________
# | A |
# |-------------|
# | | n1 |----+----->[ Resource X ]
# |______|______| |
# |
# INVALID _____________
# | | D |
# | |-------------|
# +----->| pX | |
# |______|______|
#
# THEN an InvalidConnection exception is raised
self.assertRaises(InvalidConnection, m.add_connection, 'A', 'n1', 'D', 'pX')
def test_adding_connection_from_port_to_edge_resource(self):
# GIVEN a mesh with the following component and resource
#
# _____________
# | A |
# |-------------|
# | | n1 | [ Resource X ]
# |______|______|
#
m = Mesh()
m.add_component('A', needs_ports=['n1'])
m.add_resource('Resource X')
# WHEN we connect A:n1 to Resource X
m.add_connection_to_resource('A', 'n1', 'Resource X')
# THEN we get the following mesh representation
#
# _____________
# | A |
# |-------------|
# | | n1 |--------->[ Resource X ]
# |______|______|
#
self.assertEqual({
'components': [
{'name': 'A', 'needs_ports': ['n1'], 'provides_ports': []},
],
'resources': ['Resource X'],
'connections': [
{
'consumer_component': 'A',
'consumer_port': 'n1',
'resource': 'Resource X',
},
],
}, m.as_dict())
def test_textual_representation_can_be_rendered_for_a_populated_mesh(self):
# GIVEN the following mesh
#
# _____________ _____________
# | A | | B |
# |-------------| |-------------|
# | | n1 |----------->| p1 | |
# | |______| |______| nY |----->[ Resource Y ]
# | | n2 |-----+ +-->| p2 | |
# |______|______| | | |______|______|
# +- | -+
# | |
# _____________ | | _____________
# | C | | | | D |
# |-------------| | | |-------------|
# | | nX |--+ +---->| pX | nX |------>[ Resource X ]
# |______|______| |______|______|
#
m = Mesh()
m.add_component('A', needs_ports=['n1', 'n2'])
m.add_component('B', provides_ports=['p1', 'p2'], needs_ports=['nY'])
m.add_component('C', needs_ports=['nX'])
m.add_component('D', provides_ports=['pX'], needs_ports=['nX'])
m.add_resource('Resource X')
m.add_resource('Resource Y')
m.add_connection('A', 'n1', 'B', 'p1')
m.add_connection('A', 'n2', 'D', 'pX')
m.add_connection('C', 'nX', 'B', 'p2')
m.add_connection_to_resource('D', 'nX', 'Resource X')
m.add_connection_to_resource('B', 'nY', 'Resource Y')
# WHEN the textual representation is rendered
out = render(m, self.JSON_TEMPLATE)
# THEN the rendered output reflects that of the mesh
out_data = json.loads(out)
self.assertEqual({
'components': [
{'name': 'A', 'needs_ports': ['n1', 'n2'], 'provides_ports': []},
{'name': 'B', 'needs_ports': ['nY'], 'provides_ports': ['p1', 'p2']},
{'name': 'C', 'needs_ports': ['nX'], 'provides_ports': []},
{'name': 'D', 'needs_ports': ['nX'], 'provides_ports': ['pX']},
],
'resources': ['Resource X', 'Resource Y'],
'connections': [
{
'consumer_component': 'A',
'consumer_port': 'n1',
'producer_component': 'B',
'producer_port': 'p1',
},
{
'consumer_component': 'A',
'consumer_port': 'n2',
'producer_component': 'D',
'producer_port': 'pX',
},
{
'consumer_component': 'C',
'consumer_port': 'nX',
'producer_component': 'B',
'producer_port': 'p2',
},
{
'consumer_component': 'D',
'consumer_port': 'nX',
'resource': 'Resource X',
},
{
'consumer_component': 'B',
'consumer_port': 'nY',
'resource': 'Resource Y',
},
],
}, out_data)
