ahem/spec/ast/conditional_spec.rb

RSpec.describe AST::Conditional do
  it 'evaluates a true if statement' do
    expect(
      AST::Conditional.new(
        [
          AST::Branch.new(
            AST::Binary.new(
              AST::Operators::GREATER_THAN,
              AST::Number.new(5.0),
              AST::Number.new(4.0),
            ),
            AST::Block.new(
              [
                AST::String.new("5 is greater than 4")
              ]
            )
          )
        ]
      ).execute(Environment.new)
    ).to eq("5 is greater than 4")
  end

  it 'evaluates an else statement' do
    expect(
      AST::Conditional.new(
        [
          AST::Branch.new(
            AST::Binary.new(
              AST::Operators::GREATER_THAN,
              AST::Number.new(3.0),
              AST::Number.new(4.0),
            ),
            AST::Block.new(
              [
                AST::String.new("3 is greater than 4")
              ]
            )
          ),
          AST::Branch.new(
            AST::Boolean.new(true),
            AST::Block.new(
              [
                AST::String.new("3 is _not_ greater than 4")
              ]
            )
          )
        ]
      ).execute(Environment.new)
    ).to eq("3 is _not_ greater than 4")
  end

  it 'evaluates an else if statement' do
    expect(
      AST::Conditional.new(
        [
          AST::Branch.new(
            AST::Binary.new(
              AST::Operators::GREATER_THAN,
              AST::Number.new(3.0),
              AST::Number.new(4.0),
            ),
            AST::Block.new(
              [
                AST::String.new("3 is greater than 4")
              ]
            )
          ),
          AST::Branch.new(
            AST::Binary.new(
              AST::Operators::GREATER_THAN,
              AST::Number.new(3.0),
              AST::Number.new(2.0),
            ),
            AST::Block.new(
              [
                AST::String.new("3 is greater than 2")
              ]
            )
          ),
          AST::Branch.new(
            AST::Boolean.new(true),
            AST::Block.new(
              [
                AST::String.new("3 is _not_ greater than 4")
              ]
            )
          )
        ]
      ).execute(Environment.new)
    ).to eq("3 is greater than 2")
  end
end