wakame-adapters-tengine 0.0.1

data/lib/controllers/controller.rb

@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+
+require 'rubygems'
+require 'right_aws'
+
+module Tama
+  module Controllers
+    A = Tama::Apis
+
+    # A Controller just passes any method calls it gets to its api
+    # The api can be whatever class you want it to be.
+    #
+    # A Controller can contain one api or an array of apis. In case of an array
+    # TamaController checks each of its apis in turn and executes the method
+    # on the first API that supports it
+    class Controller
+      attr_accessor :api
+      
+      def initialize(api)
+        self.api = api
+      end
+      
+      def method_missing(method_name,*args)
+        if self.api.is_a? Array
+          index = 0
+          begin
+            self.api[index].send(method_name,*args)
+          rescue NoMethodError => e
+            index += 1
+            raise if index == self.api.length || (not method_name == e.name)
+            retry
+          end
+        else
+          api.send(method_name)
+        end
+      end
+    end
+    
+    # The TamaController is a controller that has two apis
+    # A RightAws::Ec2 object for handling Amazon Ec2 requests and
+    # a WakameApi for handling requests to Wakame-vdc
+    class TamaController < Controller
+      def initialize(access_key,ec2_host,ec2_port,ec2_protocol,wakame_host,wakame_port,wakame_protocol)
+        super([
+          A::WakameApi.new(access_key,wakame_host,wakame_port,wakame_protocol),
+          RightAws::Ec2.new(access_key,"dummy",{:server => ec2_host,:port => ec2_port,:protocol => ec2_protocol})
+        ])
+      end
+    end
+    
+    # TamaTestController does the same as TamaController but
+    # reads output from a file on disk. It doesn't actually make
+    # any requests to Wakame or Ec2
+    class TamaTestController < TamaController
+      def initialize
+        self.api = [A::WakameApiTest.new,A::Ec2ApiTest.new]
+      end
+    end
+    
+    class ControllerFactory
+      def self.create_controller(*args)
+        case 
+          when args.first == :test || args.first == "test" then
+            TamaTestController.new
+          else TamaController.new(*args)
+        end
+      end
+    end
+  end
+end

data/lib/tama.rb

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+
+require 'rubygems'
+
+load_path = "#{File.expand_path(File.dirname(__FILE__))}/lib"
+$LOAD_PATH.unshift(load_path) unless $LOAD_PATH.include?(load_path)
+
+require 'apis/ec2'
+require 'apis/wakame'
+require 'controllers/controller'
+
+module Tama
+  # Just a wrapper class for our controller to make the syntax a bit more beautiful ^_~
+  class Tama
+    def initialize(*args)
+      @controller = Controllers::ControllerFactory.create_controller(*args)
+    end
+    
+    def method_missing(m,*args)
+      @controller.send(m,*args)
+    end
+  end
+end

data/test/calltama

