mirror of
				https://github.com/neovim/neovim.git
				synced 2025-11-04 09:44:31 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			221 lines
		
	
	
		
			7.8 KiB
		
	
	
	
		
			Lua
		
	
	
	
	
	
			
		
		
	
	
			221 lines
		
	
	
		
			7.8 KiB
		
	
	
	
		
			Lua
		
	
	
	
	
	
local helpers = require('test.unit.helpers')
 | 
						|
 | 
						|
local cimport = helpers.cimport
 | 
						|
local eq = helpers.eq
 | 
						|
local neq = helpers.neq
 | 
						|
local ffi = helpers.ffi
 | 
						|
local cstr = helpers.cstr
 | 
						|
local to_cstr = helpers.to_cstr
 | 
						|
local NULL = helpers.NULL
 | 
						|
 | 
						|
require('lfs')
 | 
						|
 | 
						|
local env = cimport('./src/nvim/os/os.h')
 | 
						|
 | 
						|
describe('env function', function()
 | 
						|
  local function os_setenv(name, value, override)
 | 
						|
    return env.os_setenv((to_cstr(name)), (to_cstr(value)), override)
 | 
						|
  end
 | 
						|
 | 
						|
  local function os_unsetenv(name, _, _)
 | 
						|
    return env.os_unsetenv((to_cstr(name)))
 | 
						|
  end
 | 
						|
 | 
						|
  local function os_getenv(name)
 | 
						|
    local rval = env.os_getenv((to_cstr(name)))
 | 
						|
    if rval ~= NULL then
 | 
						|
      return ffi.string(rval)
 | 
						|
    else
 | 
						|
      return NULL
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe('os_setenv', function()
 | 
						|
    local OK = 0
 | 
						|
 | 
						|
    it('sets an env variable and returns OK', function()
 | 
						|
      local name = 'NEOVIM_UNIT_TEST_SETENV_1N'
 | 
						|
      local value = 'NEOVIM_UNIT_TEST_SETENV_1V'
 | 
						|
      eq(nil, os.getenv(name))
 | 
						|
      eq(OK, (os_setenv(name, value, 1)))
 | 
						|
      eq(value, os.getenv(name))
 | 
						|
    end)
 | 
						|
 | 
						|
    it("dosn't overwrite an env variable if overwrite is 0", function()
 | 
						|
      local name = 'NEOVIM_UNIT_TEST_SETENV_2N'
 | 
						|
      local value = 'NEOVIM_UNIT_TEST_SETENV_2V'
 | 
						|
      local value_updated = 'NEOVIM_UNIT_TEST_SETENV_2V_UPDATED'
 | 
						|
      eq(OK, (os_setenv(name, value, 0)))
 | 
						|
      eq(value, os.getenv(name))
 | 
						|
      eq(OK, (os_setenv(name, value_updated, 0)))
 | 
						|
      eq(value, os.getenv(name))
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('os_getenv', function()
 | 
						|
    it('reads an env variable', function()
 | 
						|
      local name = 'NEOVIM_UNIT_TEST_GETENV_1N'
 | 
						|
      local value = 'NEOVIM_UNIT_TEST_GETENV_1V'
 | 
						|
      eq(NULL, os_getenv(name))
 | 
						|
      -- need to use os_setenv, because lua dosn't have a setenv function
 | 
						|
      os_setenv(name, value, 1)
 | 
						|
      eq(value, os_getenv(name))
 | 
						|
    end)
 | 
						|
 | 
						|
    it('returns NULL if the env variable is not found', function()
 | 
						|
      local name = 'NEOVIM_UNIT_TEST_GETENV_NOTFOUND'
 | 
						|
      return eq(NULL, os_getenv(name))
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('os_unsetenv', function()
 | 
						|
    it('unsets environment variable', function()
 | 
						|
      local name = 'TEST_UNSETENV'
 | 
						|
      local value = 'TESTVALUE'
 | 
						|
      os_setenv(name, value, 1)
 | 
						|
      os_unsetenv(name)
 | 
						|
      neq(os_getenv(name), value)
 | 
						|
      -- Depending on the platform the var might be unset or set as ''
 | 
						|
      assert.True(os_getenv(name) == nil or os_getenv(name) == '')
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('os_getenvname_at_index', function()
 | 
						|
    it('returns names of environment variables', function()
 | 
						|
      local test_name = 'NEOVIM_UNIT_TEST_GETENVNAME_AT_INDEX_1N'
 | 
						|
      local test_value = 'NEOVIM_UNIT_TEST_GETENVNAME_AT_INDEX_1V'
 | 
						|
      os_setenv(test_name, test_value, 1)
 | 
						|
      local i = 0
 | 
						|
      local names = { }
 | 
						|
      local found_name = false
 | 
						|
      local name = env.os_getenvname_at_index(i)
 | 
						|
      while name ~= NULL do
 | 
						|
        table.insert(names, ffi.string(name))
 | 
						|
        if (ffi.string(name)) == test_name then
 | 
						|
          found_name = true
 | 
						|
        end
 | 
						|
        i = i + 1
 | 
						|
        name = env.os_getenvname_at_index(i)
 | 
						|
      end
 | 
						|
      eq(true, (table.getn(names)) > 0)
 | 
						|
      eq(true, found_name)
 | 
						|
    end)
 | 
						|
 | 
						|
    it('returns NULL if the index is out of bounds', function()
 | 
						|
      local huge = ffi.new('size_t', 10000)
 | 
						|
      local maxuint32 = ffi.new('size_t', 4294967295)
 | 
						|
      eq(NULL, env.os_getenvname_at_index(huge))
 | 
						|
      eq(NULL, env.os_getenvname_at_index(maxuint32))
 | 
						|
 | 
						|
      if ffi.abi('64bit') then
 | 
						|
        -- couldn't use a bigger number because it gets converted to
 | 
						|
        -- double somewere, should be big enough anyway
 | 
						|
        -- maxuint64 = ffi.new 'size_t', 18446744073709551615
 | 
						|
        local maxuint64 = ffi.new('size_t', 18446744073709000000)
 | 
						|
        eq(NULL, env.os_getenvname_at_index(maxuint64))
 | 
						|
      end
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('os_get_pid', function()
 | 
						|
    it('returns the process ID', function()
 | 
						|
      local stat_file = io.open('/proc/self/stat')
 | 
						|
      if stat_file then
 | 
						|
        local stat_str = stat_file:read('*l')
 | 
						|
        stat_file:close()
 | 
						|
        local pid = tonumber((stat_str:match('%d+')))
 | 
						|
        eq(pid, tonumber(env.os_get_pid()))
 | 
						|
      else
 | 
						|
        -- /proc is not available on all systems, test if pid is nonzero.
 | 
						|
        eq(true, (env.os_get_pid() > 0))
 | 
						|
      end
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('os_get_hostname', function()
 | 
						|
    it('returns the hostname', function()
 | 
						|
      local handle = io.popen('hostname')
 | 
						|
      local hostname = handle:read('*l')
 | 
						|
      handle:close()
 | 
						|
      local hostname_buf = cstr(255, '')
 | 
						|
      env.os_get_hostname(hostname_buf, 255)
 | 
						|
      eq(hostname, (ffi.string(hostname_buf)))
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
 | 
						|
  describe('expand_env_esc', function()
 | 
						|
    it('expands environment variables', function()
 | 
						|
      local name = 'NEOVIM_UNIT_TEST_EXPAND_ENV_ESCN'
 | 
						|
      local value = 'NEOVIM_UNIT_TEST_EXPAND_ENV_ESCV'
 | 
						|
      os_setenv(name, value, 1)
 | 
						|
      -- TODO(bobtwinkles) This only tests UNIX expansions. There should be a
 | 
						|
      -- test for windows as well
 | 
						|
      local input1 = to_cstr('$NEOVIM_UNIT_TEST_EXPAND_ENV_ESCN/test')
 | 
						|
      local input2 = to_cstr('${NEOVIM_UNIT_TEST_EXPAND_ENV_ESCN}/test')
 | 
						|
      local output_buff1 = cstr(255, '')
 | 
						|
      local output_buff2 = cstr(255, '')
 | 
						|
      local output_expected = 'NEOVIM_UNIT_TEST_EXPAND_ENV_ESCV/test'
 | 
						|
      env.expand_env_esc(input1, output_buff1, 255, false, true, NULL)
 | 
						|
      env.expand_env_esc(input2, output_buff2, 255, false, true, NULL)
 | 
						|
      eq(output_expected, ffi.string(output_buff1))
 | 
						|
      eq(output_expected, ffi.string(output_buff2))
 | 
						|
    end)
 | 
						|
 | 
						|
    it('expands ~ once when one is true', function()
 | 
						|
      local input = '~/foo ~ foo'
 | 
						|
      local homedir = cstr(255, '')
 | 
						|
      env.expand_env_esc(to_cstr('~'), homedir, 255, false, true, NULL)
 | 
						|
      local output_expected = ffi.string(homedir) .. "/foo ~ foo"
 | 
						|
      local output = cstr(255, '')
 | 
						|
      env.expand_env_esc(to_cstr(input), output, 255, false, true, NULL)
 | 
						|
      eq(ffi.string(output), ffi.string(output_expected))
 | 
						|
    end)
 | 
						|
 | 
						|
    it('expands ~ every time when one is false', function()
 | 
						|
      local input = to_cstr('~/foo ~ foo')
 | 
						|
      local homedir = cstr(255, '')
 | 
						|
      env.expand_env_esc(to_cstr('~'), homedir, 255, false, true, NULL)
 | 
						|
      homedir = ffi.string(homedir)
 | 
						|
      local output_expected = homedir .. "/foo " .. homedir .. " foo"
 | 
						|
      local output = cstr(255, '')
 | 
						|
      env.expand_env_esc(input, output, 255, false, false, NULL)
 | 
						|
      eq(output_expected, ffi.string(output))
 | 
						|
    end)
 | 
						|
 | 
						|
    it('respects the dstlen parameter without expansion', function()
 | 
						|
      local input = to_cstr('this is a very long thing that will not fit')
 | 
						|
      -- The buffer is long enough to actually contain the full input in case the
 | 
						|
      -- test fails, but we don't tell expand_env_esc that
 | 
						|
      local output = cstr(255, '')
 | 
						|
      env.expand_env_esc(input, output, 5, false, true, NULL)
 | 
						|
      -- Make sure the first few characters are copied properly and that there is a
 | 
						|
      -- terminating null character
 | 
						|
      for i=0,3 do
 | 
						|
        eq(input[i], output[i])
 | 
						|
      end
 | 
						|
      eq(0, output[4])
 | 
						|
    end)
 | 
						|
 | 
						|
    it('respects the dstlen parameter with expansion', function()
 | 
						|
      local varname = to_cstr('NVIM_UNIT_TEST_EXPAND_ENV_ESC_DSTLENN')
 | 
						|
      local varval = to_cstr('NVIM_UNIT_TEST_EXPAND_ENV_ESC_DSTLENV')
 | 
						|
      env.os_setenv(varname, varval, 1)
 | 
						|
      -- TODO(bobtwinkles) This test uses unix-specific environment variable accessing,
 | 
						|
      -- should have some alternative for windows
 | 
						|
      local input = to_cstr('$NVIM_UNIT_TEST_EXPAND_ENV_ESC_DSTLENN/even more stuff')
 | 
						|
      -- The buffer is long enough to actually contain the full input in case the
 | 
						|
      -- test fails, but we don't tell expand_env_esc that
 | 
						|
      local output = cstr(255, '')
 | 
						|
      env.expand_env_esc(input, output, 5, false, true, NULL)
 | 
						|
      -- Make sure the first few characters are copied properly and that there is a
 | 
						|
      -- terminating null character
 | 
						|
      -- expand_env_esc SHOULD NOT expand the variable if there is not enough space to
 | 
						|
      -- contain the result
 | 
						|
      for i=0,3 do
 | 
						|
        eq(output[i], input[i])
 | 
						|
      end
 | 
						|
      eq(output[4], 0)
 | 
						|
    end)
 | 
						|
  end)
 | 
						|
end)
 |