Update TestCustomProtocolSimulation.scala

Added some assertions