test_google_drive.py

import random
import string
import unittest

from parsons import GoogleDrive
from test.utils import mark_live_test

# Test Slides: https://docs.google.com/presentation/d/19I-kicyaJV53KoPNwt77KJL10fHzWFdZ_c2mW4XJaxc


@mark_live_test
class TestGoogleDrive(unittest.TestCase):
    def setUp(self):
        self.gd = GoogleDrive()

    def test_get_permissions(self):
        file_id = "19I-kicyaJV53KoPNwt77KJL10fHzWFdZ_c2mW4XJaxc"
        p = self.gd.get_permissions(file_id)
        assert "anyoneWithLink" in [x["id"] for x in p["permissions"]]

    def test_share_object(self):
        file_id = "19I-kicyaJV53KoPNwt77KJL10fHzWFdZ_c2mW4XJaxc"
        email = "".join(random.choices(string.ascii_letters, k=10)) + "@gmail.com"
        email_addresses = [email]

        before = self.gd.get_permissions(file_id)["permissions"]
        self.gd.share_object(file_id, email_addresses)
        after = self.gd.get_permissions(file_id)["permissions"]
        assert len(after) > len(before)