Add missing bits to TestKit

Contains AwaitAssert, ExpectMsgAllOf, ReceiveN and FishForMessages.
Also contains a fix for ReceiveWhile so that it puts the last message back into the queue