Moved test behaviour classes back to the src/test folder