Update sdk-tweak-model.ts (#1317)

* Update sdk-tweak-model.ts For head API, the response should be 204 * Update sdk-tweak-model.ts
Azure · Mar 11, 2024 · c33b901 · c33b901
1 parent ba17668
commit c33b901
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/powershell/plugins/sdk-tweak-model.ts b/powershell/plugins/sdk-tweak-model.ts
@@ -324,7 +324,8 @@ async function tweakOperation(state: State) {
         respCountWithBody = schemas.size + specialBinaryResponse;
         const isHead = operation.requests && operation.requests[0].protocol.http?.method === 'head';
         if (isHead) {
-          initializeResponseBody = '_result.Body = (_statusCode == System.Net.HttpStatusCode.OK);';
+          const succeedCode = operation.responses.filter(r => (<string>r.protocol.http?.statusCodes[0]).startsWith('2'))[0].protocol.http?.statusCodes[0];
+          initializeResponseBody = `_result.Body = (_statusCode == System.Net.HttpStatusCode.${(<any>StatusCodes)[succeedCode]});`;
         }
         const responses = operation.responses.filter(r => (<any>r).schema);
         const hasHeaderResponse = operation.responses.some(r => (<any>r).protocol.http.headers);
@@ -469,4 +470,4 @@ export async function tweakSdkModelPlugin(service: Host) {
     }
     throw E;
   }
-}
+}