Use a test behavior instead of a production one

Rather than depending on a real behavior (CloseImmediately), we create a simple behavior for testing and use it here.