public async Task ShouldRetry_Handles_Retry_Cases(string fileName, bool enableEnhancedPreparedStatements, bool shouldRetry) { // Arrange using var stream = ResourceHelper.ReadResourceAsStream(fileName); using var streamingResult = new StreamingQueryResult <dynamic>(stream, new DefaultSerializer(), ErrorContextFactory); await streamingResult.InitializeAsync().ConfigureAwait(false); // Act var actual = streamingResult.ShouldRetry(enableEnhancedPreparedStatements); // Assert Assert.Equal(shouldRetry, actual); }