Merge pull request #144 from gustavo-bertoldi/master

Add Travel Restrictions API

Author: tsolakoua

README.md

@@ -498,6 +498,14 @@ amadeus.analytics.itineraryPriceMetrics.get({
    destinationIataCode: 'CDG',
    departureDate: '2022-03-13',
  })
+
+// Travel Restrictions
+// The Travel Restrictions API provides up-to-date data on COVID-19 caseloads and travel restrictions for over 200 countries and territories,
+// as well as hundreds of cities and regions worldwide.
+amadeus.dutyOfCare.diseases.covid19AreaReport.get({
+  countryCode: 'US',
+  cityCode: 'NYC' 
+});
 
 ```
 

spec/amadeus/namespaces.test.js

@@ -93,6 +93,10 @@ describe('Namespaces', () => {
       expect(amadeus.location).toBeDefined();
       expect(amadeus.location.analytics).toBeDefined();
       expect(amadeus.location.analytics.categoryRatedAreas).toBeDefined();
+
+      expect(amadeus.dutyOfCare).toBeDefined();
+      expect(amadeus.dutyOfCare.diseases).toBeDefined();
+      expect(amadeus.dutyOfCare.diseases.covid19AreaReport).toBeDefined();
     });
 
     it('should define all expected .get methods', () => {
@@ -146,6 +150,7 @@ describe('Namespaces', () => {
 
       expect(amadeus.location.analytics.categoryRatedAreas.get).toBeDefined();
 
+      expect(amadeus.dutyOfCare.diseases.covid19AreaReport.get).toBeDefined();
     });
 
     it('should define all expected .post methods', () => {
@@ -498,5 +503,12 @@ describe('Namespaces', () => {
         .toHaveBeenCalledWith('/v1/shopping/flight-offers/upselling', {});
     });
 
+    it('.amadeus.dutyOfCare.diseases.covid19AreaReport.get', () => {
+      amadeus.client.get = jest.fn();
+      amadeus.dutyOfCare.diseases.covid19AreaReport.get();
+      expect(amadeus.client.get)
+        .toHaveBeenCalledWith('/v1/duty-of-care/diseases/covid19-area-report', {});
+    });
+
   });
 });

src/amadeus.js

@@ -12,6 +12,7 @@ import Safety        from './amadeus/namespaces/safety';
 import Schedule      from './amadeus/namespaces/schedule';
 import Analytics     from './amadeus/namespaces/analytics';
 import Location      from './amadeus/namespaces/location';
+import DutyOfCare    from './amadeus/namespaces/duty_of_care';
 
 
 /**
@@ -76,6 +77,7 @@ class Amadeus {
     this.schedule       = new Schedule(this.client);
     this.analytics      = new Analytics(this.client);
     this.location       = new Location(this.client);
+    this.dutyOfCare     = new DutyOfCare(this.client);
   }
 
   /**

src/amadeus/namespaces/duty_of_care.js

@@ -0,0 +1,19 @@
+import Diseases from './duty_of_care/diseases';
+/**
+ * A namespaced client for the
+ * `/v1/duty-of-care` endpoint
+ *
+ * Access via the {@link Amadeus} object
+ *
+ * @param {Client} client
+ * @property {Diseases} diseases
+ */
+class DutyOfCare {
+  constructor(client) {
+    this.client   = client;
+    this.diseases = new Diseases(client);
+  }
+
+}
+
+export default DutyOfCare;

src/amadeus/namespaces/duty_of_care/diseases.js

@@ -0,0 +1,22 @@
+import Covid19AreaReport from './diseases/covid19_area_report';
+/**
+ * A namespaced client for the
+ * `/v1/duty-of-care/diseases` endpoints
+ *
+ * Access via the {@link Amadeus} object
+ *
+ * ```js
+ * let amadeus = new Amadeus();
+ * amadeus.dutyOfCare.diseases;
+ * ```
+ *
+ * @param {Client} client
+ */
+class Diseases {
+  constructor(client) {
+    this.client = client;
+    this.covid19AreaReport = new Covid19AreaReport(client);
+  }
+}
+
+export default Diseases;

src/amadeus/namespaces/duty_of_care/diseases/covid19_area_report.js

@@ -0,0 +1,45 @@
+/**
+ * A namespaced client for the
+ * `/v1/duty-of-care/diseases/covid19-area-report` endpoints
+ *
+ * Access via the {@link Amadeus} object
+ *
+ * ```js
+ * let amadeus = new Amadeus();
+ * amadeus.dutyOfCare.diseases.covid19AreaReport;
+ * ```
+ *
+ * @param {Client} client
+ */
+class Covid19AreaReport {
+  constructor(client) {
+    this.client = client;
+  }
+
+  /**
+  * Returns Covid-19 related restrictions for a given country
+  *
+  * @param {Object} params
+  * @param {string} params.countryCode  ISO 3166 Alpha-2 code. e.g. "US" United States of America.
+  * @return {Promise.<Response,ResponseError>} a Promise
+  *
+  * Returns Covid-19 restrictions in the United States
+  *
+  * ```js
+  * amadeus.dutyOfCare.diseases.covid19AreaReport.get({
+  *     countryCode: 'US'
+  * }).then(function(response){
+  *     console.log(response.data);
+  * }).catch(function(responseError){
+  *     console.log(responseError);
+  * });
+  * ```
+  */
+
+
+  get(params = {}) {
+    return this.client.get('/v1/duty-of-care/diseases/covid19-area-report', params);
+  }
+}
+
+export default Covid19AreaReport;