30
30
31
31
def create_mock_drive_item (is_file , name , children = None ):
32
32
"""Helper function to create a mock drive item."""
33
- mock_item = MagicMock (properties = {
34
- "@microsoft.graph.downloadUrl" : "test_url" ,
35
- "lastModifiedDateTime" : datetime (1991 , 8 , 24 ),
36
- "createdDateTime" : datetime (1991 , 8 , 24 ),
37
- })
33
+ mock_item = MagicMock (
34
+ properties = {
35
+ "@microsoft.graph.downloadUrl" : "test_url" ,
36
+ "lastModifiedDateTime" : datetime (1991 , 8 , 24 ),
37
+ "createdDateTime" : datetime (1991 , 8 , 24 ),
38
+ }
39
+ )
38
40
mock_item .is_file = is_file
39
41
mock_item .name = name
40
42
mock_item .children .get .return_value .execute_query = Mock (return_value = children or [])
@@ -70,13 +72,13 @@ def create_mock_drive_files():
70
72
uri = "file1.csv" ,
71
73
download_url = "https://example.com/file1.csv" ,
72
74
last_modified = datetime (2021 , 1 , 1 ),
73
- created_at = datetime (2021 , 1 , 1 )
75
+ created_at = datetime (2021 , 1 , 1 ),
74
76
),
75
77
MicrosoftSharePointRemoteFile (
76
78
uri = "file2.txt" ,
77
79
download_url = "https://example.com/file2.txt" ,
78
80
last_modified = datetime (2021 , 1 , 1 ),
79
- created_at = datetime (2021 , 1 , 1 )
81
+ created_at = datetime (2021 , 1 , 1 ),
80
82
),
81
83
]
82
84
@@ -91,13 +93,13 @@ def create_mock_shared_drive_files():
91
93
uri = "file3.csv" ,
92
94
download_url = "https://example.com/file3.csv" ,
93
95
last_modified = datetime (2021 , 3 , 1 ),
94
- created_at = datetime (2021 , 3 , 1 )
96
+ created_at = datetime (2021 , 3 , 1 ),
95
97
),
96
98
MicrosoftSharePointRemoteFile (
97
99
uri = "file4.txt" ,
98
100
download_url = "https://example.com/file4.txt" ,
99
101
last_modified = datetime (2021 , 4 , 1 ),
100
- created_at = datetime (2021 , 4 , 1 )
102
+ created_at = datetime (2021 , 4 , 1 ),
101
103
),
102
104
]
103
105
@@ -406,10 +408,20 @@ def test_list_directories_and_files():
406
408
407
409
assert len (result ) == 2
408
410
assert result == [
409
- MicrosoftSharePointRemoteFile (uri = 'https://example.com/root/folder1/file1.txt' , last_modified = datetime (1991 , 8 , 24 , 0 , 0 ),
410
- mime_type = None , download_url = 'test_url' , created_at = datetime (1991 , 8 , 24 , 0 , 0 )),
411
- MicrosoftSharePointRemoteFile (uri = 'https://example.com/root/file2.txt' , last_modified = datetime (1991 , 8 , 24 , 0 , 0 ),
412
- mime_type = None , download_url = 'test_url' , created_at = datetime (1991 , 8 , 24 , 0 , 0 )),
411
+ MicrosoftSharePointRemoteFile (
412
+ uri = "https://example.com/root/folder1/file1.txt" ,
413
+ last_modified = datetime (1991 , 8 , 24 , 0 , 0 ),
414
+ mime_type = None ,
415
+ download_url = "test_url" ,
416
+ created_at = datetime (1991 , 8 , 24 , 0 , 0 ),
417
+ ),
418
+ MicrosoftSharePointRemoteFile (
419
+ uri = "https://example.com/root/file2.txt" ,
420
+ last_modified = datetime (1991 , 8 , 24 , 0 , 0 ),
421
+ mime_type = None ,
422
+ download_url = "test_url" ,
423
+ created_at = datetime (1991 , 8 , 24 , 0 , 0 ),
424
+ ),
413
425
]
414
426
415
427
@@ -507,7 +519,7 @@ def test_get_shared_files_from_all_drives(
507
519
"name" : "TestFile.txt" ,
508
520
"@microsoft.graph.downloadUrl" : "http://example.com/download" ,
509
521
"lastModifiedDateTime" : "2021-01-01T00:00:00Z" ,
510
- "createdDateTime" : "2021-01-01T00:00:00Z"
522
+ "createdDateTime" : "2021-01-01T00:00:00Z" ,
511
523
}
512
524
513
525
empty_folder_response = {"folder" : True , "value" : []}
@@ -532,7 +544,7 @@ def test_get_shared_files_from_all_drives(
532
544
"name" : "NestedFile.txt" ,
533
545
"@microsoft.graph.downloadUrl" : "http://example.com/nested" ,
534
546
"lastModifiedDateTime" : "2021-01-02T00:00:00Z" ,
535
- "createdDateTime" : "2021-01-02T00:00:00Z"
547
+ "createdDateTime" : "2021-01-02T00:00:00Z" ,
536
548
}
537
549
],
538
550
"name" : "subfolder2" ,
@@ -547,7 +559,13 @@ def test_get_shared_files_from_all_drives(
547
559
file_response ,
548
560
[],
549
561
[
550
- MicrosoftSharePointRemoteFile (uri = 'http://example.com/TestFile.txt' , last_modified = datetime (2021 , 1 , 1 , 0 , 0 ), mime_type = None , download_url = 'http://example.com/download' , created_at = datetime (2021 , 1 , 1 , 0 , 0 )),
562
+ MicrosoftSharePointRemoteFile (
563
+ uri = "http://example.com/TestFile.txt" ,
564
+ last_modified = datetime (2021 , 1 , 1 , 0 , 0 ),
565
+ mime_type = None ,
566
+ download_url = "http://example.com/download" ,
567
+ created_at = datetime (2021 , 1 , 1 , 0 , 0 ),
568
+ ),
551
569
],
552
570
False ,
553
571
None ,
@@ -564,7 +582,13 @@ def test_get_shared_files_from_all_drives(
564
582
not_empty_subfolder_response ,
565
583
],
566
584
[
567
- MicrosoftSharePointRemoteFile (uri = 'http://example.com/subfolder2/NestedFile.txt' , last_modified = datetime (2021 , 1 , 2 , 0 , 0 ), mime_type = None , download_url = 'http://example.com/nested' , created_at = datetime (2021 , 1 , 2 , 0 , 0 ))
585
+ MicrosoftSharePointRemoteFile (
586
+ uri = "http://example.com/subfolder2/NestedFile.txt" ,
587
+ last_modified = datetime (2021 , 1 , 2 , 0 , 0 ),
588
+ mime_type = None ,
589
+ download_url = "http://example.com/nested" ,
590
+ created_at = datetime (2021 , 1 , 2 , 0 , 0 ),
591
+ )
568
592
],
569
593
False ,
570
594
None ,
0 commit comments