ruby/tests/tc_210_setvalue.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 TestSetValue < Test::Unit::TestCase
  24   def test_set_value
  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, "B")
  32     b = h.node_get_child(root, "B")
  33
  34     values = [
  35               { :key => "Key1", :type => 3, :value => "ABC" },
  36               { :key => "Key2", :type => 2, :value => "DEF" }
  37              ]
  38     h.node_set_values(b, values)
  39
  40     value1 = { :key => "Key3", :type => 3, :value => "GHI" }
  41     h.node_set_value(b, value1)
  42
  43     value2 = { :key => "Key1", :type => 3, :value => "JKL" }
  44     h.node_set_value(b, value2)
  45
  46     val = h.node_get_value(b, "Key1")
  47     hash = h.value_value(val)
  48     assert (hash[:type] == 3)
  49     assert (hash[:value] == "JKL")
  50     assert (hash[:len] == 3)
  51
  52     val = h.node_get_value(b, "Key2")
  53     hash = h.value_value(val)
  54     assert (hash[:type] == 2)
  55     assert (hash[:value] == "DEF")
  56     assert (hash[:len] == 3)
  57
  58     val = h.node_get_value(b, "Key3")
  59     hash = h.value_value(val)
  60     assert (hash[:type] == 3)
  61     assert (hash[:value] == "GHI")
  62     assert (hash[:len] == 3)
  63
  64     # Don't actually commit here because that would overwrite
  65     # the original file.
  66     # h.commit()
  67   end
  68 end