From b389b181d5dd18379823dec7df08b0fd69bad6e0 Mon Sep 17 00:00:00 2001
From: Olga Nemt <on.nemtina@digitaldrilling.ru>
Date: Thu, 29 Feb 2024 09:10:15 +0500
Subject: [PATCH] =?UTF-8?q?=D0=98=D0=BD=D1=82=D0=B5=D0=B3=D1=80=D0=B0?=
 =?UTF-8?q?=D1=86=D0=B8=D0=BE=D0=BD=D0=BD=D1=8B=D0=B5=20=D1=82=D0=B5=D1=81?=
 =?UTF-8?q?=D1=82=20=D0=BD=D0=B0=20=D1=81=D0=BA=D0=B0=D0=B8=D1=87=D0=B2?=
 =?UTF-8?q?=D0=B0=D0=BD=D0=B8=D0=B5=20=D0=A0=D0=A2=D0=9A-=D0=BE=D1=82?=
 =?UTF-8?q?=D1=87=D0=B5=D1=82=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Clients/IProcessMapReportDrilling.cs      |  4 ++
 .../ProcessMapReportDrillingControllerTest.cs | 38 ++++++++++++++++++-
 2 files changed, 41 insertions(+), 1 deletion(-)

diff --git a/AsbCloudWebApi.IntegrationTests/Clients/IProcessMapReportDrilling.cs b/AsbCloudWebApi.IntegrationTests/Clients/IProcessMapReportDrilling.cs
index b721df75..12adac91 100644
--- a/AsbCloudWebApi.IntegrationTests/Clients/IProcessMapReportDrilling.cs
+++ b/AsbCloudWebApi.IntegrationTests/Clients/IProcessMapReportDrilling.cs
@@ -13,4 +13,8 @@ public interface IProcessMapReportDrilling
 
     [Get($"{BaseRoute}/report")]
     Task<IApiResponse<IEnumerable<ProcessMapReportDataSaubStatDto>>> GetReportAsync(int idWell, [FromQuery] DataSaubStatRequest request, CancellationToken cancellationToken);
+
+    [Get($"{BaseRoute}/report/export")]
+    Task<IApiResponse<IActionResult>> ExportReportAsync(int idWell, [FromQuery] DataSaubStatRequest request, CancellationToken cancellationToken);
+
 }
diff --git a/AsbCloudWebApi.IntegrationTests/Controllers/ProcessMapPlan/ProcessMapReportDrillingControllerTest.cs b/AsbCloudWebApi.IntegrationTests/Controllers/ProcessMapPlan/ProcessMapReportDrillingControllerTest.cs
index 2d08b0c8..77951c23 100644
--- a/AsbCloudWebApi.IntegrationTests/Controllers/ProcessMapPlan/ProcessMapReportDrillingControllerTest.cs
+++ b/AsbCloudWebApi.IntegrationTests/Controllers/ProcessMapPlan/ProcessMapReportDrillingControllerTest.cs
@@ -172,7 +172,6 @@ public class ProcessMapReportDrillingControllerTest: BaseIntegrationTest
 
     public ProcessMapReportDrillingControllerTest(WebAppFactoryFixture factory) : base(factory)
     {
-        dbContext.CleanupDbSet<ProcessMapPlanDrilling>();
         client = factory.GetAuthorizedHttpClient<IProcessMapReportDrilling>();
     }
 
@@ -181,6 +180,9 @@ public class ProcessMapReportDrillingControllerTest: BaseIntegrationTest
     [Fact]
     public async Task Get_rtk_report_returns_success()
     {
+        //clear
+        dbContext.CleanupDbSet<ProcessMapPlanDrilling>();
+
         //arrange
         var dbset = dbContext.Set<ProcessMapPlanDrilling>();
         dbset.Add(entity);
@@ -234,4 +236,38 @@ public class ProcessMapReportDrillingControllerTest: BaseIntegrationTest
         //assert
         Assert.Equal(HttpStatusCode.BadRequest, response.StatusCode);
     }
+
+    [Fact]
+    public async Task Export_rtk_report_returns_success()
+    {
+        //clear
+        dbContext.CleanupDbSet<ProcessMapPlanDrilling>();
+
+        //arrange
+        var dbset = dbContext.Set<ProcessMapPlanDrilling>();
+        dbset.Add(entity);
+        entity.Id = 1;
+        dbset.Add(entity);
+        entity.Id = 2;
+        dbset.Add(entity);
+
+        var dbSetWellOperationCategory = dbContext.Set<WellOperationCategory>();
+        dbSetWellOperationCategory.Add(wellOperationCategory);
+
+        var dbSetWellOperation = dbContext.Set<WellOperation>();
+        dbSetWellOperation.Add(wellOperation);
+
+        var dbSetDataSaubStat = dbContext.Set<DataSaubStat>();
+        dbSetDataSaubStat.AddRange(dataSaubStats);
+
+        dbContext.SaveChanges();
+
+        //act
+        var request = new DataSaubStatRequest();
+        var response = await client.ExportReportAsync(1, request, CancellationToken.None);
+
+        //assert
+        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
+        Assert.NotNull(response.Content);       
+    }
 }