Added all fields of connection to the TestConnectionBuilder class and removed unused imports