moving more test logic to parent class