Add tests for _assert_count_message