python/t/800-explicit-close.py

   1 # libguestfs Python bindings
   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 # Test implicit vs explicit closes of the handle (RHBZ#717786).
  19
  20 import os
  21 import guestfs
  22
  23 g = guestfs.GuestFS ()
  24
  25 g.close ()                      # explicit close
  26 del g                           # implicit close - should be no error/warning
  27
  28 # Expect an exception if we call a method on a closed handle.
  29 g = guestfs.GuestFS ()
  30 g.close ()
  31 try:
  32     g.set_memsize (512)
  33     raise Exception("expected an exception from previous statement")
  34 except guestfs.ClosedHandle:
  35     pass
  36 del g
  37
  38 # Verify that the handle is really being closed by g.close, by setting
  39 # up a close event and testing that it happened.
  40 g = guestfs.GuestFS ()
  41
  42 close_invoked = 0
  43
  44 def close_callback (ev, eh, buf, array):
  45     global close_invoked
  46     close_invoked += 1
  47
  48 g.set_event_callback (close_callback, guestfs.EVENT_CLOSE)
  49
  50 if close_invoked != 0:
  51     raise Exception("close_invoked should be 0")
  52
  53 g.close ()
  54 if close_invoked != 1:
  55     raise Exception("close_invoked should be 1")
  56
  57 del g
  58 if close_invoked != 1:
  59     raise Exception("close_invoked should be 1")