chef-vpc-toolkit 2.1.0 → 2.2.0

data/test/client_test.rb

@@ -0,0 +1,108 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+module ChefVPCToolkit
+module CloudServersVPC
+
+class ClientTest < Test::Unit::TestCase
+
+  include ChefVPCToolkit::CloudServersVPC
+
+  def setup
+    @tmp_dir=TmpDir.new_tmp_dir
+    Client.data_dir=@tmp_dir
+  end
+
+  def teardown
+    FileUtils.rm_rf(@tmp_dir)
+  end
+
+  def test_new
+    client=Client.new(:name => "test", :description => "zz", :status => "Pending")
+    assert_equal "test", client.name
+    assert_equal "zz", client.description
+    assert_equal 0, client.vpn_network_interfaces.size
+  end
+
+  def test_from_xml
+    client=Client.from_xml(CLIENT_XML)
+    assert_equal "local", client.name
+    assert_equal "Toolkit Client: local", client.description
+    assert_equal 5, client.id
+    assert_equal 11, client.server_group_id
+    vni=client.vpn_network_interfaces[0] 
+    assert_not_nil vni.client_key
+    assert_not_nil vni.client_cert
+    assert_not_nil vni.ca_cert
+  end
+
+  def test_client_to_and_from_xml
+    client=Client.from_xml(CLIENT_XML)
+    xml=client.to_xml
+    assert_not_nil xml
+    client=Client.from_xml(xml) 
+    assert_equal "local", client.name
+    assert_equal "Toolkit Client: local", client.description
+    assert_equal 5, client.id
+    assert_equal 11, client.server_group_id
+    vni=client.vpn_network_interfaces[0] 
+    assert_not_nil vni.client_key
+    assert_not_nil vni.client_cert
+    assert_not_nil vni.ca_cert
+  end
+
+  def test_fetch
+
+    tmp_dir=TmpDir.new_tmp_dir
+    File.open("#{tmp_dir}/5.xml", 'w') do |f|
+        f.write(CLIENT_XML)
+    end
+    Client.data_dir=tmp_dir
+
+    Connection.stubs(:get).returns(CLIENT_XML)
+
+    # should raise exception if no ID is set and doing a remote lookup
+    assert_raises(RuntimeError) do
+      Client.fetch
+    end
+
+    client=Client.fetch(:id => "1234")
+    assert_not_nil client
+    assert_equal "Toolkit Client: local", client.description
+
+    client=Client.fetch(:id => "5", :source => "cache")
+    assert_not_nil client
+    assert_equal "Toolkit Client: local", client.description
+
+    #nonexistent group from cache
+    ENV['GROUP_ID']="1234"
+    assert_raises(RuntimeError) do
+      Client.fetch(:source => "cache")
+    end
+
+    #invalid fetch source
+    assert_raises(RuntimeError) do
+      Client.fetch(:id => "5", :source => "asdf")
+    end
+
+  end
+
+  def test_delete
+
+    client=Client.from_xml(CLIENT_XML)
+    client.delete
+    assert_equal false, File.exists?(File.join(Client.data_dir, "#{client.id}.xml"))
+
+  end
+
+  def test_create
+
+    Connection.stubs(:post).returns(CLIENT_XML)
+    client=Client.create(ServerGroup.from_xml(SERVER_GROUP_XML), "local")
+    assert_equal "local", client.name
+
+  end
+
+end
+
+end
+end

data/test/server_group_test.rb

