def test_list_files(self, mock_build):
# mock
drive_service = DriveService(self.credentials)
with patch.object(drive_service.service.files(), 'list', return_value=MagicMock()) as mock_list_files:
# act
drive_service.list_files()
# assert
mock_list_files.assert_called_with(fields='nextPageToken, files(id, name)', pageSize=10)
def test_create_metadata(self):
with patch.object(GoogleService, "__init__", return_value=None) as mock_constructor:
drive_service = DriveService(self.credentials)
# act
file_metadata = drive_service.create_file_metadata("filename", "mimetype")
# assert
assert file_metadata["name"] == "filename"
assert file_metadata["mimeType"] == "mimetype"
def test_create_media(self, mock_fileupload):
# mock
with patch.object(GoogleService, "__init__", return_value=None) as mock_constructor:
drive_service = DriveService(self.credentials)
# act
media = drive_service.create_media("filename", "mimetype")
# assert
assert isinstance(media, MediaFileUpload)
mock_fileupload.assert_called_with("filename", mimetype="mimetype", resumable=True)
def test_create_file(self, mock_build, mock_fileupload):
# mock
drive_service = DriveService(self.credentials)
with patch.object(drive_service.service.files(), 'create', return_value=Mock()) as mock_create_file:
file_metadata = {
"name": "filename",
"mimeType": "mimetype"
}
# act
drive_service.create_file("filename", "mimetype")
# assert
mock_create_file.assert_called_with(body=file_metadata, fields="id", media_body=mock_fileupload())
