local helpers = require('test.functional.helpers')(after_each)
local Screen = require('test.functional.ui.screen')

local clear = helpers.clear
local command = helpers.command
local feed = helpers.feed

describe("update_menu notification", function()

  local screen

  before_each(function()
    clear()
    screen = Screen.new()
    screen:attach()
  end)

  local function expect_sent(expected)
    screen:expect{condition=function()
      if screen.update_menu ~= expected then
        if expected then
          error('update_menu was expected but not sent')
        else
          error('update_menu was sent unexpectedly')
        end
      end
    end, unchanged=(not expected)}
  end

  it("should be sent when adding a menu", function()
    command('menu Test.Test :')
    expect_sent(true)
  end)

  it("should be sent when deleting a menu", function()
    command('menu Test.Test :')
    screen.update_menu = false

    command('unmenu Test.Test')
    expect_sent(true)
  end)

  it("should not be sent unnecessarily", function()
    feed('i12345<ESC>:redraw<CR>')
    expect_sent(false)
  end)

end)