Events.cs :  » Persistence-Frameworks » FileHelpers-Library » FileHelpersTests » CommonTests » C# / CSharp Open Source

using System;
using System.IO;
using FileHelpers;
using NUnit.Framework;

namespace FileHelpersTests.CommonTests{
  [TestFixture]
  public class Events
  {
    FileHelperEngine engine;

    [Test]
    public void ReadEvents()
    {
      before = 0;
      after = 0;

      engine = new FileHelperEngine(typeof (SampleType));
      engine.BeforeReadRecord += new BeforeReadRecordHandler(BeforeEvent);
      engine.AfterReadRecord += new AfterReadRecordHandler(AfterEvent);

      object[] res = Common.ReadTest(engine, @"Good\test1.txt");

      Assert.AreEqual(4, res.Length);
      Assert.AreEqual(4, engine.TotalRecords);
      Assert.AreEqual(4, before);
      Assert.AreEqual(4, after);
    }
    

    [Test]
    public void WriteEvents()
    {
      before = 0;
      after = 0;

      engine = new FileHelperEngine(typeof (SampleType));

            engine.BeforeWriteRecord += new BeforeWriteRecordHandler(engine_BeforeWriteRecord);
      engine.AfterWriteRecord += new AfterWriteRecordHandler(engine_AfterWriteRecord);

            SampleType[] res = new SampleType[2];

      res[0] = new SampleType();
      res[1] = new SampleType();

      res[0].Field1 = DateTime.Now.AddDays(1);
      res[0].Field2 = "je";
      res[0].Field3 = 0;

      res[1].Field1 = DateTime.Now;
      res[1].Field2 = "ho";
      res[1].Field3 = 2;

      engine.WriteString(res);

      Assert.AreEqual(2, engine.TotalRecords);
      Assert.AreEqual(2, before);
      Assert.AreEqual(2, after);

    }

    
    [Test]
    public void ReadEventsCancelAfter()
    {
      before = 0;
      after = 0;

      engine = new FileHelperEngine(typeof (SampleType));
      engine.AfterReadRecord += new AfterReadRecordHandler(AfterEvent2);

      object[] res = Common.ReadTest(engine, @"Good\test1.txt");

      Assert.AreEqual(0, res.Length);
      Assert.AreEqual(4, engine.TotalRecords);
      Assert.AreEqual(0, before);
      Assert.AreEqual(4, after);
    }

    [Test]
    public void ReadEventsCancelBefore()
    {
      before = 0;
      after = 0;

      engine = new FileHelperEngine(typeof (SampleType));
      engine.BeforeReadRecord += new BeforeReadRecordHandler(BeforeEvent2);

      object[] res = Common.ReadTest(engine, @"Good\test1.txt");

      Assert.AreEqual(0, res.Length);
      Assert.AreEqual(4, engine.TotalRecords);
      Assert.AreEqual(4, before);
      Assert.AreEqual(0, after);
    }
    
    [Test]
    public void ReadEventsCancelAll()
    {
      before = 0;
      after = 0;

      engine = new FileHelperEngine(typeof (SampleType));
      engine.BeforeReadRecord += new BeforeReadRecordHandler(BeforeEvent2);
      engine.AfterReadRecord += new AfterReadRecordHandler(AfterEvent2);

      object[] res = Common.ReadTest(engine, @"Good\test1.txt");

      Assert.AreEqual(0, res.Length);
      Assert.AreEqual(4, engine.TotalRecords);
      Assert.AreEqual(4, before);
      Assert.AreEqual(0, after);
    }
    
    int before = 0;
    int after = 0;

    private void BeforeEvent(EngineBase sender, BeforeReadRecordEventArgs e)
    {
      if (e.RecordLine.StartsWith(" ") || e.RecordLine.StartsWith("-"))
        e.SkipThisRecord = true;

      before++;
    }

    private void AfterEvent(EngineBase sender, AfterReadRecordEventArgs e)
    {
      after++;
    }

    private void engine_BeforeWriteRecord(EngineBase sender, BeforeWriteRecordEventArgs e)
    {
      before++;
    }

    private void engine_AfterWriteRecord(EngineBase sender, AfterWriteRecordEventArgs e)
    {
      after++;
    }
    
    private void AfterEvent2(EngineBase sender, AfterReadRecordEventArgs e)
    {
      e.SkipThisRecord = true;
      after++;
    }

    private void BeforeEvent2(EngineBase sender, BeforeReadRecordEventArgs e)
    {
      e.SkipThisRecord = true;
      before++;
    }

  }
}