ruby-llvm/ruby-llvm

# frozen_string_literal: true

require "test_helper"
require "tempfile"

class ModuleTestCase < Minitest::Test

  def setup
    LLVM.init_jit
  end

  def simple_function
    run_function([], [], LLVM::Int) do |builder, function, *arguments|
      entry = function.basic_blocks.append
      builder.position_at_end(entry)
      builder.ret(LLVM::Int(1))
    end
  end

  def test_simple_module
    assert_equal 1, simple_function().to_i
  end

  def test_global_variable
    yielded = false

    define_module('test_globals_add') do |mod|
      mod.globals.add(LLVM::Int32, 'i') do |var|
        yielded = true

        assert var.kind_of?(LLVM::GlobalVariable)

        # unnamed_addr
        assert !var.unnamed_addr?
        var.unnamed_addr = true
        assert_predicate var, :unnamed_addr?

        assert (var.dll_storage_class == :default)
        var.dll_storage_class = :dll_import
        assert (var.dll_storage_class == :dll_import)

        # global_constant
        assert !var.global_constant?
        var.global_constant = true
        assert_predicate var, :global_constant?

        assert_output("", "Warning: Passing Integer value to LLVM::GlobalValue#global_constant=(Boolean) is deprecated.\n") do
          var.global_constant = 0
        end
        assert !var.global_constant?
      end
    end

    assert yielded, 'LLVM::Module::GlobalCollection#add takes block'
  end

  def test_to_s
    mod = LLVM::Module.new('test_print')
    assert_equal mod.to_s,
      "; ModuleID = 'test_print'\nsource_filename = \"test_print\"\n"
  end

  def test_dump
    mod = LLVM::Module.new('test_print')
    expected_pattern = /^; ModuleID = 'test_print'$/

    Tempfile.open('test_dump.1') do |tmpfile|
      # debug stream (stderr)
      stderr_old = $stderr.dup
      $stderr.reopen(tmpfile.path, 'a')
      begin
        mod.dump
        $stderr.flush
        assert_match expected_pattern, File.read(tmpfile.path)
      ensure
        $stderr.reopen(stderr_old)
      end
    end
  end

  def test_module_properties
    mod = LLVM::Module.new('mod')

    assert_equal '', mod.triple

    mod.triple = 'x86-linux-gnu'
    assert_equal 'x86-linux-gnu', mod.triple

    assert_equal '', mod.data_layout

    mod.data_layout = 'e-p:32:32'
    assert_equal 'e-p:32:32', mod.data_layout
  end

  def test_clone
    mod1 = LLVM::Module.new('mod')
    mod1.globals.add(LLVM::Int32, 'a')
    assert_match "@a = external global i32", mod1.to_s
    mod2 = mod1.clone_module
    assert_match "@a = external global i32", mod2.to_s
    mod2.globals.add(LLVM::Int32, 'b')
    assert_match "@b = external global i32", mod2.to_s
    refute_match "@b = external global i32", mod1.to_s
  end

end