TrackingControllerIntegrationTests.cs

using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Net.Http.Json;
using System.Threading.Tasks;
using FluentAssertions;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Mvc.Testing;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using ParcelTracking.Infrastructure.Data;
using ParcelTracking.Application.DTOs.Parcels;
using ParcelTracking.Application.DTOs.TrackingEvents;
using ParcelTracking.Domain.Enums;
using ParcelTracking.Domain.Entities;
using Xunit;

namespace ParcelTracking.API.IntegrationTests
{
    /// <summary>
    /// Integration tests for the Tracking Controller endpoints.
    /// 
    /// These tests require a PostgreSQL database. To run with Docker:
    /// docker run -d --name parceltrack-test-db -e POSTGRES_USER=test -e POSTGRES_PASSWORD=test -e POSTGRES_DB=parceltracking_test -p 5433:5432 postgres:15
    /// 
    /// Then update the connection string in this test class to point to your database instance.
    /// </summary>
    public class TrackingControllerIntegrationTests : IAsyncLifetime
    {
        private WebApplicationFactory<Program>? _factory;
        private HttpClient? _client;
        private string? _connectionString;

        public async Task InitializeAsync()
        {
            // Use a connection string pointing to a running Postgres instance
            // Set via environment variable or use default test database
            _connectionString = Environment.GetEnvironmentVariable("PARCELTRACK_TEST_DB")
                ?? "Host=localhost;Port=5433;Database=parceltracking_test;Username=test;Password=test;Pooling=false;";

            try
            {
                _factory = new WebApplicationFactory<Program>()
                    .WithWebHostBuilder(builder =>
                    {
                        builder.ConfigureAppConfiguration((context, cfg) =>
                        {
                            var inMemConfig = new Dictionary<string, string?>
                            {
                                {"ConnectionStrings:DefaultConnection", _connectionString}
                            };
                            cfg.AddInMemoryCollection(inMemConfig);
                        });
                    });

                _client = _factory.CreateClient();

                // Ensure database is created and migrated
                var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                    .UseNpgsql(_connectionString)
                    .Options;

                using (var db = new ParcelTrackingDbContext(options))
                {
                    db.Database.EnsureCreated();
                }
            }
            catch (Exception)
            {
                // If Docker PostgreSQL is not available, silently skip these tests
                _connectionString = null;
                _factory?.Dispose();
                _factory = null;
                _client = null;
            }
        }

        public async Task DisposeAsync()
        {
            _client?.Dispose();
            _factory?.Dispose();
        }

        #region GetByTrackingNumber Tests

        [Fact]
        public async Task GetByTrackingNumber_WithValidTrackingNumber_Returns200WithParcelData()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available. Start Docker container with: docker run -d --name parceltrack-test-db -e POSTGRES_USER=test -e POSTGRES_PASSWORD=test -e POSTGRES_DB=parceltracking_test -p 5433:5432 postgres:15");
            }

