[fix][test] Fix query server unit tests to check for status code (#6241)

[fix][test] Fix query server unit tests to check for status code (#6241) #1787