public void WritesAfterMessagesAreSent() { // Unlike most other tests, here we write a message, the server completes the // task, write another message, server completes etc. var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 5); // First message var task1 = writer.WriteAsync("1"); fake.CompleteCurrentTask(); AssertCompletedWithStatus(task1, TaskStatus.RanToCompletion); // Second message var task2 = writer.WriteAsync("2"); fake.CompleteCurrentTask(); AssertCompletedWithStatus(task2, TaskStatus.RanToCompletion); // Completion var completionTask = writer.WriteCompleteAsync(); fake.CompleteCurrentTask(); AssertCompletedWithStatus(completionTask, TaskStatus.RanToCompletion); fake.AssertMessages("1", "2"); fake.AssertCompleted(); }
public void CompleteAfterMessages() { var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 5); var task1 = writer.WriteAsync("1"); var task2 = writer.WriteAsync("2"); var completionTask = writer.WriteCompleteAsync(); AssertNotCompleted(task1, task2, completionTask); // Server handles first message fake.CompleteCurrentTask(); AssertCompletedWithStatus(task1, TaskStatus.RanToCompletion); AssertNotCompleted(task2, completionTask); // Server handles second message fake.CompleteCurrentTask(); AssertCompletedWithStatus(task2, TaskStatus.RanToCompletion); AssertNotCompleted(completionTask); // Server handles completion fake.CompleteCurrentTask(); AssertCompletedWithStatus(completionTask, TaskStatus.RanToCompletion); fake.AssertMessages("1", "2"); fake.AssertCompleted(); }
public void CompleteWithNoWrites() { var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 5); var completionTask = writer.WriteCompleteAsync(); fake.CompleteCurrentTask(); AssertCompletedWithStatus(completionTask, TaskStatus.RanToCompletion); fake.AssertMessages(); fake.AssertCompleted(); }
public void CompletionUsesBufferSpace() { var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 2); writer.WriteAsync("1"); writer.WriteAsync("2"); // Try completion, should throw because queue is full. // The (object) cast is because to make xUnit understand that the call itself should throw; // we don't return a failed task. Assert.Throws <InvalidOperationException>(() => (object)writer.WriteCompleteAsync()); // Try completion, should return null because queue is full. Assert.Null(writer.TryWriteCompleteAsync()); fake.CompleteCurrentTask(); fake.CompleteCurrentTask(); writer.WriteCompleteAsync(); fake.CompleteCurrentTask(); fake.AssertCompleted(); }
public void TryWriteBeyondBuffer() { var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 2); var task1 = writer.WriteAsync("1"); var task2 = writer.WriteAsync("2"); var task3 = writer.TryWriteAsync("3"); Assert.Null(task3); // Couldn't write. fake.CompleteCurrentTask(); // Message 1 WaitForSpace(writer); // Now the buffer is smaller, we can write again. var task4 = writer.TryWriteAsync("4"); Assert.NotNull(task4); // Try to write completion, will fail as buffer is full. var completionTask1 = writer.TryWriteCompleteAsync(); Assert.Null(completionTask1); fake.CompleteCurrentTask(); // Message 2 WaitForSpace(writer); // Now the buffer is smaller, we can write completion. var completionTask2 = writer.TryWriteCompleteAsync(); Assert.NotNull(completionTask2); fake.CompleteCurrentTask(); // Message 4 fake.CompleteCurrentTask(); // Completion AssertCompletedWithStatus(task1, TaskStatus.RanToCompletion); AssertCompletedWithStatus(task2, TaskStatus.RanToCompletion); AssertCompletedWithStatus(task4, TaskStatus.RanToCompletion); AssertCompletedWithStatus(completionTask2, TaskStatus.RanToCompletion); fake.AssertMessages("1", "2", "4"); fake.AssertCompleted(); }
public void WriteBeyondBuffer() { var fake = new FakeWriter(); var writer = new BufferedClientStreamWriter <string>(fake, 2); var task1 = writer.WriteAsync("1"); var task2 = writer.WriteAsync("2"); // The (object) cast is because to make xUnit understand that the call itself should throw; // we don't return a failed task. Assert.Throws <InvalidOperationException>(() => (object)writer.WriteAsync("3")); fake.CompleteCurrentTask(); // Message 1 WaitForSpace(writer); // Now the buffer is smaller, we can write again. var task4 = writer.WriteAsync("4"); // Completion fails, no space in buffer Assert.Throws <InvalidOperationException>(() => (object)writer.WriteCompleteAsync()); fake.CompleteCurrentTask(); // Message 2 WaitForSpace(writer); // Completion succeeds, there is now space in the buffer. var completionTask = writer.WriteCompleteAsync(); fake.CompleteCurrentTask(); // Message 4 fake.CompleteCurrentTask(); // Completion AssertCompletedWithStatus(task1, TaskStatus.RanToCompletion); AssertCompletedWithStatus(task2, TaskStatus.RanToCompletion); AssertCompletedWithStatus(task4, TaskStatus.RanToCompletion); AssertCompletedWithStatus(completionTask, TaskStatus.RanToCompletion); fake.AssertMessages("1", "2", "4"); fake.AssertCompleted(); }