ruby/tests/tc_200_write.rb

   1 # hivex Ruby bindings -*- ruby -*-
   2 # Copyright (C) 2009-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", "hivex"))
  21 require 'hivex'
  22
  23 class TestWrite < Test::Unit::TestCase
  24   def test_write
  25     h = Hivex::open("../images/minimal", {:write => 1})
  26     assert_not_nil (h)
  27
  28     root = h.root()
  29     assert (root)
  30
  31     h.node_add_child(root, "A")
  32     h.node_add_child(root, "B")
  33     b = h.node_get_child(root, "B")
  34     assert (b)
  35
  36     values = [
  37               { :key => "Key1", :type => 3, :value => "ABC" },
  38               { :key => "Key2", :type => 3, :value => "DEF" }
  39              ]
  40     h.node_set_values(b, values)
  41
  42     # Don't actually commit here because that would overwrite
  43     # the original file.
  44     # h.commit()
  45   end
  46 end