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 init_storage_manager(self):
self.storage_manager = StorageManager()
drive_api = GoogleDriveAPI(demerio_config_dir)
self.storage_manager.add_storage("google_drive", drive_api)
dropbox = DropboxAPI(demerio_config_dir)
self.storage_manager.add_storage("dropbox", dropbox)
box = BoxAPI(demerio_config_dir)
self.storage_manager.add_storage("box", box)
class TestGoogleDriveAuth(unittest.TestCase):
def setUp(self):
self.tempdir = create_random_dir()
self.google_drive = GoogleDriveAPI(self.tempdir)
def tearDown(self):
delete_dir(self.tempdir)
def test_without_credential_file_i_am_not_connected(self):
self.assertFalse(self.google_drive.is_connected())
@attr(slow=True)
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()
@attr(slow=True)
def test_user_cancel_connection(self):
# Given
driver = webdriver.Firefox()
driver.get(self.google_drive.get_auth_url())
try:
enter_text_by_id(driver, "Email", DRIVE_EMAIL)
enter_text_by_id(driver, "Passwd", DRIVE_PASSWD)
# When
click_button_by_id(driver, "submit_deny_access")
wait_for_code(driver, 'access_denied')
except:
raise Exception("timeout for authorization")
# Then
driver.quit()
def test_i_do_not_need_to_go_through_oauth_if_file_exists(self):
# Given
google_drive_new = GoogleDriveAPI(self.tempdir)
# Then
self.assertTrue(google_drive_new.is_connected())
def setUp(self):
self.tempdir = create_random_dir()
self.google_drive = GoogleDriveAPI(self.tempdir)
