def setUpClass(cls): cls.driver = webdriver.Firefox() cls.tempdir = create_random_dir() cls.google_drive = GoogleDriveAPI(cls.tempdir) code = authorize_drive_user(cls.driver, cls.google_drive.get_auth_url(), DRIVE_EMAIL, DRIVE_PASSWD) cls.google_drive.authorize_with_code(code) cls.google_drive.build()
def test_i_can_connect_with_mail_adress_and_password(self): # Given driver = webdriver.Firefox() code = authorize_drive_user(driver, self.google_drive.get_auth_url(), DRIVE_EMAIL, DRIVE_PASSWD) # When self.google_drive.authorize_with_code(code) # Then self.assertTrue(self.google_drive.is_connected()) driver.quit()