BadUnitTestContextBuilder.cs

Go to the documentation of this file.

using BadScript2.IO;
using BadScript2.Optimizations.Folding;
using BadScript2.Optimizations.Substitution;
using BadScript2.Parser;
using BadScript2.Parser.Expressions;
using BadScript2.Runtime.Objects;
using BadScript2.Runtime.Objects.Functions;
using BadScript2.Runtime.Objects.Native;
 
namespace BadScript2.Interop.NUnit;
 
public class BadUnitTestContextBuilder
{
    private readonly List<BadNUnitTestCase> m_Cases = new List<BadNUnitTestCase>();
 
    private readonly BadRuntime m_Runtime;
 
    private readonly List<BadFunction> m_Setup = new List<BadFunction>();
 
    private readonly List<BadFunction> m_Teardown = new List<BadFunction>();
 
    public BadUnitTestContextBuilder(BadRuntime runtime)
    {
        BadNUnitConsoleApi api = new BadNUnitConsoleApi();
        api.SetContext(this);
        m_Runtime = runtime.Clone()
            .UseApi(new BadNUnitApi())
            .UseApi(api);
    }
 
 
    public void Register(bool optimizeFolding, bool optimizeSubstitution, params string[] files)
    {
        foreach (string file in files)
        {
            SetupStage(file, optimizeFolding, optimizeSubstitution);
        }
    }
 
    public BadUnitTestContext CreateContext(string workingDir)
    {
        return new BadUnitTestContext(workingDir, m_Cases.ToList(), m_Setup.ToList(), m_Teardown.ToList(), m_Runtime);
    }
 
    public void AddTest(BadFunction function, BadObject testName, bool allowCompile = true)
    {
        string? name = null;
 
        if (testName != BadObject.Null)
        {
            name = (testName as IBadString)?.Value ?? throw new InvalidOperationException("Test name must be a string");
        }
 
        m_Cases.Add(new BadNUnitTestCase(function, name, allowCompile));
    }
 
    public void AddSetup(BadFunction function)
    {
        m_Setup.Add(function);
    }
 
 
    public void AddTeardown(BadFunction function)
    {
        m_Teardown.Add(function);
    }
 
    public void Reset()
    {
        m_Teardown.Clear();
        m_Setup.Clear();
        m_Cases.Clear();
    }
 
 
    private void SetupStage(string file, bool optimizeFolding = false, bool optimizeSubstitution = false)
    {
        //Load expressions
        IEnumerable<BadExpression> expressions = BadSourceParser.Create(file, BadFileSystem.ReadAllText(file)).Parse();
 
        if (optimizeFolding)
        {
            expressions = BadConstantFoldingOptimizer.Optimize(expressions);
        }
 
        if (optimizeSubstitution)
        {
            expressions = BadConstantSubstitutionOptimizer.Optimize(expressions);
        }
 
        m_Runtime.Execute(expressions, Path.GetDirectoryName(BadFileSystem.Instance.GetFullPath(file)) ?? "/");
    }
}