evilhornets 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f2eb8b072d3546ff3b3252cec33ada70b49e7279
-  data.tar.gz: d8cad9948b5b5affda0e054f1ef594b17dc0770c
+  metadata.gz: c10bfd3c02ad6e2a2e9a733472153138070848d0
+  data.tar.gz: 66f5a60a21c35600d2805cfcaa6d68b902e38008
 SHA512:
-  metadata.gz: a8e5645179e857ab39824dcf742b81f75c9c4c67051a1ce6d027102ffaaab38ff983d9644ad9b73e990b8683715442a2426b1d99093e62b9e91a1d61c24b2973
-  data.tar.gz: 18d0774defbaed47e8b39a4e542cf1f628be92537747bd11d793188d3c9b4151f3290244ad3ed5e15ba82c63eaa8657e8ba8d8cb1ab6a9ca6b7713a33528657a
+  metadata.gz: 9b1d2874b8531d0ab4522b8ffe76e9d5a1fce0cb42697f29eff01f5fffaf079917b5618c1e3dd9193ebf5e761454268e400df87f2852ee295940149516016aad
+  data.tar.gz: ae6ada4ebb63dd822023c9fdc498d82f2226a315a34744f667cc4bf10e54bb1d811faf7c7a195d2e6225f1cdd8382957ec882060d0ea568f6aa644d4f0cd2492

data/lib/hornet.rb

@@ -10,6 +10,17 @@ Commands = {
   'report' => 'Report the load testing result.',
   'help'   => 'Print the help document.',
 }
+AMIs = {
+  'us-east-1'      => 'ami-fbb35590',
+  'us-west-1'      => 'ami-831cf4c7',
+  'us-west-2'      => 'ami-318eb201',
+  'sa-east-1'      => 'ami-9ba92886',
+  'eu-west-1'      => 'ami-094f3f7e',
+  'eu-central-1'   => 'ami-a67e47bb',
+  'ap-northeast-1' => 'ami-be73a2be',
+  'ap-southeast-1' => 'ami-2c19217e',
+  'ap-southeast-2' => 'ami-9b9de4a1'
+}
 
 class Hornet
 
@@ -101,7 +112,13 @@ class Hornet
           raise ArgumentError.new 'Missing argument: --url'
         end
       when 'up'
-        ops = {:region => 'us-east-1', :username => 'ubuntu', :number => 1, :image_id => 'ami-c3e997f9'}.merge options.to_h
+        ops = {:region => 'us-east-1', :username => 'ubuntu', :number => 1}.merge options.to_h
+        if not ops.has_key? :image_id
+          ops[:image_id] = AMIs[ops[:region]]
+        end
+        if not AMIs.keys.include? ops[:region]
+          raise ArgumentError.new 'Region must be in %s ' % AMIs.keys.join(', ')
+        end
         if not ops.has_key? :key_name
           raise ArgumentError.new 'Missing argument: --key'
         end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: evilhornets
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Andy Zhang