Exemplo n.º 1
0
    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())