appscale-tools 1.6.0

data/test/tc_app_controller_client.rb

@@ -0,0 +1,10 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'app_controller_client'
+
+require 'test/unit'
+
+
+class TestAppControllerClient < Test::Unit::TestCase
+  def test_nothing
+  end
+end

data/test/tc_appscale_add_keypair.rb

@@ -0,0 +1,44 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'appscale_tools'
+
+
+require 'rubygems'
+require 'flexmock/test_unit'
+
+
+class TestAppScaleAddKeypair < Test::Unit::TestCase
+  def setup
+    @options_no_auto = {}
+    @options_w_auto = {"auto" => "true"}
+  end
+
+  def test_not_having_ssh_copy_id_in_path
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:shell).with("which ssh-keygen").and_return("/usr/bin/ssh-keygen")
+    commonfunctions.should_receive(:shell).with("which ssh-copy-id").and_return("")
+
+    assert_raises(BadConfigurationException) {
+      AppScaleTools.add_keypair(@options_no_auto)
+    }
+  end
+
+  def test_not_having_expect_in_path
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:shell).with("which ssh-keygen").and_return("/usr/bin/ssh-keygen")
+    commonfunctions.should_receive(:shell).with("which ssh-copy-id").and_return("/usr/bin/ssh-copy-id")
+    commonfunctions.should_receive(:shell).with("which expect").and_return("")
+
+    assert_raises(BadConfigurationException) {
+      AppScaleTools.add_keypair(@options_w_auto)
+    }
+
+    # TODO(cgb) - add a test where auto isn't used - it shouldn't throw an error
+  end
+
+  def test_usage_is_up_to_date
+    AppScaleTools::ADD_KEYPAIR_FLAGS.each { |flag|
+      assert_equal(true, AppScaleTools::ADD_KEYPAIR_USAGE.include?("-#{flag}"),
+        "No usage text for #{flag}.")
+    } 
+  end
+end

data/test/tc_appscale_describe_instances.rb

@@ -0,0 +1,69 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'appscale_tools'
+
+
+require 'rubygems'
+require 'flexmock/test_unit'
+
+
+class TestAppScaleDescribeInstances < Test::Unit::TestCase
+  def setup
+    kernel = flexmock(Kernel)
+    kernel.should_receive(:puts).and_return()
+
+    @key = "booscale"
+    @options = { "keyname" => @key }
+
+    # taken from the standard two node deployment
+    node1 = {
+      'public_ip' => 'public_ip1',
+      'private_ip' => 'private_ip1',
+      'jobs' => ['load_balancer', 'shadow', 'db_master', 'zookeeper', 'login', 'memcache', 'rabbitmq_master'],
+      'instance_id' => 'instance_id1',
+      'cloud' => 'cloud1',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+ 
+    @role_info = [node1]
+ 
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:read_file).
+      with(File.expand_path("~/.appscale/locations-#{@key}.json")).
+      and_return(JSON.dump(@role_info))
+
+    # mocks for get_role_info, which will return the node's state
+    appcontroller = flexmock('appcontroller')
+    appcontroller.should_receive(:get_role_info).and_return(@role_info)
+    appcontroller.should_receive(:status).and_return("ALL OK")
+    flexmock(AppControllerClient).should_receive(:new).
+      and_return(appcontroller)
+ 
+    commonfunctions.should_receive(:write_file).with(
+      File.expand_path("~/.appscale/locations-#{@key}.json"),
+      JSON.dump(@role_info)).and_return()
+  end
+
+  def test_all_ok
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:get_secret_key).and_return("")
+    commonfunctions.should_receive(:get_head_node_ip).and_return("127.0.0.1")
+
+    flexmock(AppControllerClient).new_instances { |instance|
+      instance.should_receive(:get_all_public_ips).and_return(["127.0.0.1"])
+      instance.should_receive(:status).and_return("ALL OK")
+    }
+
+    instance_info = AppScaleTools.describe_instances(@options)
+    assert_equal(nil, instance_info[:error])
+    assert_equal(["ALL OK"], instance_info[:result])
+  end
+
+  def test_usage_is_up_to_date
+    AppScaleTools::DESCRIBE_INSTANCES_FLAGS.each { |flag|
+      assert_equal(true, 
+        AppScaleTools::DESCRIBE_INSTANCES_USAGE.include?("-#{flag}"), 
+        "No usage text for #{flag}.")
+    } 
+  end
+end