@@ -0,0 +1,93 @@
+#!/usr/bin/env ruby
+# -*- coding: utf-8 -*-
+
+load_path = "#{File.expand_path(File.dirname(__FILE__))}/../lib"
+$LOAD_PATH.unshift(load_path) unless $LOAD_PATH.include?(load_path)
+
+require 'tama'
+
+def usage
+bin_name = "calltama"
+"Usage:
+  ruby #{bin_name} [test] [command] [options]
+
+Test:
+  Using the word test as the first paremeter causes all output to be read from a file.
+
+Commands:
+  run_instances                      # Start a new instance using the data defined in #{bin_name}.yml.
+  describe_instances [instance_ids]  # Show the alive instances.
+  terminate_instances instance_ids   # Terminate alive instances.
+  describe_images [image_ids]        # Describes images in the database.
+  describe_host_nodes [host_ids]     # Describes host nodes in the database
+  describe_instance_specs [spec_ids] # Describes instance specs in the database
+
+Examples:
+  #{bin_name} test describe_instances
+  #{bin_name} run_instances
+  #{bin_name} terminate instances i-dlobuy33 i-8390cvle
+  #{bin_name} test describe_images wmi-lucid0 wmi-lucid5"
+end
+
+if ARGV[0] == "help" || ARGV[0] == "--help" || ARGV[0] == "-h"
+  puts usage
+  exit 0
+end
+
+begin
+  require 'yaml'
+  config = YAML.load_file File.expand_path('../calltama.yml', __FILE__)
+rescue
+  puts "Warning: Config file (calltama.yml) not found."
+  config = {}
+end
+
+#Arguments for Tama
+Account = config["Account"]
+
+Ec2_host = config["Ec2_host"]
+Ec2_port = config["Ec2_port"]
+Ec2_protocol = config["Ec2_protocol"]
+
+Wakame_host = config["Wakame_host"]
+Wakame_port = config["Wakame_port"]
+Wakame_protocol = config["Wakame_protocol"]
+
+#Put the instances you want to terminate in this array
+Instances_to_terminate = config["Instances_to_terminate"]
+
+#Arguments for run_instances
+Image_id = config["Image_id"]
+Min = config["Min"]
+Max = config["Max"]
+Security_Groups = config["Security_Groups"]
+SSH_Key = config["SSH_Key"]
+User_Data = config["User_Data"]
+Addressing_Type = config["Addressing_Type"]
+Instance_Type = config["Instance_Type"]
+HOST = config["HOST"]
+
+if ARGV[0] == "test"
+  c = Tama::Tama.new(:test)
+  args = ARGV.slice(1..ARGV.length) || []
+else
+  c = Tama::Tama.new(Account,Ec2_host,Ec2_port,Ec2_protocol,Wakame_host,Wakame_port,Wakame_protocol)
+  args = ARGV
+end
+
+if args.empty?
+  [:run_instances,:terminate_instances,:describe_instances,:describe_images,:describe_host_nodes,:describe_instance_specs].each { |method|
+    puts "Testing method: #{method}\n"
+    puts "Testing method: #{method}\n".gsub(/./) {|s| "-"}
+    p case method
+      when :run_instances then c.send(method,Image_id,Min,Max,Security_Groups,SSH_Key, User_Data, Addressing_Type, Instance_Type, nil, nil, HOST)
+      when :terminate_instances then c.send(method,Instances_to_terminate)
+      else c.send(method)
+    end
+    puts "\n"
+  }
+elsif args[0] == "run_instances"
+  p c.run_instances(Image_id,Min,Max,Security_Groups,SSH_Key, User_Data, Addressing_Type, Instance_Type, nil, nil, HOST)
+else
+  p c.send(args[0],args.slice(1..ARGV.length) || [])
+end

data/test/calltama.yml

@@ -0,0 +1,25 @@
+# Arguments for Tama
+Account: "a-shpoolxx"
+
+Ec2_host: "192.168.2.22"
+Ec2_port: "9005"
+Ec2_protocol: "http"
+
+Wakame_host: "192.168.2.22"
+Wakame_port: "9001"
+Wakame_protocol: "http"
+
+#Put the instances you want to terminate in this array
+Instances_to_terminate: []
+
+#Arguments for run_instances
+Image_id: 'wmi-lucid5'
+Min: 1
+Max: 1
+Security_Groups: ['sg-demofgr']#['default', 'joske', 'jefke']
+#Security_Groups: ['jefke']
+SSH_Key: 'ssh-demo'
+User_Data: 'joske'
+Addressing_Type: 'public'
+Instance_Type: 'is-demospec'
+HOST: 'hn-demo1'

data/test/generic_test.rb

@@ -0,0 +1,13 @@
+load_path = "#{File.expand_path(File.dirname(__FILE__))}/.."
+$LOAD_PATH.unshift(load_path) unless $LOAD_PATH.include?(load_path)
+
+require 'tama'
+
+c = Tama::Tama.new(:test)
+
+[:run_instances,:terminate_instances,:describe_instances,:describe_images,:describe_host_nodes,:describe_instance_specs].each { |method|
+  puts "Testing method: #{method}\n"
+  puts "Testing method: #{method}\n".gsub(/./) {|s| "-"}
+  p c.send(method)
+  puts "\n"
+}

data/test/spec/config.yml

@@ -0,0 +1,31 @@
+# Api parameters
+Account: "a-shpoolxx"
+
+Ec2_host: "192.168.2.22"
+Ec2_port: "9005"
+Ec2_protocol: "http"
+
+Wakame_host: "192.168.2.22"
+Wakame_port: "9001"
+Wakame_protocol: "http"
+
+# Run_instance parameters
+Image_id: 'wmi-lucid0'
+Min: 2
+Max: 3
+Security_Groups:
+  sg-demofgr: 'default'
+  #ng-5ltkiagj: 'joske'
+  #ng-cxiouozg: 'jefke'
+SSH_Key: 'ssh-demo'
+User_Data: 'joske'
+Addressing_Type: 'public'
+Instance_Type: 'is-demospec'
+HOST: 'hn-demo1'
+
+describe_host_nodes:
+  [hn-demo1]
+describe_instance_specs:
+  [is-demo2, is-demospec]
+describe_images:
+  [wmi-lucid0, wmi-lucid1, wmi-lucid5, wmi-lucid6]

