summaryrefslogtreecommitdiffstats
path: root/test/language/transportable.rb
blob: d8b5c10e117e643b0f0e2affc868c9190b98f3d1 (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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#!/usr/bin/env ruby

require File.expand_path(File.dirname(__FILE__) + '/../lib/puppettest')

require 'puppet'
require 'puppet/transportable'
require 'puppettest'
require 'puppettest/parsertesting'
require 'yaml'

class TestTransportable < Test::Unit::TestCase
  include PuppetTest::ParserTesting

  def test_yamldumpobject
    obj = mk_transobject
    obj.to_yaml_properties
    str = nil
    assert_nothing_raised {
      str = YAML.dump(obj)
    }

    newobj = nil
    assert_nothing_raised {
      newobj = YAML.load(str)
    }

    assert(newobj.name, "Object has no name")
    assert(newobj.type, "Object has no type")
  end

  def test_yamldumpbucket
    objects = %w{/etc/passwd /etc /tmp /var /dev}.collect { |d|
      mk_transobject(d)
    }
    bucket = mk_transbucket(*objects)
    str = nil
    assert_nothing_raised {
      str = YAML.dump(bucket)
    }

    newobj = nil
    assert_nothing_raised {
      newobj = YAML.load(str)
    }

    assert(newobj.name, "Bucket has no name")
    assert(newobj.type, "Bucket has no type")
  end

  # Make sure our 'delve' command is working
  def test_delve
    top = mk_transtree do |object, depth, width|
      object.file = :funtest if width % 2 == 1
    end

    objects = []
    buckets = []
    found = []

    count = 0
    assert_nothing_raised {
      top.delve do |object|
        count += 1
        if object.is_a? Puppet::TransBucket
          buckets << object
        else
          objects << object
          if object.file == :funtest
            found << object
          end
        end
      end
    }

    top.flatten.each do |obj|
      assert(objects.include?(obj), "Missing obj #{obj.type}[#{obj.name}]")
    end


          assert_equal(
        found.length,
      top.flatten.find_all { |o| o.file == :funtest }.length,
        
      "Found incorrect number of objects")
  end
end