Added more coverage in functional tests for FlushAsync

`// Arrange`