close #3 (#4)

picolino · Jun 6, 2021 · ec0ef27 · ec0ef27
1 parent 16f7d40
commit ec0ef27
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/isdayoff.Tests/IsDayOffApiClientTests/WhenGetData.cs b/isdayoff.Tests/IsDayOffApiClientTests/WhenGetData.cs
@@ -67,6 +67,10 @@ private static IEnumerable UrlConstructionTestData()
                          {TestName = "Ukraine passes correctly", ExpectedResult = $"{ApiBaseUrlStub}getdata?date1=20200804&date2=20200804&cc=ua"};
             yield return new TestCaseData(04.08.Of(2020), 04.08.Of(2020), Country.USA, "0")
                          {TestName = "USA passes correctly", ExpectedResult = $"{ApiBaseUrlStub}getdata?date1=20200804&date2=20200804&cc=us"};
+            yield return new TestCaseData(04.08.Of(2020), 04.08.Of(2020), Country.Uzbekistan, "0")
+                         {TestName = "Uzbekistan passes correctly", ExpectedResult = $"{ApiBaseUrlStub}getdata?date1=20200804&date2=20200804&cc=uz"};
+            yield return new TestCaseData(04.08.Of(2020), 04.08.Of(2020), Country.Turkey, "0")
+                         {TestName = "Turkey passes correctly", ExpectedResult = $"{ApiBaseUrlStub}getdata?date1=20200804&date2=20200804&cc=tr"};
         }
 
         [Test]

diff --git a/isdayoff/Contract/Country.cs b/isdayoff/Contract/Country.cs
@@ -1,12 +1,13 @@
-
-namespace isdayoff.Contract
+namespace isdayoff.Contract
 {
     public enum Country
     {
         Russia,
         Belarus,
         Ukraine,
         Kazakhstan,
-        USA
+        USA,
+        Uzbekistan,
+        Turkey
     }
 }
diff --git a/isdayoff/Core/IsDayOffApiClient.cs b/isdayoff/Core/IsDayOffApiClient.cs
@@ -120,6 +120,10 @@ private static string GetCountryCode(Country country)
                     return "kz";
                 case Country.USA:
                     return "us";
+                case Country.Uzbekistan:
+                    return "uz";
+                case Country.Turkey:
+                    return "tr";
                 default:
                     throw new ArgumentOutOfRangeException(nameof(country), country, ErrorsMessages.UnknownCountry());
             }