Fix test cases

Author: takezoe

spec/statement_client_spec.rb

@@ -540,12 +540,22 @@
         client.advance
 
         sleep 1
+
         stub_request(:get, "localhost/v1/next_uri").
           with(headers: headers).
           to_return(body: planning_response.to_json)
+
+        cancelled = false
+        stub_request(:delete, "localhost/v1/next_uri").
+          with(headers: headers).
+          to_return(status: lambda { |req| cancelled = true; 204 }) # NoContent
+
         expect do
           client.advance
         end.to raise_error(Trino::Client::TrinoQueryTimeoutError, "Query queryid timed out")
+
+        expect(cancelled).to eq true
+        expect(client.client_error?).to eq true
       end
 
       it "raises TrinoQueryTimeoutError if timeout during initial resuming" do
@@ -602,12 +612,22 @@
       client.advance
 
       sleep 1
+
       stub_request(:get, "localhost/v1/next_uri").
         with(headers: headers).
         to_return(body: late_running_response.to_json)
+
+      cancelled = false
+      stub_request(:delete, "localhost/v1/next_uri").
+        with(headers: headers).
+        to_return(status: lambda { |req| cancelled = true; 204 }) # NoContent
+
       expect do
         client.advance
       end.to raise_error(Trino::Client::TrinoQueryTimeoutError, "Query queryid timed out")
+
+      expect(cancelled).to eq true
+      expect(client.client_error?).to eq true
     end
 
     it "doesn't raise errors if query is done" do