EvilBeaver/OneScript: Исполняющая среда скриптов на языке 1С

src/OneScript.Language.Tests/TreeValidatorExtensions.cs

89 строк3 KB

Andrei Ovsiankin

Другая логика обработки директив препроцессора в виде цепочки обработчиков

18 ноя 2020, 21:59

18 ноя 2020, 21:5970d43bf

/*----------------------------------------------------------
This Source Code Form is subject to the terms of the
Mozilla Public License, v.2.0. If a copy of the MPL
was not distributed with this file, You can obtain one
at http://mozilla.org/MPL/2.0/.
----------------------------------------------------------*/

using System;
using System.Linq;
using FluentAssertions;
using OneScript.Language.SyntaxAnalysis;
using OneScript.Language.SyntaxAnalysis.AstNodes;
using Xunit;

namespace OneScript.Language.Tests
{
    public static class TreeValidatorExtensions
    {
        public static SyntaxTreeValidator Is(this SyntaxTreeValidator validator, string type)
        {
            Assert.Equal(type, validator.CurrentNode.Kind.ToString());
            return validator;
        }
        
        public static SyntaxTreeValidator Is(this SyntaxTreeValidator validator, NodeKind type)
        {
            validator.CurrentNode.Kind.Should().Be(type);
            return validator;
        }
        
        public static void Is(this TestAstNode node, NodeKind type)
        {
            node.Kind.Should().Be(type, node.Kind + " is unexpected");
        }
        
        public static SyntaxTreeValidator HasChildNodes(this SyntaxTreeValidator validator, int count)
        {
            Assert.Equal(count, validator.CurrentNode.ChildrenList.Count);
            return validator;
        }
        
        public static SyntaxTreeValidator HasChildNodes(this SyntaxTreeValidator validator)
        {
            Assert.NotEmpty(validator.CurrentNode.ChildrenList);
            return validator;
        }
        
        public static TestAstNode FirstChild(this SyntaxTreeValidator validator)
        {
            if (validator.CurrentNode.ChildrenList.Count < 0)
            {
                throw new Exception("No more children");
            }

            return validator.CurrentNode.ChildrenList[0];
        }

        public static TestAstNode WithNode(this SyntaxTreeValidator validator, NodeKind type)
        {
            var child = validator.FirstChild();
            Assert.Equal(type, child.Kind);

            return child;
        }
        
        public static SyntaxTreeValidator HasNode(this SyntaxTreeValidator validator, NodeKind type)
        {
            var child = validator.CurrentNode.ChildrenList.FirstOrDefault(x => x.Kind == type);
            Assert.NotNull(child);

            return new SyntaxTreeValidator(child);
        }
        
        public static void Equal(this TestAstNode node, string value)
        {
            Assert.Equal(value, node.Value);
        }
        
        public static void Equal(this SyntaxTreeValidator node, string value)
        {
            Assert.Equal(value, node.CurrentNode.Value);
        }

        public static T CastTo<T>(this SyntaxTreeValidator node) where T : BslSyntaxNode
        {
            return (T) node.CurrentNode.RealNode;
        }
    }
}