Updated unit tests for TestActorRef framework