ruby/tests/tc_400_events.rb

   1 # libguestfs Ruby bindings -*- ruby -*-
   2 # Copyright (C) 2011 Red Hat Inc.
   3 #
   4 # This program is free software; you can redistribute it and/or modify
   5 # it under the terms of the GNU General Public License as published by
   6 # the Free Software Foundation; either version 2 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program; if not, write to the Free Software
  16 # Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 require 'test/unit'
  19 $:.unshift(File::join(File::dirname(__FILE__), "..", "lib"))
  20 $:.unshift(File::join(File::dirname(__FILE__), "..", "ext", "guestfs"))
  21 require 'guestfs'
  22
  23 class TestLoad < Test::Unit::TestCase
  24   def test_events
  25     g = Guestfs::create()
  26
  27     log = Proc.new {| event, event_handle, buf, array |
  28       if event == Guestfs::EVENT_APPLIANCE
  29         buf.chomp!
  30       end
  31       puts "ruby event logged: event=#{event} eh=#{event_handle} buf='#{buf}' array=#{array}"
  32     }
  33
  34     close_invoked = 0
  35     close = Proc.new {| event, event_handle, buf, array |
  36       close_invoked += 1
  37       log.call(event, event_handle, buf, array)
  38     }
  39
  40     # Grab log, trace and daemon messages into our custom callback.
  41     event_bitmask = Guestfs::EVENT_APPLIANCE | Guestfs::EVENT_LIBRARY |
  42       Guestfs::EVENT_TRACE
  43     g.set_event_callback(log, event_bitmask)
  44
  45     # Check that the close event is called.
  46     g.set_event_callback(close, Guestfs::EVENT_CLOSE)
  47
  48     # Make sure we see some messages.
  49     g.set_trace(1)
  50     g.set_verbose(1)
  51
  52     # Do some stuff.
  53     g.add_drive_ro("/dev/null")
  54     g.set_autosync(1)
  55
  56     if close_invoked != 0
  57       raise "close_invoked should be 0"
  58     end
  59     g.close()
  60     if close_invoked != 1
  61       raise "close_invoked should be 1"
  62     end
  63   end
  64 end