Migrate test base class to seperate lib