[Test]: Improve the test coverage

[zhongkechen]

What would you like?

Improve the test coverage

Possible Implementation

No response

Is this a breaking change?

No

Does this require an RFC?

No

Additional Context

No response