Add network status endpoint to network report service.

helikon-labs · Jun 19, 2024 · 344bd7d · 344bd7d
1 parent cbbff8b
commit 344bd7d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/Sources/SubVTData/Service/REST/ReportService.swift b/Sources/SubVTData/Service/REST/ReportService.swift
@@ -164,4 +164,11 @@ public final class ReportService: BaseRESTService {
             method: .get
         )
     }
+
+    public func getNetworkStatus() -> ServiceResponsePublisher<NetworkStatus> {
+        return perform(
+            path: "/network/status",
+            method: .get
+        )
+    }
 }
diff --git a/Tests/SubVTDataTests/ReportServiceTests.swift b/Tests/SubVTDataTests/ReportServiceTests.swift
@@ -200,24 +200,24 @@ final class ReportServiceTests: BaseTest {
         testServiceCall(
             publisher: service.getSessionValidatorReport(
                 validatorAccountId: validatorAccountId,
-                startSessionIndex: 36217
+                startSessionIndex: 39931
             )
         ) {
             (reports, error) in
             XCTAssertNil(error)
             XCTAssertEqual(1, reports?.count ?? 0)
             let report = reports![0]
             XCTAssertNotNil(report.paraVotesSummary)
-            XCTAssertNotNil(report.heartbeatEvent)
+            XCTAssertNil(report.heartbeatEvent)
         }
     }
 
     func test16GetMultipleSessionValidatorReport() {
         testServiceCall(
             publisher: service.getSessionValidatorReport(
                 validatorAccountId: validatorAccountId,
-                startSessionIndex: 36000,
-                endSessionIndex: 36009
+                startSessionIndex: 39950,
+                endSessionIndex: 39959
             )
         ) {
             (reports, error) in
@@ -235,6 +235,15 @@ final class ReportServiceTests: BaseTest {
         }
     }
 
+    func test18GetNetworkStatus() {
+        testServiceCall(
+            publisher: service.getNetworkStatus()
+        ) {
+            (networkStatus, error) in
+            XCTAssertNil(error)
+        }
+    }
+
     static var allTests = [
         ("test01GetSingleEraReport", test01GetSingleEraReport),
         ("test02GetMultipleEraReport", test02GetMultipleEraReport),