summaryrefslogtreecommitdiff
path: root/test/dl/test_callback.rb
blob: ed3be661e7605a8156f144f5a8a7a3dfcc248a09 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
require_relative 'test_base'
require_relative '../ruby/envutil'
require 'dl/callback'
require 'dl/func'

module DL
  class TestCallback < TestBase
    include DL

    def test_remove_callback_failed
      assert_equal(false, remove_callback(0, TYPE_VOIDP))
    end

    def test_remove_callback
      addr = set_callback(TYPE_VOIDP, 1) do |str|
        str
      end
      assert remove_callback(addr, TYPE_VOIDP), 'callback removed'
    end

    def test_callback_return_value
      addr = set_callback(TYPE_VOIDP, 1) do |str|
        str
      end
      func = CFunc.new(addr, TYPE_VOIDP, 'test')
      f = Function.new(func, [TYPE_VOIDP])
      ptr = CPtr['blah']
      assert_equal ptr, f.call(ptr)
    end

    def test_callback_return_arbitrary
      foo = 'foo'
      addr = set_callback(TYPE_VOIDP, 1) do |ptr|
        CPtr[foo].to_i
      end
      func = CFunc.new(addr, TYPE_VOIDP, 'test')
      f = Function.new(func, [TYPE_VOIDP])

      ptr = CPtr['foo']
      assert_equal 'foo', f.call(ptr).to_s
    end

    def test_callback_with_string
      called_with = nil
      addr = set_callback(TYPE_VOID, 1) do |str|
        called_with = dlunwrap(str)
      end
      func = CFunc.new(addr, TYPE_VOID, 'test')
      f = Function.new(func, [TYPE_VOIDP])

      f.call(dlwrap('foo'))
      assert_equal 'foo', called_with
    end

    def test_call_callback
      called = false

      addr = set_callback(TYPE_VOID, 1) do |foo|
        called = true
      end

      func = CFunc.new(addr, TYPE_VOID, 'test')
      f = Function.new(func, [TYPE_VOIDP])
      f.call(nil)

      assert called, 'function should be called'
    end
  end
end