Update file download tests

wordpress-mobile · Jun 5, 2024 · 0e3a287 · 0e3a287
1 parent 1c4d187
commit 0e3a287
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/Tests/WordPressKitTests/Mock Data/stats-file-downloads.json b/Tests/WordPressKitTests/Mock Data/stats-file-downloads.json
@@ -5,12 +5,16 @@
         "2019-07-01": {
             "files": [
                 {
-                    "filename": "/2019/07/test.pdf",
-                    "downloads": 11
+                    "filename": "test.pdf",
+                    "downloads": 11,
+                    "relative_url": "/2019/07/test.pdf",
+                    "download_url": "https://testsite.com/2019/07/test.pdf"
                 },
                 {
-                    "filename": "/2019/07/sampleaudio.mp3",
-                    "downloads": 4
+                    "filename": "sampleaudio.mp3",
+                    "downloads": 4,
+                    "relative_url": "/2019/07/sampleaudio.mp3",
+                    "download_url": "https://testsite.com/2019/07/sampleaudio.mp3"
                 }
             ],
             "other_downloads": 0,

diff --git a/Tests/WordPressKitTests/Tests/StatsRemoteV2Tests.swift b/Tests/WordPressKitTests/Tests/StatsRemoteV2Tests.swift
@@ -393,7 +393,8 @@ class StatsRemoteV2Tests: RemoteTestCase, RESTTestable {
 
             XCTAssertEqual(fileDownloads?.fileDownloads.count, 2)
 
-            XCTAssertEqual(fileDownloads?.fileDownloads.first!.file, "/2019/07/test.pdf")
+            XCTAssertEqual(fileDownloads?.fileDownloads.first!.file, "test.pdf")
+            XCTAssertEqual(fileDownloads?.fileDownloads.first!.downloadURL, URL(string: "https://testsite.com/2019/07/test.pdf"))
             XCTAssertEqual(fileDownloads?.fileDownloads.first!.downloadCount, 11)
 
             expect.fulfill()