Masuit.LuceneEFCore.SearchEngine/Masuit.LuceneEFCore.SearchEngine.Test/Helpers/TestDbContext.cs

using Masuit.LuceneEFCore.SearchEngine.Test.Models;
using Microsoft.EntityFrameworkCore;
using System.IO;
using System.Linq;

namespace Masuit.LuceneEFCore.SearchEngine.Test.Helpers
{
    public class TestDbContext : DbContext
    {
        public DbSet<User> Users { get; set; }
        public DbSet<City> Cities { get; set; }

        public TestDbContext(DbContextOptions<TestDbContext> options) : base(options)
        {
            InitializeData();
        }

        public TestDbContext() : this(new DbContextOptionsBuilder<TestDbContext>().UseInMemoryDatabase("Add_writes_to_database").Options)
        {
        }

        private void InitializeData()
        {
            if (!Users.Any())
            {
                TextReader reader = new StreamReader("Helpers\\TestData\\MOCK_USERS.csv");

                string data = reader.ReadLine();

                while ((data = reader.ReadLine()) != null)
                {
                    string[] line = data.Split(',');
                    Users.Add(new User()
                    {
                        FirstName = line[1],
                        Surname = line[2],
                        Email = line[3],
                        JobTitle = line[5]
                    });
                }
                reader.Close();
                SaveChanges();
            }

            if (!Cities.Any())
            {
                TextReader reader = new StreamReader("Helpers\\TestData\\MOCK_CITIES.csv");

                string data = reader.ReadLine();
                while ((data = reader.ReadLine()) != null)
                {
                    string[] line = data.Split(',');
                    Cities.Add(new City()
                    {
                        Id = int.Parse(line[0]),
                        Country = line[1],
                        Code = line[2],
                        Name = line[3]
                    });
                }
                reader.Close();
                SaveChanges();
            }
        }
    }
}