package.path = '../common/libs/?.lua;../common/vendor/?.lua;' .. package.path

local lu = require('luaunit')

require('ReaUtil')

require('mock_reaper')

TestReaUtil = {}

function TestReaUtil:setUp()
  reaper.__test_setUp()
end

function TestReaUtil:testProxyMainOnCommand()
  local proxy = ReaUtil.proxy_main_on_command(1, 0)
  lu.assertEquals(type(proxy), "function")
end

function TestReaUtil:testProxyMainOnCommandProjectArgument()
  local proxy = ReaUtil.proxy_main_on_command(1, 0)

  reaper.Main_OnCommandEx = function(_command_number, _flag, proj)
    lu.assertEquals(proj, 0)
  end
  proxy()

  reaper.Main_OnCommandEx = function(_command_number, _flag, proj)
    lu.assertEquals(proj, 1)
  end
  proxy(1)
end

function TestReaUtil:testProxyMainOnCommandCallsMainOnCommandEx()
  local proxy = ReaUtil.proxy_main_on_command(1, 0)

  local main_on_command_ex_called = false

  reaper.Main_OnCommandEx = function(command_number, flag, _proj)
    main_on_command_ex_called = true
    lu.assertEquals(command_number, 1)
    lu.assertEquals(flag, 0)
  end

  proxy()
  lu.assertEquals(main_on_command_ex_called, true)
end

function TestReaUtil:testDisablerReturnsFunction()
  local disabler = ReaUtil.disabler("imgui context")
  lu.assertEquals(type(disabler), "function")
end

function TestReaUtil:testDisablerDefaultErrorHandler()
  local disabler = ReaUtil.disabler("imgui context")

  local default_handler_called = false

  reaper.ShowConsoleMsg = function(_msg)
    default_handler_called = true
  end

  disabler(true, function() error("error") end)
  lu.assertEquals(default_handler_called, true)
end

function TestReaUtil:testDisablerCustomErrorHandler()
  local custom_handler_called = false
  local disabler = ReaUtil.disabler("imgui context", function(msg)
    custom_handler_called = true
    lu.assertEquals(msg, "error")
  end)

  disabler(true, function() error("error") end)
  lu.assertEquals(custom_handler_called, true)
end

function TestReaUtil:testDisablerContext()
  local disabler = ReaUtil.disabler("imgui context")

    reaper.ImGui_BeginDisabled = function(context, _disabled)
      lu.assertEquals(context, "imgui context")
    end

    reaper.ImGui_EndDisabled = function(context)
      lu.assertEquals(context, "imgui context")
    end

    disabler(true, function() end)
end

function TestReaUtil:testDisablerWrapping()
  local disabler = ReaUtil.disabler("imgui context")

  local begin_marker = false
  local function_called_marker = false
  local end_marker = false

  reaper.ImGui_BeginDisabled = function(_context, _disabled)
    begin_marker = true
  end

  local f = function()
    function_called_marker = true
  end

  reaper.ImGui_EndDisabled = function(_context)
    end_marker = true
  end

  disabler(true, f)
  lu.assertEquals(begin_marker, true)
  lu.assertEquals(function_called_marker, true)
  lu.assertEquals(end_marker, true)

  begin_marker = false
  function_called_marker = false
  end_marker = false

  disabler(false, f)
  lu.assertEquals(begin_marker, false)
  lu.assertEquals(function_called_marker, true)
  lu.assertEquals(end_marker, false)
end

--

os.exit(lu.LuaUnit.run())