data/test/spec/real_api_spec.rb

@@ -0,0 +1,39 @@
+load_path = "#{File.expand_path(File.dirname(__FILE__))}/../../lib"
+$LOAD_PATH.unshift(load_path) unless $LOAD_PATH.include?(load_path)
+
+require 'spec_helper'
+require 'tama'
+
+require 'yaml'
+config = YAML.load_file File.expand_path('../config.yml', __FILE__)
+
+describe "TamaController" do
+  tama = Tama::Tama.new(config["Account"],config["Ec2_host"],config["Ec2_port"],config["Ec2_protocol"],config["Wakame_host"],config["Wakame_port"],config["Wakame_protocol"])
+  res = tama.run_instances(config["Image_id"],config["Min"],config["Max"],config["Security_Groups"].keys,config["SSH_Key"], config["User_Data"], config["Addressing_Type"], config["Instance_Type"], nil, nil, config["HOST"])
+  
+  describe "run_instances" do
+    it_should_behave_like "run_instances", res, config
+    it_should_behave_like "run_instances return values", res, config
+  end
+  
+  describe "describe_instances" do
+    @desc_res = tama.describe_instances(res.map{|i| i[:aws_instance_id]})
+    it_should_behave_like "run_instances return values", @desc_res, config
+    it_should_behave_like "describe_instances return values", @desc_res, config
+  end
+  
+  describe "terminate_instances" do
+    @uuids = res.map{|i| i[:aws_instance_id]}
+    it_should_behave_like "terminate_instances", tama, @uuids
+  end
+  
+  describe "describe_images" do
+    it_should_behave_like "describe_images", tama, config["describe_images"]
+  end
+  
+  [:describe_host_nodes,:describe_instance_specs].each { |method|
+    describe "#{method}" do
+      it_should_behave_like "wakame describe method", tama, method, config[method.to_s]
+    end
+  }
+end

data/test/spec/spec_helper.rb

@@ -0,0 +1,101 @@
+require 'rubygems'
+require 'rspec'
+require 'time'
+
+
+shared_examples_for "run_instances" do |result,args|
+  it "Should have started a number of instances between #{args["Min"]} and #{args["Max"]}" do
+    result.length.should >= args["Min"]
+    result.length.should <= args["Max"]
+  end
+end
+
+shared_examples_for "run_instances return values" do |result,args|
+  result.each { |inst_map|
+    it "#{inst_map[:aws_instance_id]} should have a proper time value set for :aws_launch_time" do
+      Time.parse(inst_map[:aws_launch_time]).utc.xmlschema == inst_map[:aws_launch_time]
+    end
+    
+    it "#{inst_map[:aws_instance_id]} should have #{args["Account"]} set as :aws_owner" do
+      inst_map[:aws_owner].should == args["Account"]
+    end
+    
+    it "#{inst_map[:aws_instance_id]} should have #{args["Security_Groups"].keys.inspect} set as :aws_groups" do
+      inst_map[:aws_groups].sort.should == args["Security_Groups"].keys.sort
+    end
+    
+    it "#{inst_map[:aws_instance_id]} should have 0 set as :aws_state_code" do
+      inst_map[:aws_state_code].should == 0
+    end
+    
+    it "#{inst_map[:aws_instance_id]} should have #{args["Image_id"]} set as :aws_image_id" do
+      inst_map[:aws_image_id].should == args["Image_id"]
+    end
+    
+    it "#{inst_map[:aws_instance_id]} should have #{args["Instance_Type"]} set as :aws_instance_type" do
+      inst_map[:aws_instance_type].should == args["Instance_Type"]
+    end
+    
+    #it "#{inst_map[:aws_instance_id]} should have one of #{possible_states.join(",")} set as :aws_state" do
+    
+    #end
+  }
+end
+
+shared_examples_for "describe_instances return values" do |result,args|
+  before(:all) do
+    @uuids = result.map {|i| i[:aws_instance_id]}
+  end
+  
+  result.each { |inst_map|
+    #it "#{inst_map[:aws_instance_id]} should have #{args["HOST"]} set as :aws_availability_zone" do
+      #inst_map[:aws_availablity_zone].should == args["HOST"]
+    #end
+    
+    it "#{inst_map[:aws_instance_id]} should have #{args["SSH_Key"]} set as :ssh_key_name" do
+      inst_map[:ssh_key_name].should == args["SSH_Key"]
+    end
+    
+    #it "#{inst_map[:aws_instance_id]} should have #{args[:arch]} set as :architecture" do
+      #inst_map[:architecture].should == args[:arch]
+    #end
+  }
+end
+
+shared_examples_for "terminate_instances" do |tama,uuids|
+  it "should terminate instances #{uuids.join(",")}" do
+    res = tama.terminate_instances(uuids)
+    
+    res.length.should == uuids.length
+    res.map {|i| i[:aws_instance_id]}.sort.should == uuids.sort
+  end
+end
+
+shared_examples_for "describe_images" do |tama,ids|
+  it "should describe all images" do
+    res = tama.describe_images
+    res.class.should == Array
+  end
+  
+  it "should describe #{ids.join(",")}" do
+    res = tama.describe_images(ids)
+    res.length.should == ids.length
+    res.map {|img| img[:aws_id]}.sort.should == ids.sort
+  end
+end
+
+shared_examples_for "wakame describe method" do |tama,method_name,ids|
+  it "should describe all" do
+    res = tama.send(method_name)
+    res.nil?.should be_false
+    res.class.should == Array
+    #res.empty?.should be_false
+    #res.first["results"].class.should == Array
+  end
+  
+  it "should describe #{ids.join(",")}" do
+    res = tama.send(method_name,ids)
+    res.length.should == ids.length
+    res.map {|node| node["uuid"]}.sort.should == ids.sort
+  end
+end

