Restructure for Spec to inherit from TestCase to allow unit tests as well