kytoon 1.0.0

data/lib/kytoon/xml_util.rb

@@ -0,0 +1,15 @@
+module Kytoon
+
+module XMLUtil
+
+  def self.element_text(dom, name)
+    if dom.elements[name]
+      return dom.elements[name].text
+    else
+      return nil
+    end
+  end
+
+end
+
+end

data/rake/kytoon.rake

@@ -0,0 +1,115 @@
+include Kytoon
+
+namespace :group do
+  TMP_SG=File.join(KYTOON_PROJECT, 'tmp', 'server_groups')
+
+  directory TMP_SG
+
+  task :init => [TMP_SG] do
+      ServerGroup.init
+    end
+
+  desc "Create a new group of cloud servers"
+  task :create => [ "init" ] do
+    sg = ServerGroup.create
+    puts "Server group ID #{sg.id} created."
+  end
+
+  desc "List existing cloud server groups."
+  task :list => "init" do
+
+    server_groups=nil
+    server_groups=ServerGroup.index(:source => "cache")
+    if server_groups.size > 0
+      puts "Server groups:"
+      server_groups.sort { |a,b| b.id <=> a.id }.each do |sg|
+        gw=sg.gateway_ip.nil? ? "" : " (#{sg.gateway_ip})"
+        puts "\t :id => #{sg.id}, :name => #{sg.name} #{gw}"
+      end
+    else
+      puts "No server groups."
+    end
+
+  end
+
+  desc "Print information for a cloud server group"
+  task :show => [ "init" ] do
+    sg = ServerGroup.get
+    sg.pretty_print
+  end
+
+  desc "Delete a cloud server group"
+  task :delete => ["init"] do
+
+    sg = ServerGroup.get
+    puts "Deleting cloud server group ID: #{sg.id}."
+    sg.delete
+    SshUtil.remove_known_hosts_ip(sg.gateway_ip)
+
+  end
+
+  desc "Print the VPN gateway IP address"
+  task :gateway_ip do
+    group = ServerGroup.get
+    puts group.gateway_ip
+  end
+
+end
+
+desc "SSH into the most recently created VPN gateway server."
+task :ssh => 'group:init' do
+
+  sg=ServerGroup.get
+  args=ARGV[1, ARGV.length].join(" ")
+  if (ARGV[1] and ARGV[1] =~ /^GROUP_.*/) and (ARGV[2] and ARGV[2] =~ /^GROUP_.*/)
+    args=ARGV[3, ARGV.length].join(" ")
+  elsif ARGV[1] and ARGV[1] =~ /^GROUP_.*/
+    args=ARGV[2, ARGV.length].join(" ")
+  end
+  exec("ssh -o \"StrictHostKeyChecking no\" root@#{sg.gateway_ip} #{args}")
+end
+
+desc "Print help and usage information"
+task :usage do
+
+  puts ""
+  puts "Kytoon Toolkit Version: #{Kytoon::Version::VERSION}"
+  puts ""
+  puts "The following tasks are available:"
+
+  puts %x{cd #{KYTOON_PROJECT} && rake -T}
+  puts "----"
+  puts "Example commands:"
+  puts ""
+  puts "\t- Create a new server group."
+  puts ""
+  puts "\t\t$ rake group:create"
+
+  puts ""
+  puts "\t- List your currently running server groups."
+  puts ""
+  puts "\t\t$ rake group:list"
+
+  puts ""
+  puts "\t- List all remote groups using a common Cloud Servers VPC account."
+  puts ""
+  puts "\t\t$ rake group:list"
+
+  puts ""
+  puts "\t- SSH into the current (most recently created) server group."
+  puts ""
+  puts "\t\t$ rake ssh"
+
+  puts ""
+  puts "\t- SSH into a server group with an ID of 3."
+  puts ""
+  puts "\t\t$ rake ssh GROUP_ID=3"
+
+  puts ""
+  puts "\t- Delete the server group with an ID of 3."
+  puts ""
+  puts "\t\t$ rake group:delete GROUP_ID=3"
+
+end
+
+task :default => 'usage'

data/test/client_test.rb

@@ -0,0 +1,111 @@
+$:.unshift File.dirname(__FILE__)
+require 'test_helper'
+
+module Kytoon
+module Providers
+module CloudServersVPC
+
+class ClientTest < Test::Unit::TestCase
+
+  include Kytoon::Providers::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_get
+
+    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.get
+    end
+
+    client=Client.get(:id => "1234")
+    assert_not_nil client
+    assert_equal "Toolkit Client: local", client.description
+
+    client=Client.get(: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.get(:source => "cache")
+    end
+
+    #invalid get source
+    assert_raises(RuntimeError) do
+      Client.get(: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
+end

data/test/helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'kytoon'
+
+class Test::Unit::TestCase
+end

data/test/server_group_test.rb

@@ -0,0 +1,253 @@
+$:.unshift File.dirname(__FILE__)
+require 'test_helper'
+
+require 'fileutils'
+require 'tempfile'
+
+module Kytoon
+module Providers
+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_gateway_ip
+    sg=ServerGroup.from_xml(SERVER_GROUP_XML)
+    assert_equal "184.106.205.120", sg.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(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('kytoon')
+    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_get
+
+    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)
+
+    sg=ServerGroup.get(:source => "cache")
+    assert_not_nil sg
+    assert_equal "test", sg.name
+
+    sg=ServerGroup.get(:id => "1759", :source => "cache")
+    assert_not_nil sg
+    assert_equal "test", sg.name
+
+    #nonexistent group from cache
+    assert_raises(RuntimeError) do
+      ServerGroup.get(:id => "1234", :source => "cache")
+    end
+
+    #invalid get source
+    assert_raises(RuntimeError) do
+      ServerGroup.get(:id => "1759", :source => "asdf")
+    end
+
+  end
+
+  def test_index_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.index
+	
+	assert_equal 1, server_groups.size
+	assert_equal 1759, server_groups[0].id
+
+  end
+
+  def test_index_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.index(:source => "remote")
+	
+	assert_equal 1, server_groups.size
+	assert_equal 1759, server_groups[0].id
+
+  end
+
+  def test_create
+
+    sg=ServerGroup.from_json(TEST_JSON_CONFIG)
+
+    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(:post).returns(SERVER_GROUP_XML)
+    Connection.stubs(:get).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_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
+end