data/test/test_files/describe_images.json

@@ -0,0 +1,65 @@
+[
+  {
+    "owner_total": 4,
+    "start": 0,
+    "results": [
+      {
+        "updated_at": "2011-10-18T03:53:37Z",
+        "account_id": "a-shpoolxx",
+        "boot_dev_type": 1,
+        "description": "ubuntu-10.04_with-metadata_kvm_i386.raw volume",
+        "source": "--- \n:snapshot_id: snap-lucid6\n",
+        "state": "init",
+        "arch": "x86",
+        "is_public": false,
+        "created_at": "2011-10-18T03:53:37Z",
+        "uuid": "wmi-lucid6",
+        "id": "wmi-lucid6",
+        "md5sum": "8e0d8d37d704751a076fdf704a311350"
+      },
+      {
+        "updated_at": "2011-10-18T03:53:36Z",
+        "account_id": "a-shpoolxx",
+        "boot_dev_type": 2,
+        "description": "ubuntu-10.04_with-metadata_kvm_i386.raw local",
+        "source": "--- \n:uri: file:///home/wakame/work/wakame-vdc/tmp/images/ubuntu-10.04_with-metadata_kvm_i386.raw\n",
+        "state": "init",
+        "arch": "x86",
+        "is_public": false,
+        "created_at": "2011-10-18T03:53:36Z",
+        "uuid": "wmi-lucid5",
+        "id": "wmi-lucid5",
+        "md5sum": "8e0d8d37d704751a076fdf704a311350"
+      },
+      {
+        "updated_at": "2011-10-18T03:53:35Z",
+        "account_id": "a-shpoolxx",
+        "boot_dev_type": 1,
+        "description": "ubuntu-10.04_without-metadata_kvm_i386.raw volume",
+        "source": "--- \n:snapshot_id: snap-lucid1\n",
+        "state": "init",
+        "arch": "x86",
+        "is_public": false,
+        "created_at": "2011-10-18T03:53:35Z",
+        "uuid": "wmi-lucid1",
+        "id": "wmi-lucid1",
+        "md5sum": "8b2a8fb606f5b6ceeca11d9f71e034a8"
+      },
+      {
+        "updated_at": "2011-10-18T03:53:35Z",
+        "account_id": "a-shpoolxx",
+        "boot_dev_type": 2,
+        "description": "ubuntu-10.04_without-metadata_kvm_i386.raw local",
+        "source": "--- \n:uri: file:///home/wakame/work/wakame-vdc/tmp/images/ubuntu-10.04_without-metadata_kvm_i386.raw\n",
+        "state": "init",
+        "arch": "x86",
+        "is_public": false,
+        "created_at": "2011-10-18T03:53:35Z",
+        "uuid": "wmi-lucid0",
+        "id": "wmi-lucid0",
+        "md5sum": "8b2a8fb606f5b6ceeca11d9f71e034a8"
+      }
+    ],
+    "limit": null
+  }
+]