Added test for Drain.
Fixed test.