Add helper function to set rlimit for tests
am: 0a1a9a11d9

Change-Id: I6c71f8a79be2a8bd74405f1c850230e56e7afdfc