            // Arrange: seed test data
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var trackingNumber = "PKT" + Guid.NewGuid().ToString().Substring(0, 9);
            Guid parcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "NYC", Street1 = "123 Main" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Boston", Street1 = "456 Oak" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                var parcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = trackingNumber,
                    Status = ParcelStatus.InTransit,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Standard,
                    Weight = 2.5m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 150m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow.AddDays(-1),
                    UpdatedAt = DateTime.UtcNow.AddDays(-1),
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(2)
                };
                db.Parcels.Add(parcel);
                await db.SaveChangesAsync();
                parcelId = parcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
            parcelDto!.TrackingNumber.Should().Be(trackingNumber);
            parcelDto.Id.Should().Be(parcelId);
            parcelDto.Weight.Should().Be(2.5m);
            parcelDto.DeclaredValue.Should().Be(150m);
        }

        [Fact]
        public async Task GetByTrackingNumber_WithNonExistentTrackingNumber_Returns404()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange
            var nonExistentTrackingNumber = "NONEXISTENT" + Guid.NewGuid().ToString().Substring(0, 10);

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{nonExistentTrackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.NotFound);
        }

        [Fact]
        public async Task GetByTrackingNumber_WithDeliveredParcel_Returns200WithDeliveredStatus()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange: seed test data
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var trackingNumber = "DELIVER" + Guid.NewGuid().ToString().Substring(0, 7);
            Guid parcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Seattle", Street1 = "789 Pine" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "LA", Street1 = "321 Elm" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                var parcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = trackingNumber,
                    Status = ParcelStatus.Delivered,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Express,
                    Weight = 0.5m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 500m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow.AddDays(-5),
                    UpdatedAt = DateTime.UtcNow.AddDays(-5),
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(-1)
                };
                db.Parcels.Add(parcel);
                await db.SaveChangesAsync();
                parcelId = parcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
            parcelDto!.TrackingNumber.Should().Be(trackingNumber);
        }

        [Fact]
        public async Task GetByTrackingNumber_WithPickedUpParcel_Returns200WithPickedUpStatus()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange: seed test data
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var trackingNumber = "PICKEDUP" + Guid.NewGuid().ToString().Substring(0, 7);
            Guid parcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Denver", Street1 = "999 Ridge" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Chicago", Street1 = "555 Lake" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                var parcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = trackingNumber,
                    Status = ParcelStatus.PickedUp,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Standard,
                    Weight = 1.2m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 75m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow.AddHours(-12),
                    UpdatedAt = DateTime.UtcNow.AddHours(-12),
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(3)
                };
                db.Parcels.Add(parcel);
                await db.SaveChangesAsync();
                parcelId = parcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
        }

        [Fact]
        public async Task GetByTrackingNumber_WithSpecialCharactersInTrackingNumber_ReturnsNotFound()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange
            var trackingNumberWithSpecialChars = "PKT-123_456@789";

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumberWithSpecialChars}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.NotFound);
        }

        [Fact]
        public async Task GetByTrackingNumber_WithExceptionParcel_Returns200WithExceptionStatus()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange: seed test data
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var trackingNumber = "EXCEPTION" + Guid.NewGuid().ToString().Substring(0, 6);
            Guid parcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Phoenix", Street1 = "777 Desert" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Boston", Street1 = "222 Harbor" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                var parcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = trackingNumber,
                    Status = ParcelStatus.Exception,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Standard,
                    Weight = 3m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 200m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow.AddDays(-2),
                    UpdatedAt = DateTime.UtcNow.AddDays(-2),
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(1)
                };
                db.Parcels.Add(parcel);
                await db.SaveChangesAsync();
                parcelId = parcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
        }

        [Fact]
        public async Task GetByTrackingNumber_WithReturnedParcel_Returns200WithReturnedStatus()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange: seed test data
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var trackingNumber = "RETURNED" + Guid.NewGuid().ToString().Substring(0, 7);
            Guid parcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Austin", Street1 = "444 Congress" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Portland", Street1 = "888 Willamette" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                var parcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = trackingNumber,
                    Status = ParcelStatus.Returned,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Standard,
                    Weight = 1.8m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 120m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow.AddDays(-4),
                    UpdatedAt = DateTime.UtcNow.AddDays(-4),
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(-1)
                };
                db.Parcels.Add(parcel);
                await db.SaveChangesAsync();
                parcelId = parcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{trackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
        }

        [Fact]
        public async Task GetByTrackingNumber_WithMultipleParcels_ReturnsCorrectParcelByTrackingNumber()
        {
            // Skip test if Docker database is not available
            if (_client == null || _connectionString == null)
            {
                throw new SkipTestException("PostgreSQL database not available");
            }

            // Arrange: seed test data with multiple parcels
            var options = new DbContextOptionsBuilder<ParcelTrackingDbContext>()
                .UseNpgsql(_connectionString)
                .Options;

            var targetTrackingNumber = "TARGET" + Guid.NewGuid().ToString().Substring(0, 7);
            var otherTrackingNumber = "OTHER" + Guid.NewGuid().ToString().Substring(0, 8);
            Guid targetParcelId;

            using (var db = new ParcelTrackingDbContext(options))
            {
                db.Database.EnsureCreated();

                var shipper = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Miami", Street1 = "100 Ocean" };
                var recipient = new Address { Id = Guid.NewGuid(), CountryCode = "US", City = "Tampa", Street1 = "200 Bay" };
                db.Addresses.Add(shipper);
                db.Addresses.Add(recipient);
                await db.SaveChangesAsync();

                // Create target parcel
                var targetParcel = new Parcel
                {
                    Id = Guid.NewGuid(),
                    TrackingNumber = targetTrackingNumber,
                    Status = ParcelStatus.InTransit,
                    ShipperAddressId = shipper.Id,
                    RecipientAddressId = recipient.Id,
                    ServiceType = ServiceType.Standard,
                    Weight = 1m,
                    WeightUnit = WeightUnit.Kg,
                    DeclaredValue = 100m,
                    Currency = "USD",
                    CreatedAt = DateTime.UtcNow,
                    UpdatedAt = DateTime.UtcNow,
                    EstimatedDeliveryDate = DateTime.UtcNow.AddDays(3)
                };
                db.Parcels.Add(targetParcel);

                // Create other parcels to ensure we get the right one
                for (int i = 0; i < 5; i++)
                {
                    var otherParcel = new Parcel
                    {
                        Id = Guid.NewGuid(),
                        TrackingNumber = $"OTHER{i}" + Guid.NewGuid().ToString().Substring(0, 8),
                        Status = ParcelStatus.Delivered,
                        ShipperAddressId = shipper.Id,
                        RecipientAddressId = recipient.Id,
                        ServiceType = ServiceType.Express,
                        Weight = 2m,
                        WeightUnit = WeightUnit.Kg,
                        DeclaredValue = 200m,
                        Currency = "USD",
                        CreatedAt = DateTime.UtcNow.AddDays(-1),
                        UpdatedAt = DateTime.UtcNow.AddDays(-1),
                        EstimatedDeliveryDate = DateTime.UtcNow
                    };
                    db.Parcels.Add(otherParcel);
                }

                await db.SaveChangesAsync();
                targetParcelId = targetParcel.Id;
            }

            // Act
            var response = await _client!.GetAsync($"/api/v1/Tracking/{targetTrackingNumber}");

            // Assert
            response.StatusCode.Should().Be(System.Net.HttpStatusCode.OK);

            var parcelDto = await response.Content.ReadFromJsonAsync<ParcelDto>();
            parcelDto.Should().NotBeNull();
            parcelDto!.Id.Should().Be(targetParcelId);
            parcelDto.TrackingNumber.Should().Be(targetTrackingNumber);
        }

        #endregion
    }
}