Refactor test DeleteTrip_TripExists_ReturnsOk

oreze · Jun 2, 2024 · 4b85a4b · 4b85a4b
1 parent 3ed4fe6
commit 4b85a4b
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/BackendTripRecruitmentTask.IntegrationTests/Features/TripTests.cs b/BackendTripRecruitmentTask.IntegrationTests/Features/TripTests.cs
@@ -311,24 +311,24 @@ public async Task EditTrip_NumberOfSeatsLowerThanRegisteredUsers_ReturnsBadReque
     [Fact]
     public async Task DeleteTrip_TripExists_ReturnsOk()
     {
-        var createTripDto = new CreateTripDto(
+        var country = _dbContext.Countries.First(x => x.ThreeLetterCode == "POL");
+        var trip = Trip.Create(
             Guid.NewGuid().ToString(),
             "Random description",
             DateTime.UtcNow.AddDays(10),
             50,
-            "Poland");
-
-        var createTripJson = JsonSerializer.Serialize(createTripDto);
-        var createTripContent = new StringContent(createTripJson, Encoding.UTF8, "application/json");
-        var createTripResponse = await _httpClient.PostAsync("/trips", createTripContent);
-        createTripResponse.EnsureSuccessStatusCode();
-        var tripID = await createTripResponse.Content.ReadFromJsonAsync<int>();
-        Assert.NotEqual(0, tripID);
+            country);
+
+        await _dbContext.AddAsync(trip);
+        await _dbContext.SaveChangesAsync();
 
-        var deleteTripResponse = await _httpClient.DeleteAsync($"/trips/{tripID}");
+        var deleteTripResponse = await _httpClient.DeleteAsync($"/trips/{trip.ID}");
 
         deleteTripResponse.EnsureSuccessStatusCode();
-        var deletedTrip = await _dbContext.Trips.FindAsync(tripID);
+        var wasDeleted = await deleteTripResponse.Content.ReadFromJsonAsync<bool>();
+        Assert.True(wasDeleted);
+
+        var deletedTrip = await _dbContext.Trips.FirstOrDefaultAsync(x => x.ID == trip.ID);
         Assert.Null(deletedTrip);
     }