add flight availabilities and test

tsolakoua · tsolakoua · commit 49382e9163f2 · 2021-05-21T15:54:17.000+02:00
diff --git a/README.md b/README.md
@@ -233,6 +233,9 @@ amadeus.schedule.flights.get(params:["carrierCode":"AZ",
                                      "flightNumber":"319",
                                      "scheduledDepartureDate":"2021-03-13"], onCompletion: { ... })
 
+// Flight Availabilities Search 
+amadeus.shopping.availability.flightAvailabilities.post(body: jsonBody, onCompletion: { ... })
+
 // Seatmap Display
 amadeus.shopping.seatMaps.get(params: ["flight-orderId":"eJzTd9f3NjIJdzUGAAp%2fAiY="],
                               onCompletion: { ... })
diff --git a/Sources/Amadeus/Shopping.swift b/Sources/Amadeus/Shopping.swift
@@ -16,6 +16,7 @@ public class Shopping {
     public let hotelOffers: HotelOffers
     public let hotelOfferByHotel: HotelOfferByHotel
     public let activities: Activities
+    public let availability: Availability
 
     private let client: Client
 
@@ -28,6 +29,7 @@ public class Shopping {
         hotelOffers = HotelOffers(client: client)
         hotelOfferByHotel = HotelOfferByHotel(client: client)
         activities = Activities(client: client)
+        self.availability = Availability(client: client)
         self.client = client
     }
 
diff --git a/Sources/Amadeus/Shopping/Availability.swift b/Sources/Amadeus/Shopping/Availability.swift
@@ -0,0 +1,16 @@
+import Foundation
+
+/// A namespaced client for the `v1/shopping/availability` endpoints
+///
+/// Access via the `Amadeus` object
+/// ```swift
+/// let amadeus = Amadeus(client_id, secret_id)
+/// amadeus.shopping.availability
+/// ```
+public class Availability {
+    public let flightAvailabilities: FlightAvailabilities
+
+    public init(client: Client) {
+        flightAvailabilities = FlightAvailabilities(client: client)
+    }
+}
diff --git a/Sources/Amadeus/Shopping/FlightOffers/FlightAvailabilities.swift b/Sources/Amadeus/Shopping/FlightOffers/FlightAvailabilities.swift
@@ -0,0 +1,23 @@
+
+import Foundation
+import SwiftyJSON
+
+public class FlightAvailabilities {
+    private var client: Client
+    
+    public init(client: Client) {
+        self.client = client
+    }
+    
+    /// Gets available seats in different fare classes
+    ///
+    /// - Returns:
+    ///    `JSON` object
+    ///
+    public func post(body: JSON, params: [String: String] = [:], onCompletion: @escaping AmadeusResponse) {
+        
+        client.post(path: "/v1/shopping/availability/flight-availabilities", body: body, params: params, onCompletion: { result in
+            onCompletion(result)
+        })
+    }
+}
diff --git a/Tests/AmadeusTests/AirShoppingTests.swift b/Tests/AmadeusTests/AirShoppingTests.swift
@@ -150,6 +150,54 @@ class AirShoppingTests: XCTestCase {
         }
     }
     
+    func testFlightAvailabilitiesPost() {
+        let expectation = XCTestExpectation(description: "TimeOut")
+        
+        let jsonString: String = """
+        {
+            "originDestinations": [
+                {
+                    "id": "1",
+                    "originLocationCode": "MIA",
+                    "destinationLocationCode": "ATL",
+                    "departureDateTime": {
+                        "date": "2021-11-01"
+                    }
+                }
+            ],
+            "travelers": [
+                {
+                    "id": "1",
+                    "travelerType": "ADULT"
+                }
+            ],
+            "sources": [
+                "GDS"
+            ]
+        }
+        """
+        
+        let dataFromString = jsonString.data(using: .utf8, allowLossyConversion: false)
+        do {
+            let body: JSON = try JSON(data: dataFromString!)
+            
+            amadeus.shopping.availability.flightAvailabilities.post(body: body, onCompletion: {
+                result in
+                switch result {
+                case .success(let response):
+                    XCTAssertEqual(response.statusCode, 200)
+                case .failure(let error):
+                    fatalError(error.localizedDescription)
+                }
+                expectation.fulfill()
+            })
+            
+            wait(for: [expectation], timeout: 60)
+            
+        } catch _ as NSError {
+            assertionFailure("JSON not valid")
+        }
+    }
     func testFlightDates() {
         let expectation = XCTestExpectation(description: "TimeOut")
         