@@ -0,0 +1,259 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+require 'fileutils'
+require 'tempfile'
+
+module ChefVPCToolkit
+module CloudServersVPC
+
+class ServerGroupTest < Test::Unit::TestCase
+
+  def setup
+    @tmp_dir=TmpDir.new_tmp_dir
+    ServerGroup.data_dir=@tmp_dir
+  end
+
+  def teardown
+    FileUtils.rm_rf(@tmp_dir)
+  end
+
+  TEST_JSON_CONFIG = %{{
+        "name": "test",
+        "domain_name": "vpc",
+        "description": "test description",
+        "servers": {
+            "login": {
+                "image_id": "51",
+                "flavor_id": "2",
+                "openvpn_server": "true"
+            },
+            "client1": {
+                "image_id": "69",
+                "flavor_id": "3"
+            }
+        }
+    }}
+
+  def test_server_new
+    sg=ServerGroup.new(:name => "test", :domain_name => "vpc", :description => "zz")
+    assert_equal "test", sg.name
+    assert_equal "zz", sg.description
+    assert_equal "vpc", sg.domain_name
+    assert_equal "172.19.0.0", sg.vpn_network
+    assert_equal "255.255.128.0", sg.vpn_subnet
+  end
+
+  def test_vpn_gateway_ip
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert_equal "184.106.205.120", sg.vpn_gateway_ip
+    assert_equal 1759, sg.id
+    assert_equal "test description", sg.description
+    assert_equal "dan.prince", sg.owner_name
+    assert_equal "172.19.0.0", sg.vpn_network
+    assert_equal "255.255.128.0", sg.vpn_subnet
+    assert_equal 2, sg.servers.size
+  end
+
+  def test_vpn_gateway_name
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert_equal "login1", sg.vpn_gateway_name
+  end
+
+  def test_server_group_from_json_config
+    sg=ServerGroup.from_json_config(TEST_JSON_CONFIG)
+    assert_equal "vpc", sg.domain_name
+    assert_equal "test", sg.name
+    assert_equal "test description", sg.description
+    assert_equal 2, sg.servers.size
+    assert_equal 1, sg.ssh_public_keys.size
+
+    # validate the login server
+    login_server=sg.server("login") 
+    assert_equal 51, login_server.image_id
+    assert_equal 2, login_server.flavor_id
+    assert_equal true, login_server.openvpn_server?
+
+    # validate the client1 server
+    client1_server=sg.server("client1") 
+    assert_equal 69, client1_server.image_id
+    assert_equal 3, client1_server.flavor_id
+    assert_equal false, client1_server.openvpn_server?
+
+  end
+
+  def test_server_group_from_xml
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert_equal "mydomain.net", sg.domain_name
+    assert_equal "test", sg.name
+    assert_equal "test description", sg.description
+    assert_equal 2, sg.servers.size
+    assert_equal 1759, sg.id
+  end
+
+  def test_server_group_to_xml
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert_equal "mydomain.net", sg.domain_name
+    assert_equal "test", sg.name
+    assert_equal "test description", sg.description
+    assert_equal 2, sg.servers.size
+    assert_equal 1759, sg.id
+    xml=sg.to_xml
+  end
+
+  def test_print_server_group
+
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    tmp = Tempfile.open('chef-cloud-toolkit')
+    begin
+        $stdout = tmp
+        sg.pretty_print
+        tmp.flush
+        output=IO.read(tmp.path)
+        $stdout = STDOUT
+        assert output =~ /login1/
+        assert output =~ /test1/
+        assert output =~ /184.106.205.120/
+    ensure
+        $stdout = STDOUT
+    end
+
+  end
+
+  def test_server_names
+
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    names=sg.server_names
+
+    assert_equal 2, names.size
+    assert names.include?("login1")
+    assert names.include?("test1")
+
+  end
+
+  def test_fetch
+
+    tmp_dir=TmpDir.new_tmp_dir
+    File.open("#{tmp_dir}/5.xml", 'w') do |f|
+        f.write(SERVER_GROUP_XML)
+    end
+    ServerGroup.data_dir=tmp_dir
+
+    Connection.stubs(:get).returns(SERVER_GROUP_XML)
+
+    sg=ServerGroup.fetch
+    assert_not_nil sg
+    assert_equal "test", sg.name
+
+    sg=ServerGroup.fetch(:id => "1234")
+    assert_not_nil sg
+    assert_equal "test", sg.name
+
+    sg=ServerGroup.fetch(:id => "5", :source => "cache")
+    assert_not_nil sg
+    assert_equal "test", sg.name
+
+    #nonexistent group from cache
+    ENV['GROUP_ID']="1234"
+    assert_raises(RuntimeError) do
+      ServerGroup.fetch(:source => "cache")
+    end
+
+    #invalid fetch source
+    assert_raises(RuntimeError) do
+      ServerGroup.fetch(:id => "5", :source => "asdf")
+    end
+
+  end
+
+  def test_list_from_cache
+
+    tmp_dir=TmpDir.new_tmp_dir
+    File.open("#{tmp_dir}/1759.xml", 'w') do |f|
+        f.write(SERVER_GROUP_XML)
+    end
+    ServerGroup.data_dir=tmp_dir
+
+    server_groups = ServerGroup.list
+	
+	assert_equal 1, server_groups.size
+	assert_equal 1759, server_groups[0].id
+
+  end
+
+  def test_list_from_remote
+
+    tmp_dir=TmpDir.new_tmp_dir
+    File.open("#{tmp_dir}/1759.xml", 'w') do |f|
+        f.write(SERVER_GROUP_XML)
+    end
+    ServerGroup.data_dir=tmp_dir
+
+    Connection.stubs(:get).returns(SERVER_GROUP_XML)
+    server_groups = ServerGroup.list(:source => "remote")
+	
+	assert_equal 1, server_groups.size
+	assert_equal 1759, server_groups[0].id
+
+  end
+
+  def test_create
+
+    sg=ServerGroup.from_json_config(TEST_JSON_CONFIG)
+
+    Connection.stubs(:post).returns(SERVER_GROUP_XML)
+    sg=ServerGroup.create(sg)
+    assert_not_nil sg
+    assert_equal "mydomain.net", sg.domain_name
+    assert_equal "test", sg.name
+    assert_equal "test description", sg.description
+    assert_equal 2, sg.servers.size
+    assert_equal 1759, sg.id
+
+  end
+
+  def test_most_recent
+
+    File.open("#{ServerGroup.data_dir}/5.xml", 'w') do |f|
+        f.write(SERVER_GROUP_XML)
+    end
+
+    sg=ServerGroup.most_recent
+
+    assert_equal "mydomain.net", sg.domain_name
+    assert_equal 1759, sg.id
+    assert_equal 2, sg.servers.size
+
+  end
+
+  def test_os_types
+
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    os_types=sg.os_types
+
+    assert_equal 2, os_types.size
+    assert_equal "rhel", os_types["login1"]
+    assert_equal "ubuntu", os_types["test1"]
+
+  end
+
+  def test_cache_to_disk
+
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert sg.cache_to_disk
+    assert File.exists?(File.join(ServerGroup.data_dir, "#{sg.id}.xml"))
+
+  end
+
+  def test_delete
+
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    Connection.stubs(:delete).returns("")
+    sg.delete
+    assert_equal false, File.exists?(File.join(ServerGroup.data_dir, "#{sg.id}.xml"))
+
+  end
+
+end
+
+end
+end

