Adding a test for message flow diagram
Fixing an issue with test data, resetting the layout before each test