Added drain method to testing modules

Change-Id: I3509602cdcc12ba1071a3e3d5e09e77902044845