Fix failing tests

microsoft · Feb 16, 2022 · d5d3c02 · d5d3c02
1 parent a7cf475
commit d5d3c02
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/Kiota.Builder.Tests/Writers/Shell/ShellCodeMethodWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/Shell/ShellCodeMethodWriterTests.cs
@@ -254,7 +254,7 @@ public void WritesRootCommandWithCommandBuilderMethods()
         Assert.Contains("command.AddOption(qOption);", result);
         Assert.Contains("command.AddOption(outputOption);", result);
         Assert.Contains("var requestInfo = CreateGetRequestInformation", result);
-        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo);", result);
+        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo, errorMapping: default, cancellationToken: cancellationToken);", result);
         Assert.Contains("return command;", result);
     }
 
@@ -301,7 +301,7 @@ public void WritesRootCommandWithCommandBuilderMethods()
         Assert.Contains("command.AddOption(bodyOption);", result);
         Assert.Contains("command.AddOption(outputOption);", result);
         Assert.Contains("var requestInfo = CreatePostRequestInformation", result);
-        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo);", result);
+        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo, errorMapping: default, cancellationToken: cancellationToken);", result);
         Assert.Contains("return command;", result);
     }
 
@@ -347,7 +347,7 @@ public void WritesExecutableCommandForGetStreamRequest()
         Assert.Contains("var fileOption = new Option<FileInfo>(\"--file\");", result);
         Assert.Contains("command.AddOption(fileOption);", result);
         Assert.Contains("var requestInfo = CreateGetRequestInformation", result);
-        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo);", result);
+        Assert.Contains("var response = await RequestAdapter.SendPrimitiveAsync<Stream>(requestInfo, errorMapping: default, cancellationToken: cancellationToken);", result);
         Assert.Contains("return command;", result);
     }
 
@@ -396,7 +396,7 @@ public void WritesExecutableCommandForGetStreamRequest()
         Assert.Contains("bodyOption.IsRequired = true;", result);
         Assert.Contains("command.AddOption(bodyOption);", result);
         Assert.Contains("var requestInfo = CreatePostRequestInformation", result);
-        Assert.Contains("await RequestAdapter.SendNoContentAsync(requestInfo);", result);
+        Assert.Contains("await RequestAdapter.SendNoContentAsync(requestInfo, errorMapping: default, cancellationToken: cancellationToken);", result);
         Assert.Contains("console.WriteLine(\"Success\");", result);
         Assert.Contains("return command;", result);
     }