data/test/tc_appscale_remove_app.rb

@@ -0,0 +1,128 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'appscale_tools'
+
+
+require 'rubygems'
+require 'flexmock/test_unit'
+
+
+class TestAppScaleRemoveApp < Test::Unit::TestCase
+  def setup
+    @key = "appscale"
+    @options = {"appname" => "boo", "keyname" => @key}
+
+    @secret = "booooo"
+    @fake_yaml = {
+      :load_balancer => "127.0.0.1",
+      :shadow => "127.0.0.1",
+      :secret => @secret,
+      :db_master => "node-1",
+      :table => "cassandra",
+      :instance_id => "i-FOOBARZ"
+    }
+    yaml = flexmock(YAML)
+    yaml.should_receive(:load_file).and_return(@fake_yaml)
+
+    # taken from the standard two node deployment
+    node1 = {
+      'public_ip' => 'public_ip1',
+      'private_ip' => 'private_ip1',
+      'jobs' => ['load_balancer', 'shadow', 'db_master', 'zookeeper', 'login', 'memcache', 'rabbitmq_master'],
+      'instance_id' => 'instance_id1',
+      'cloud' => 'cloud1',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+ 
+    node2 = {
+      'public_ip' => 'public_ip2',
+      'private_ip' => 'private_ip2',
+      'jobs' => ['load_balancer', 'db_slave', 'memcache', 'rabbitmq_slave', 'appengine'],
+      'instance_id' => 'instance_id2',
+      'cloud' => 'cloud2',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+ 
+    @role_info = [node1, node2]
+ 
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:read_file).
+      with(File.expand_path("~/.appscale/locations-#{@key}.json")).
+      and_return(JSON.dump(@role_info))
+    commonfunctions.should_receive(:write_file).with(
+      File.expand_path("~/.appscale/locations-#{@key}.json"),
+      JSON.dump(@role_info)).and_return()
+
+    # mocks for get_role_info, which will return the node's state
+    appcontroller = flexmock('appcontroller')
+    appcontroller.should_receive(:get_role_info).and_return(@role_info)
+    appcontroller.should_receive(:get_userappserver_ip).and_return("127.0.0.1")
+    appcontroller.should_receive(:stop_app).and_return("true")
+    appcontroller.should_receive(:app_is_running?).and_return(true, false)
+    flexmock(AppControllerClient).should_receive(:new).
+      and_return(appcontroller)
+    
+    kernel = flexmock(Kernel)
+    kernel.should_receive(:print).and_return()
+    kernel.should_receive(:puts).and_return()
+    kernel.should_receive(:sleep).and_return()
+  end
+
+  def test_no_appname_given
+    options = {"keyname" => @key}
+    assert_raises(AppScaleException) {
+      AppScaleTools.remove_app(options)
+    }
+  end
+
+  def test_user_cancels_removal
+    stdin = flexmock(STDIN)
+    stdin.should_receive(:gets).and_return("no\n")
+
+    assert_raises(AppScaleException) {
+      AppScaleTools.remove_app(@options)
+    }
+  end
+
+  def test_remove_app_that_doesnt_exist
+    stdin = flexmock(STDIN)
+    stdin.should_receive(:gets).and_return("yes\n")
+    
+    file = flexmock(File)
+    file.should_receive(:exists?).and_return(true)
+
+    userappclient = flexmock(UserAppClient)
+    userappclient.new_instances { |instance|
+      instance.should_receive(:does_app_exist?).and_return(false)
+    }
+
+    assert_raises(AppEngineConfigException) {
+      AppScaleTools.remove_app(@options)
+    }
+  end
+
+  def test_remove_app_that_does_exist
+    stdin = flexmock(STDIN)
+    stdin.should_receive(:gets).and_return("yes\n")
+    
+    file = flexmock(File)
+    file.should_receive(:exists?).and_return(true)
+
+    userappclient = flexmock(UserAppClient)
+    userappclient.new_instances { |instance|
+      instance.should_receive(:does_app_exist?).and_return(true)
+    }
+
+    assert_nothing_raised(Exception) {
+      AppScaleTools.remove_app(@options)
+    }
+  end
+
+  def test_usage_is_up_to_date
+    AppScaleTools::REMOVE_APP_FLAGS.each { |flag|
+      assert_equal(true, AppScaleTools::REMOVE_APP_USAGE.include?("-#{flag}"),
+        "No usage text for #{flag}.")
+    } 
+  end
+end

data/test/tc_appscale_reset_pwd.rb

@@ -0,0 +1,156 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'appscale_tools'
+
+require 'test/unit'
+
+
+class TestAppScaleResetPassword < Test::Unit::TestCase
+  def setup
+    @keyname = "booscale"
+    @options = {"keyname" => @keyname}
+
+    kernel = flexmock(Kernel)
+    kernel.should_receive(:print).and_return()
+    kernel.should_receive(:puts).and_return()
+
+    stdout = flexmock(STDOUT)
+    stdout.should_receive(:flush).and_return()
+  end
+
+  def test_appscale_not_running
+    file = flexmock(File)
+    file.should_receive(:exists?).and_return(true)
+
+    @secret = "booooo"
+    @fake_yaml = {:load_balancer => "",
+      :shadow => "127.0.0.1",
+      :secret => @secret,
+      :db_master => "node-1",
+      :table => "cassandra",
+      :instance_id => "i-FOOBARZ"}
+    yaml = flexmock(YAML)
+    yaml.should_receive(:load_file).and_return(@fake_yaml)
+
+    # taken from the standard two node deployment
+    node1 = {
+      'public_ip' => 'public_ip1',
+      'private_ip' => 'private_ip1',
+      'jobs' => ['shadow', 'db_master', 'zookeeper', 'login', 'memcache', 'rabbitmq_master'],
+      'instance_id' => 'instance_id1',
+      'cloud' => 'cloud1',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+ 
+    node2 = {
+      'public_ip' => 'public_ip2',
+      'private_ip' => 'private_ip2',
+      'jobs' => ['db_slave', 'memcache', 'rabbitmq_slave', 'appengine'],
+      'instance_id' => 'instance_id2',
+      'cloud' => 'cloud2',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+
+    @role_info = [node1, node2]
+
+    appcontroller = flexmock('appcontroller')
+    appcontroller.should_receive(:get_role_info).and_return(@role_info)
+    flexmock(AppControllerClient).should_receive(:new).and_return(appcontroller)
+ 
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:read_file).
+      with(File.expand_path("~/.appscale/locations-#{@keyname}.json")).
+      and_return(JSON.dump(@role_info))
+    commonfunctions.should_receive(:write_file).with(
+      File.expand_path("~/.appscale/locations-#{@keyname}.json"),
+      JSON.dump(@role_info)).and_return()
+
+    assert_raises(BadConfigurationException) {
+      AppScaleTools.reset_password(@options)
+    }
+  end
+
+  def test_all_ok
+    file = flexmock(File)
+    file.should_receive(:exists?).and_return(true)
+
+    @secret = "booooo"
+    @fake_yaml = {:load_balancer => "127.0.0.1",
+      :shadow => "127.0.0.1",
+      :secret => @secret,
+      :db_master => "node-1",
+      :table => "cassandra",
+      :instance_id => "i-FOOBARZ"}
+    yaml = flexmock(YAML)
+    yaml.should_receive(:load_file).and_return(@fake_yaml)
+
+    # taken from the standard two node deployment
+    node1 = {
+      'public_ip' => 'public_ip1',
+      'private_ip' => 'private_ip1',
+      'jobs' => ['load_balancer', 'shadow', 'db_master', 'zookeeper', 'login', 'memcache', 'rabbitmq_master'],
+      'instance_id' => 'instance_id1',
+      'cloud' => 'cloud1',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+ 
+    node2 = {
+      'public_ip' => 'public_ip2',
+      'private_ip' => 'private_ip2',
+      'jobs' => ['load_balancer', 'db_slave', 'memcache', 'rabbitmq_slave', 'appengine'],
+      'instance_id' => 'instance_id2',
+      'cloud' => 'cloud2',
+      'creation_time' => nil,
+      'destruction_time' => nil
+    }
+
+    @role_info = [node1, node2]
+ 
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:read_file).
+      with(File.expand_path("~/.appscale/locations-#{@keyname}.json")).
+      and_return(JSON.dump(@role_info))
+
+    @user = "a@a.a"
+    @password = "aaaaaa"
+
+    stdin = flexmock(STDIN)
+    stdin.should_receive(:gets).and_return(@user + "\n", @password + "\n")
+
+    kernel = flexmock(Kernel)
+    kernel.should_receive(:system).with("stty -echo").and_return()
+    kernel.should_receive(:system).with("stty echo").and_return()
+
+    appcontroller = flexmock('appcontroller')
+    appcontroller.should_receive(:get_role_info).and_return(@role_info)
+    appcontroller.should_receive(:get_userappserver_ip).and_return("127.0.0.1")
+    flexmock(AppControllerClient).should_receive(:new).and_return(appcontroller)
+ 
+    commonfunctions = flexmock(CommonFunctions)
+    commonfunctions.should_receive(:read_file).
+      with(File.expand_path("~/.appscale/locations-#{@keyname}.json")).
+      and_return(JSON.dump(@role_info))
+    commonfunctions.should_receive(:write_file).with(
+      File.expand_path("~/.appscale/locations-#{@keyname}.json"),
+      JSON.dump(@role_info)).and_return()
+
+    userappclient = flexmock(UserAppClient)
+    userappclient.new_instances { |instance|
+      instance.should_receive(:change_password).and_return("OK")
+    }
+
+    assert_nothing_raised(Exception) {
+      AppScaleTools.reset_password(@options)
+    }
+  end
+
+  def test_usage_is_up_to_date
+    AppScaleTools::RESET_PASSWORD_FLAGS.each { |flag|
+      assert_equal(true, 
+        AppScaleTools::RESET_PASSWORD_USAGE.include?("-#{flag}"), 
+        "No usage text for #{flag}.")
+    } 
+  end
+end

data/test/tc_appscale_run_instances.rb

@@ -0,0 +1,48 @@
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+require 'appscale_tools'
+
+require 'test/unit'
+
+
+class TestAppScaleRunInstances < Test::Unit::TestCase
+  def setup
+    EC2_ENVIRONMENT_VARIABLES.each { |var|
+      ENV[var] = "BOO"
+    }
+
+    file = flexmock(File)
+    file.should_receive(:exists?).and_return(true)
+  end
+
+  def test_machine_not_set_in_cloud_deployments
+    options = {
+      "infrastructure" => "euca"
+    }
+    assert_raises(BadConfigurationException) {
+      AppScaleTools.run_instances(options)
+    }
+  end
+
+  def test_environment_variables_not_set_in_cloud_deployments
+    options = {
+      "infrastructure" => "euca",
+      "machine" => "emi-ABCDEFG"
+    }
+
+    EC2_ENVIRONMENT_VARIABLES.each { |var|
+      ENV[var] = nil
+    }
+
+    assert_raises(BadConfigurationException) {
+      AppScaleTools.run_instances(options)
+    }
+  end
+
+  def test_usage_is_up_to_date
+    AppScaleTools::RUN_INSTANCES_FLAGS.each { |flag|
+      assert_equal(true, 
+        AppScaleTools::RUN_INSTANCES_USAGE.include?("-#{flag}"), 
+        "No usage text for #{flag}.")
+    } 
+  end
+end