data/test/server_test.rb

@@ -0,0 +1,66 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+module ChefVPCToolkit
+module CloudServersVPC
+
+class ServerTest < Test::Unit::TestCase
+
+  include ChefVPCToolkit::CloudServersVPC
+
+  def setup
+    @tmp_dir=TmpDir.new_tmp_dir
+    ServerGroup.data_dir=@tmp_dir
+  end
+
+  def teardown
+    FileUtils.rm_rf(@tmp_dir)
+  end
+
+  def test_vpn_server_rebuild_fails
+    group=ServerGroup.from_xml(SERVER_GROUP_XML)
+    server=group.server("login1")
+    assert_raises(RuntimeError) do
+        server.rebuild
+    end
+  end
+
+  def test_rebuild
+    group=ServerGroup.from_xml(SERVER_GROUP_XML)
+    server=group.server("test1")
+    Connection.stubs(:post).returns("")
+    server.rebuild
+  end
+
+  def test_from_to_xml
+    server=Server.from_xml(SERVER_XML)
+    server=Server.from_xml(server.to_xml)
+    assert_equal "db1", server.name
+    assert_equal "blah", server.description
+    assert_equal 1234, server.id
+    assert_equal 888, server.cloud_server_id_number
+    assert_equal 999, server.server_group_id
+    assert_equal "10.119.225.116", server.internal_ip_addr
+    assert_equal "123.100.100.100", server.external_ip_addr
+    assert_equal "Online", server.status
+  end
+
+  def test_create
+
+    Connection.stubs(:post).returns(SERVER_XML)
+    server=Server.create(Server.from_xml(SERVER_XML))
+    assert_equal "db1", server.name
+
+  end
+
+  def test_delete
+
+    server=Server.from_xml(SERVER_XML)
+    Connection.stubs(:delete).returns("")
+    assert server.delete
+
+  end
+
+end
+
+end
+end

data/test/test_helper.rb

@@ -172,3 +172,24 @@ QRiYX+gvycrIph1yyIGA1qeHYnjhQp4ZijhcwSFUAAyF
   </vpn-network-interfaces>
 </client>
 }
+
+SERVER_XML = %{
+
+<linux-server> 
+  <cloud-server-id-number type="integer">888</cloud-server-id-number> 
+  <created-at type="datetime">2010-07-29T10:27:33-04:00</created-at> 
+  <description>blah</description> 
+  <error-message nil="true"></error-message> 
+  <external-ip-addr>123.100.100.100</external-ip-addr> 
+  <flavor-id type="integer">3</flavor-id> 
+  <id type="integer">1234</id> 
+  <image-id type="integer">14</image-id> 
+  <internal-ip-addr>10.119.225.116</internal-ip-addr> 
+  <name>db1</name> 
+  <openvpn-server type="boolean">false</openvpn-server> 
+  <retry-count type="integer">0</retry-count> 
+  <server-group-id type="integer">999</server-group-id> 
+  <status>Online</status> 
+  <updated-at type="datetime">2010-07-29T11:19:04-04:00</updated-at> 
+</linux-server>
+}