Class: Puppet::Test::TestHelper

Inherits:
Object
  • Object
show all
Defined in:
lib/vendor/puppet/test/test_helper.rb

Overview

This class is intended to provide an API to be used by external projects

when they are running tests that depend on puppet core.  This should
allow us to vary the implementation details of managing puppet's state
for testing, from one version of puppet to the next--without forcing
the external projects to do any of that state management or be aware of
the implementation details.

For now, this consists of a few very simple signatures. The plan is

that it should be the responsibility of the puppetlabs_spec_helper
to broker between external projects and this API; thus, if any
hacks are required (e.g. to determine whether or not a particular)
version of puppet supports this API, those hacks will be consolidated in
one place and won't need to be duplicated in every external project.

This should also alleviate the anti-pattern that we’ve been following,

wherein each external project starts off with a copy of puppet core's
test_helper.rb and is exposed to risk of that code getting out of
sync with core.

Since this class will be “library code” that ships with puppet, it does

not use API from any existing test framework such as rspec.  This should
theoretically allow it to be used with other unit test frameworks in the
future, if desired.

Note that in the future this API could potentially be expanded to handle

other features such as "around_test", but we didn't see a compelling
reason to deal with that right now.

Class Method Summary collapse

Class Method Details

.after_all_testsObject

Call this method once, at the end of a test run, when no more tests

will be run.

Returns:

  • nil



40
41
42
# File 'lib/vendor/puppet/test/test_helper.rb', line 40

def self.after_all_tests()

end

.after_each_testObject

Call this method once per test, after execution of each individual test.

Returns:

  • nil



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# File 'lib/vendor/puppet/test/test_helper.rb', line 90

def self.after_each_test()
  clear_settings_after_each()

  Puppet::Util::Storage.clear
  Puppet::Util::ExecutionStub.reset

  # Restore the indirector configuration.  See before hook.
  indirections = Puppet::Indirector::Indirection.send(:class_variable_get, :@@indirections)
  indirections.each do |indirector|
    $saved_indirection_state.fetch(indirector.name, {}).each do |variable, value|
      indirector.instance_variable_set(variable, value)
    end
  end
  $saved_indirection_state = nil


  # Some tests can cause us to connect, in which case the lingering
  # connection is a resource that can cause unexpected failure in later
  # tests, as well as sharing state accidentally.
  # We're testing if ActiveRecord::Base is defined because some test cases
  # may stub Puppet.features.rails? which is how we should normally
  # introspect for this functionality.
  ActiveRecord::Base.remove_connection if defined?(ActiveRecord::Base)

end

.before_all_testsObject

Call this method once, when beginning a test run–prior to running

any individual tests.

Returns:

  • nil



33
34
35
# File 'lib/vendor/puppet/test/test_helper.rb', line 33

def self.before_all_tests()

end

.before_each_testObject

Call this method once per test, prior to execution of each invididual test.

Returns:

  • nil



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
# File 'lib/vendor/puppet/test/test_helper.rb', line 46

def self.before_each_test()
  # We need to preserve the current state of all our indirection cache and
  # terminus classes.  This is pretty important, because changes to these
  # are global and lead to order dependencies in our testing.
  #
  # We go direct to the implementation because there is no safe, sane public
  # API to manage restoration of these to their default values.  This
  # should, once the value is proved, be moved to a standard API on the
  # indirector.
  #
  # To make things worse, a number of the tests stub parts of the
  # indirector.  These stubs have very specific expectations that what
  # little of the public API we could use is, well, likely to explode
  # randomly in some tests.  So, direct access.  --daniel 2011-08-30
  $saved_indirection_state = {}
  indirections = Puppet::Indirector::Indirection.send(:class_variable_get, :@@indirections)
  indirections.each do |indirector|
    $saved_indirection_state[indirector.name] = {
        :@terminus_class => indirector.instance_variable_get(:@terminus_class),
        :@cache_class    => indirector.instance_variable_get(:@cache_class)
    }
  end

  initialize_settings_before_each()

  # Longer keys are secure, but they sure make for some slow testing - both
  # in terms of generating keys, and in terms of anything the next step down
  # the line doing validation or whatever.  Most tests don't care how long
  # or secure it is, just that it exists, so these are better and faster
  # defaults, in testing only.
  #
  # I would make these even shorter, but OpenSSL doesn't support anything
  # below 512 bits.  Sad, really, because a 0 bit key would be just fine.
  Puppet[:req_bits]  = 512
  Puppet[:keylength] = 512

  Puppet::Node::Environment.clear
  Puppet::Parser::Functions.reset

  Puppet.clear_deprecation_warnings
end