add tests for dx-link reflection issue

dxFeed · May 20, 2024 · 17d00ad · 17d00ad
1 parent 4b364e5
commit 17d00ad
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/DXFeedFrameworkTests/DXConnectionTest.swift b/DXFeedFrameworkTests/DXConnectionTest.swift
@@ -60,6 +60,42 @@ final class DXConnectionTest: XCTestCase {
         wait(for: [receivedEventsExpectation], timeout: 4)
     }
 
+    func testDXLinkConnectionTheoPrice() throws {
+        // For token-based authorization, use the following address format:
+        // "dxlink:wss://demo.dxfeed.com/dxlink-ws[login=dxlink:token]"
+        let endpoint = try DXEndpoint.builder()
+            .build()
+
+        let subscription = try endpoint.getFeed()?.createSubscription(TheoPrice.self)
+        let receivedEventsExpectation = expectation(description: "Received events")
+        let eventListener = DXConnectionListener(expectation: receivedEventsExpectation)
+        try subscription?.add(listener: eventListener)
+        try subscription?.addSymbols(".AAPL240524C110")
+        try endpoint.connect("dxlink:wss://demo.dxfeed.com/dxlink-ws")
+        defer {
+            try? endpoint.closeAndAwaitTermination()
+        }
+        wait(for: [receivedEventsExpectation], timeout: 4)
+    }
+
+    func testDXLinkConnectionGreeks() throws {
+        // For token-based authorization, use the following address format:
+        // "dxlink:wss://demo.dxfeed.com/dxlink-ws[login=dxlink:token]"
+        let endpoint = try DXEndpoint.builder()
+            .build()
+
+        let subscription = try endpoint.getFeed()?.createSubscription(Greeks.self)
+        let receivedEventsExpectation = expectation(description: "Received events")
+        let eventListener = DXConnectionListener(expectation: receivedEventsExpectation)
+        try subscription?.add(listener: eventListener)
+        try subscription?.addSymbols(".AAPL240524C110")
+        try endpoint.connect("dxlink:wss://demo.dxfeed.com/dxlink-ws")
+        defer {
+            try? endpoint.closeAndAwaitTermination()
+        }
+        wait(for: [receivedEventsExpectation], timeout: 4)
+    }
+
     func testConnection() throws {
         // For token-based authorization, use the following address format:
         // "demo.dxfeed.com:7300[login=entitle:token]"