Restructure test setup.

As we have a base TestCase class lets add some stuff to it.