Taking advantage of stem's new stem.util.test_tools

Clearing out a big chunk of code we copied from stem now that it's available in
its utils.