Add LocalStoreDrain to use in tests