summaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorLuke Kanies <luke@madstop.com>2008-04-11 11:34:51 -0500
committerLuke Kanies <luke@madstop.com>2008-04-11 11:34:51 -0500
commit4aaad26509b2dc9bd45782ec45231e6ea53a4a37 (patch)
treee0715ba0328f149bf030b63a9ce30e3fb57a5d26 /test
parent2925ad1cb9aa820785afca58c4fb6e34274dadd4 (diff)
downloadpuppet-4aaad26509b2dc9bd45782ec45231e6ea53a4a37.tar.gz
puppet-4aaad26509b2dc9bd45782ec45231e6ea53a4a37.tar.xz
puppet-4aaad26509b2dc9bd45782ec45231e6ea53a4a37.zip
Modified the 'master' handler to use the Catalog class to
compile node configurations, rather than using the Configuration handler, which was never used directly. I removed the Configuration handler as a result. Modified the 'master' handler (responsible for sending configurations to clients) to always return Time.now as its compile date, so configurations will always get recompiled.
Diffstat (limited to 'test')
-rwxr-xr-xtest/network/handler/configuration.rb160
-rwxr-xr-xtest/network/handler/master.rb53
2 files changed, 9 insertions, 204 deletions
diff --git a/test/network/handler/configuration.rb b/test/network/handler/configuration.rb
deleted file mode 100755
index 36c5d9e54..000000000
--- a/test/network/handler/configuration.rb
+++ /dev/null
@@ -1,160 +0,0 @@
-#!/usr/bin/env ruby
-
-require File.dirname(__FILE__) + '/../../lib/puppettest'
-
-require 'puppettest'
-require 'puppet/network/handler/configuration'
-
-class TestHandlerConfiguration < Test::Unit::TestCase
- include PuppetTest
-
- Config = Puppet::Network::Handler.handler(:configuration)
-
- # Check all of the setup stuff.
- def test_initialize
- config = nil
- assert_nothing_raised("Could not create local config") do
- config = Config.new(:Local => true)
- end
-
- assert(config.local?, "Config is not considered local after being started that way")
- end
-
- # Test creation/returning of the interpreter
- def test_interpreter
- config = Config.new
-
- # First test the defaults
- config.expects(:create_interpreter).returns(:interp)
- assert_equal(:interp, config.send(:interpreter), "Did not return the interpreter")
-
- # Now run it again and make sure we get the same thing
- assert_equal(:interp, config.send(:interpreter), "Did not cache the interpreter")
- end
-
- def test_create_interpreter
- config = Config.new(:Local => false)
- args = {}
-
- # Try it first with defaults.
- Puppet::Parser::Interpreter.expects(:new).returns(:interp)
- assert_equal(:interp, config.send(:create_interpreter), "Did not return the interpreter")
- end
-
- # Make sure node objects get appropriate data added to them.
- def test_add_node_data
- # First with no classes
- config = Config.new
-
- fakenode = Object.new
- # Set the server facts to something
- config.instance_variable_set("@server_facts", :facts)
- fakenode.expects(:merge).with(:facts)
- config.send(:add_node_data, fakenode)
-
- # Now try it with classes.
- config.classes = %w{a b}
- list = []
- fakenode = Object.new
- fakenode.expects(:merge).with(:facts)
- fakenode.expects(:classes).returns(list).times(2)
- config.send(:add_node_data, fakenode)
- assert_equal(%w{a b}, list, "Did not add classes to node")
- end
-
- def test_compile
- config = Config.new
-
- # First do a local
- node = mock 'node'
- node.stubs(:name).returns(:mynode)
- node.stubs(:environment).returns(:myenv)
-
- interp = mock 'interpreter'
- interp.stubs(:environment)
- interp.expects(:compile).with(node).returns(:config)
- config.expects(:interpreter).returns(interp)
-
- Puppet.expects(:notice) # The log message from benchmarking
-
- assert_equal(:config, config.send(:compile, node), "Did not return config")
-
- # Now try it non-local
- node = mock 'node'
- node.stubs(:name).returns(:mynode)
- node.stubs(:environment).returns(:myenv)
-
- interp = mock 'interpreter'
- interp.stubs(:environment)
- interp.expects(:compile).with(node).returns(:config)
-
- config = Config.new(:Local => true)
- config.expects(:interpreter).returns(interp)
-
- assert_equal(:config, config.send(:compile, node), "Did not return config")
- end
-
- def test_set_server_facts
- config = Config.new
- assert_nothing_raised("Could not call :set_server_facts") do
- config.send(:set_server_facts)
- end
- facts = config.instance_variable_get("@server_facts")
- %w{servername serverversion serverip}.each do |fact|
- assert(facts.include?(fact), "Config did not set %s fact" % fact)
- end
- end
-
- def test_translate
- # First do a local config
- config = Config.new(:Local => true)
- assert_equal(:plain, config.send(:translate, :plain), "Attempted to translate local config")
-
- # Now a non-local
- config = Config.new(:Local => false)
- assert(! config.local?, "Config wrongly thinks it's local")
- obj = mock 'dumpee'
- yamld = mock 'yaml'
- obj.expects(:to_yaml).with(:UseBlock => true).returns(yamld)
- CGI.expects(:escape).with(yamld).returns(:translated)
- assert_equal(:translated, config.send(:translate, obj), "Did not return translated config")
- end
-
- # Check that we're storing the node freshness into the rails db. Hackilicious.
- def test_update_node_check
- # This is stupid.
- config = Config.new
- node = Object.new
- node.expects(:name).returns(:hostname)
- now = Object.new
- Time.expects(:now).returns(now)
- host = Object.new
- host.expects(:last_freshcheck=).with(now)
- host.expects(:save)
-
- # Only test the case where rails is there
- Puppet[:storeconfigs] = true
- Puppet.features.expects(:rails?).returns(true)
- Puppet::Rails.expects(:connect)
- Puppet::Rails::Host.expects(:find_or_create_by_name).with(:hostname).returns(host)
-
- config.send(:update_node_check, node)
- end
-
- def test_version
- # First try the case where we can't look up the node
- config = Config.new
- node = Object.new
- Puppet::Node.stubs(:find_by_any_name).with(:client).returns(false, node)
- interp = Object.new
- assert_instance_of(Bignum, config.version(:client), "Did not return configuration version")
-
- # And then when we find the node.
- config = Config.new
- config.expects(:update_node_check).with(node)
- interp = Object.new
- interp.expects(:configuration_version).returns(:version)
- config.expects(:interpreter).returns(interp)
- assert_equal(:version, config.version(:client), "Did not return configuration version")
- end
-end
diff --git a/test/network/handler/master.rb b/test/network/handler/master.rb
index 55522237b..e91ec2f47 100755
--- a/test/network/handler/master.rb
+++ b/test/network/handler/master.rb
@@ -13,52 +13,17 @@ class TestMaster < Test::Unit::TestCase
Puppet::Indirector::Indirection.clear_cache
end
- # Make sure that files are reread when they change.
- def test_filereread
- # Start with a normal setting
- Puppet[:filetimeout] = 15
-
- manifest = mktestmanifest()
-
- facts = Puppet::Network::Client.master.facts
- # Store them, so we don't determine frshness based on facts.
- Puppet::Util::Storage.cache(:configuration)[:facts] = facts
-
- file2 = @createdfile + "2"
- @@tmpfiles << file2
-
- client = master = nil
- Puppet[:manifest] = manifest
- assert_nothing_raised() {
- # this is the default server setup
- master = Puppet::Network::Handler.master.new(
- :Local => true
- )
- }
-
- config = master.getconfig({"hostname" => "blah"})
-
- # Cache this value for later
- parse1 = master.freshness("mynode")
-
- sleep 1.5
- # Create a new manifest
- File.open(manifest, "w") { |f|
- f.puts "file { \"%s\": ensure => file }\n" % file2
- }
-
- # Verify that the master doesn't immediately reparse the file; we
- # want to wait through the timeout
- assert_equal(parse1, master.freshness("mynode"), "Master did not wait through timeout")
-
- # Then eliminate it
- Puppet[:filetimeout] = 0
+ def test_freshness_is_always_now
+ master = Puppet::Network::Handler.master.new(
+ :Manifest => tempfile,
+ :UseNodes => true,
+ :Local => true
+ )
- # Now make sure the master does reparse
- #Puppet.notice "%s vs %s" % [parse1, master.freshness]
- assert(parse1 != master.freshness("mynode"), "Master did not reparse file")
+ now1 = mock 'now1'
+ Time.expects(:now).returns(now1)
- assert(master.getconfig({"hostname" => "blah"}) != config, "Did not use reloaded config")
+ assert_equal(master.freshness, now1, "Did not return current time as freshness")
end
# Make sure we're correctly doing clientname manipulations.