DMS/PMSWPF.Tests/ExcelHelperTests.cs

using NUnit.Framework;
using PMSWPF.Helper;
using System.Collections.Generic;
using System.Data;
using System.IO;

namespace PMSWPF.Tests
{
    [TestFixture]
    public class ExcelHelperTests
    {
        private string _testFilePath;

        [SetUp]
        public void Setup()
        {
            // Create a temporary file for testing
            // _testFilePath = Path.Combine(Path.GetTempPath(), "test.xlsx");
            _testFilePath =  "e:/test.xlsx";
        }

        [TearDown]
        public void TearDown()
        {
            // Clean up the temporary file
            if (File.Exists(_testFilePath))
            {
                File.Delete(_testFilePath);
            }
        }

        [Test]
        public void ExportToExcel_WithListOfObjects_CreatesFile()
        {
            // Arrange
            var data = new List<TestData>
            {
                new TestData { Id = 1, Name = "Test1" },
                new TestData { Id = 2, Name = "Test2" }
            };

            // Act
            ExcelHelper.ExportToExcel(data, _testFilePath);

            // Assert
            Assert.IsTrue(File.Exists(_testFilePath));
        }

        [Test]
        public void ExportToExcel_WithDataTable_CreatesFile()
        {
            // Arrange
            var dataTable = new DataTable();
            dataTable.Columns.Add("Id", typeof(int));
            dataTable.Columns.Add("Name", typeof(string));
            dataTable.Rows.Add(1, "Test1");
            dataTable.Rows.Add(2, "Test2");

            // Act
            ExcelHelper.ExportToExcel(dataTable, _testFilePath);

            // Assert
            Assert.IsTrue(File.Exists(_testFilePath));
        }

        [Test]
        public void ImportFromExcel_ToDataTable_ReturnsCorrectData()
        {
            // Arrange
            var dataTable = new DataTable();
            dataTable.Columns.Add("Id", typeof(int));
            dataTable.Columns.Add("Name", typeof(string));
            dataTable.Rows.Add(1, "Test1");
            dataTable.Rows.Add(2, "Test2");
            ExcelHelper.ExportToExcel(dataTable, _testFilePath);

            // Act
            var result = ExcelHelper.ImportFromExcel(_testFilePath);

            // Assert
            Assert.AreEqual(2, result.Rows.Count);
            Assert.AreEqual("1", result.Rows[0]["Id"]);
            Assert.AreEqual("Test1", result.Rows[0]["Name"]);
        }

        [Test]
        public void ImportFromExcel_ToListOfObjects_ReturnsCorrectData()
        {
            // Arrange
            var data = new List<TestData>
            {
                new TestData { Id = 1, Name = "Test1" },
                new TestData { Id = 2, Name = "Test2" }
            };
            ExcelHelper.ExportToExcel(data, _testFilePath);

            // Act
            var result = ExcelHelper.ImportFromExcel<TestData>(_testFilePath);

            // Assert
            Assert.AreEqual(2, result.Count);
            Assert.AreEqual(1, result[0].Id);
            Assert.AreEqual("Test1", result[0].Name);
        }

        private class TestData
        {
            public int Id { get; set; }
            public string Name { get; set; }
        }
    }
}