just getting started with unit tests for sudo.

1 files changed, 34 insertions, 0 deletions

diff --git a/spec/unit/puppet/util/ec2.rb b/spec/unit/puppet/util/ec2.rb
new file mode 100644
index 0000000..c8757ee
--- /dev/null
+++ b/spec/unit/puppet/util/ec2.rb
@@ -0,0 +1,34 @@
+Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }
+
+require 'puppet/util/ec2'
+
+class Ec2Helper
+  include Puppet::Util::Ec2
+end
+
+# LAK: This way the constants exist, but I expect we'll regret this
+unless Puppet.features.aws?
+  class AWS
+      class EC2
+          class Base
+          end
+      end
+  end
+end
+
+describe Puppet::Util::Ec2 do
+  before do
+    @helper = Ec2Helper.new
+  end
+
+  it "should use AWS::Base to create an EC2 connection" do
+    AWS::EC2::Base.expects(:new).with(:access_key_id => "myuser", :secret_access_key => "mypass")
+    @helper.ec2_connection("myuser", "mypass")
+  end
+
+  it "should call foo and bar when calling baz" do
+    @helper.stubs(:foo).returns "yay"
+    @helper.expects(:bar).with("yay").returns "yip"
+    @helper.baz.should == "yip"
+  end
+end