bixby-agent 0.3.0

data/test/support/root_dir/bixby.yml

@@ -0,0 +1,8 @@
+---
+port: 18000
+manager_uri: http://localhost:3000
+uuid: 839b20d8aab74902bd4aa846567abe23
+mac_address: 00:1f:5b:3d:6b:08
+access_key: "12345"
+secret_key: "my_voice_is_my_passport"
+log_level: info

data/test/support/root_dir/id_rsa

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEpAIBAAKCAQEA30PBCsqplS1HmJ91OQEGYuNyZw3Qf7WQ2eAvesPbVecXxBru
+oRditkBeVMivLem2FK6TNao3ps+5gQR6N50pG2sRI+Itxepo2VVUhztxRwysPfED
+8pkCP0LS4uzF11Lb9VIHDdQTvgNmayqGYq5/PuiioKzDkIsHiR3V6FT/sXbGAMRi
+pAmzMDjNa0OVZCbr8zXh76MPL+bVTJHvn0BysNhCdvn5tHSvHPw54y5WAbC5FEie
+5nySThSR0IsFWrOAiE+PPbmHy7YCC7ROIjJDvI9dfesmBn2KH2aesaP/6QLzZsft
+iF/9D4hItQVRTk5yzXYeZFsC4be3sVyO4iOIiQIDAQABAoIBAQDMpoWHx9cEd9nd
+66R+2RKqiIQwNHh+58ZcAGRlKqydpFOjnXBJV29TUdNl8gKpmkYd1Rv5IAsdK+GF
+gr4oxjWM/WpbsxJ0DXeIZ/AweWtPTi2AJbGRqXFm3XEDMzfyNPCMgWEX6uMh1GoI
+J5XBVOQsnP+dOl+U2evRYvjVmzw5xMfM8GKWLj+qcAVyyzHrl4yTu/o6phRwdflh
+805+XC6knuKj7rDfrXnoR1TX6MbMBWMLlvdNB+Hd6ZB+sDOvQezycSJ3nE2AVyqQ
+n1otVbL7xhfKiKrfdGz4d/CcvbrCpDyMLBdkvXBWn35XW334u1SAb8DE89gjF2mn
+1iFCEplRAoGBAPs6iUw0S1t7HT1b/BhdO/lEpirR87WIwrDRiFElpfZlvIliJSgy
+Z6VmbfkrmaXctkHg3VtAVRE9Eb19UMWA+OMNkQ9Du2EIo8SvgCqI4+LMOsZBDDzh
+ilfjgv2Skx8/qCHbHlr/Tlc8Qj/sbz3hDRRCQFYs4zxnmZxr7UGxjk4lAoGBAOOB
+QggYHYlHQatdIShW6YyrWIrDmI4fSaK8YVTLoCYl+/lMZTcb0zPv2awHWmWBZXeG
+Wx0bs8aJun0vbIH7hAH1mjFFtjZ+HhzSBW4H/pKR/DNywD9kVpfuLnKnz6hg6fGg
+5h38D7z/u/zu2LYWKpp9tWiI7YCrgpW3H52ckcmVAoGAe556IL06t3EFGkRA79ww
+vp9BBKD6wsDgzBDOet+DopK9Cvz4O6dRyNl2E72YhTztkc0MHvKw6SKbsBoRvJ2R
+2me6kUBK25xcuVOCRWvGdrr30MYBN5PjdwvEUPHm4uF34fDoJu6ppdNI5ht5QYGT
+5fwTnFPKHNb9hzK0NDciRuECgYAHZOIShUYgehJmUWaMexaoHcGI8QZ6SAYKpfi7
+2spfswhXd+YqPltE6ty2PJ7xgZ0cn19CuO3E3CEVhH8QUm5d6e+S/mMHrl2oLiND
+Cjt55AjAW1BoPwiZc4JK861KGaxA0CvqeZDBYcZoWBLvCBVmZAOLXDfr4Kcj2onK
+g5YxaQKBgQC0UDAALa7yW8W4rHyIOs4u5Fthrk3cYWx8bM7Vy4rOWmXNL1rseD3B
+dwM51dqrnvqW0XtgJlfgCXbzAqFY+xz33fSxqueojA0N7QcgnBy8mm9SE9RPOSVG
+7HWlwi7dbepPK7+3mnLGathE6xcl6EddZiLSvqlWkjJB/asY3iZvyQ==
+-----END RSA PRIVATE KEY-----

data/test/support/root_dir/server

@@ -0,0 +1,27 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIEpAIBAAKCAQEAu9t/atvz1Xzb3cLPkiNXFj25CF7BucYLbEmNG7ctjJuuchiG
+Vm8D0hq2aar6P7DZIrIRYd/Z6iK7F/6e4m2aP2AMkJF1Ypp1I0nADLHB1+A1kyCt
+w61csJYzTdUNlOnE3Qno8s2iLNvRDv7S/pYUw4aJNZG9JyBextpDeRc2q6NTEwU+
+TjvynS2oWAO/te+wbiTpX0uOM5mtgggmHU3b2I+HhQQ1w4XTfivZbhzgB3hKtAAL
+aMI5ycamCK+OMSjZU0YA7xcQbmYXZPgLT29YB11GSqjULOBqVyvJX/AREFQDiLU0
+SKLtx60G6NgBGPwHUvSBxOcAUzlBrtBek6OHMQIDAQABAoIBAFCnH2BYiKpKzfa6
+0fVTGOuimlNeo1vIDFV0ipGwpmys1kuD/gET7pcBAvoA/ynjGoZL3yyfN+bO/HaD
+yeE63zOAYESdPsGTwODipAKPOsMdLsM5D4FoZ9gk1GgBJbCeYPfpbkr4QD10LqB6
+hUnqU4XSuAv4wzYOLsK/KKLPdZhgbQmKfBV3AWaJFGFnZ/ByKakR1hHyleJOJS/y
+fxb6PIx0G5aqMfFYwaEMI67NEBn1WYHp/XDPzdtDHMEQbgHpiIPKzldmX8yRV9Gu
+tsT1P6hrT5dkLvuey6EzDahhf+BB9Bwfi6GoK17awg7fuw7x/A8u3bantkfONM60
+KBtt8hUCgYEA6BpKIq2VuaycbEqBgylXfBKUiM4Z8T3M8tnEET5CnEYZCD8eaJED
+bicIE8BFgMXJ7vGi4taPHaPskQiTUXLGpktoy1YqWwrRzDL2kvfV+HcX9KDJzWtL
+YhLOvschVtl98a62xVyC/j8frIyCE+PWjrVMSzB7l4vAlar+92X7mWsCgYEAzzMA
+Qjt07chSLLxhergxX8XYpOEk4IHZp0SUC+ez3PLtNkB3ybnU6wPWYlgL19YwbuQB
+ShBa4b649G+U3MfG+k8HRRAK7nH4t7qtGNlXXkr1vFgkkhlbGdOu8in2BVK+C/kq
+/XbXAM1FUCfbvIsku/Epjl++0s55AMRLL6CtPNMCgYEA4QUD1MT3gJBBApflIfUF
+tOsKh+tQqVpiBRtsg88ZowouoJkY8QWODs/We9ePSn32/SnU4hhjnbfb1Ow8bt1Y
+i8ZWLbQ20hcWOOM7vFZ8YiuL1QXBP98CsW1MdRb3Uoz/Fj1tQsAi2zz1h0KdpYzg
+6XAJLGgvmkfKCRBDqjR8DUsCgYAUqSpCcHlVaJ9BFLvHp+xMd4cF0emCdZBFjKIS
+mdUmmBatTTpkpCImGSjcf5kWuaX0uMmkIANf9UuuHB5E8arWtOCts07TvgumDnhK
+PQnoPiNI16U16E9cQO51+Vct6rVGrl7qDVqdGtrphj/0Cg/sqbySFNwLW0lXmTtC
+EYyUvwKBgQCYV6AaO/6YtNO9GlYjQdm3EIxwuZTylM58ylPuuaw+lORpJpYqPnOR
+7OgXui6Unh/1UtYDtoISOXNVr/K6T3wMQwZjpn4gRyJFs0ADs7/HoQIo3KeP1sY9
+6rIr8iVsdb/fZW7MaFv4nEMMJPVvzUsHRLA/a+wVTa/Mo/doZkYITQ==
+-----END RSA PRIVATE KEY-----

data/test/support/root_dir/server.pub

@@ -0,0 +1,9 @@
+-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAu9t/atvz1Xzb3cLPkiNX
+Fj25CF7BucYLbEmNG7ctjJuuchiGVm8D0hq2aar6P7DZIrIRYd/Z6iK7F/6e4m2a
+P2AMkJF1Ypp1I0nADLHB1+A1kyCtw61csJYzTdUNlOnE3Qno8s2iLNvRDv7S/pYU
+w4aJNZG9JyBextpDeRc2q6NTEwU+TjvynS2oWAO/te+wbiTpX0uOM5mtgggmHU3b
+2I+HhQQ1w4XTfivZbhzgB3hKtAALaMI5ycamCK+OMSjZU0YA7xcQbmYXZPgLT29Y
+B11GSqjULOBqVyvJX/AREFQDiLU0SKLtx60G6NgBGPwHUvSBxOcAUzlBrtBek6OH
+MQIDAQAB
+-----END PUBLIC KEY-----

data/test/support/test_bundle/bin/cat

@@ -0,0 +1,2 @@
+#!/bin/bash
+cat -

data/test/support/test_bundle/bin/echo

@@ -0,0 +1,2 @@
+#!/bin/sh
+echo $*

data/test/support/test_bundle/digest

@@ -0,0 +1,17 @@
+{
+  "digest": "c1af0e59a74367e83492a7501f6bdd7ed33de005c3f727a302c5ddfafa8c6f70",
+  "files": [
+    {
+      "file": "bin/cat",
+      "digest": "6d14ab63802f3ef51bdc16db1cfeb6df93167fe7a2aa09a6c737293d42c118b2"
+    },
+    {
+      "file": "bin/echo",
+      "digest": "5ca36fa0f9bd59e8169cc379215e6a481363df5b1e745f0442170fe6a56e1118"
+    },
+    {
+      "file": "manifest.json",
+      "digest": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"
+    }
+  ]
+}

data/test/test_agent.rb

@@ -0,0 +1,110 @@
+
+require 'helper'
+
+module Bixby
+module Test
+
+class TestAgent < TestCase
+
+  def setup
+    super
+    ENV["BIXBY_HOME"] = nil
+  end
+
+  def teardown
+    super
+    ENV["BIXBY_HOME"] = nil
+    ENV["BIXBY_LOG"] = nil
+  end
+
+  def test_create_new_agent
+    @agent = create_new_agent()
+    @agent.save_config()
+    assert(@agent.new?)
+    assert( File.exists? File.join(@root_dir, "etc", "bixby.yml") )
+    assert ENV["BIXBY_HOME"]
+    assert_equal ENV["BIXBY_HOME"], @root_dir
+
+    assert_equal 1, Logging::Logger.root.level # default is info
+  end
+
+  def test_load_existing_agent
+    setup_existing_agent()
+    @agent = create_new_agent()
+    assert(!@agent.new?)
+    assert ENV["BIXBY_HOME"]
+    assert_equal ENV["BIXBY_HOME"], @root_dir
+
+    assert_equal 1, Logging::Logger.root.level # info level is in yaml
+  end
+
+  def test_load_existing_agent_override_log
+    ENV["BIXBY_LOG"] = "debug"
+    setup_existing_agent()
+    @agent = create_new_agent()
+    assert(!@agent.new?)
+    assert ENV["BIXBY_HOME"]
+    assert_equal ENV["BIXBY_HOME"], @root_dir
+
+    assert_equal 0, Logging::Logger.root.level
+  end
+
+  def test_load_existing_agent_using_env
+    setup_existing_agent()
+    ENV["BIXBY_HOME"] = @root_dir
+
+    @agent = Agent.create()
+    assert @agent
+    assert(!@agent.new?)
+    assert ENV["BIXBY_HOME"]
+    assert_equal ENV["BIXBY_HOME"], @root_dir
+    assert_equal @root_dir, Bixby.root
+  end
+
+  def test_register_with_manager
+    @agent = create_new_agent()
+
+    response_str = MultiJson.dump({
+                      :data => {:server_key => "-----BEGIN RSA PUBLIC KEY-----",
+                                :access_key => "foo",
+                                :secret_key => "bar"},
+                      :code    => nil,
+                      :status  => "success",
+                      :message => nil })
+
+    # stub out http request
+    stub_request(:post, "#{@manager_uri}/api").with { |req|
+      req.body =~ /inventory:register_agent/ and req.body =~ /"tenant":"pixelcop"/
+
+    }.to_return(:body => response_str, :status => 200)
+
+    response = @agent.register_agent(@manager_uri, @tenant, @password)
+    assert response.status == "success"
+
+    key_file = Bixby.path("etc", "server.pub")
+    assert File.exists? key_file
+    assert File.read(key_file).include? "PUBLIC KEY"
+
+    assert_equal "foo", @agent.access_key
+    assert_equal "bar", @agent.secret_key
+  end
+
+  def test_bad_config
+    setup_existing_agent()
+    File.open(config_file(), 'w') { |f| f.write("foo") }
+    assert_throws(SystemExit) do
+      Agent.create
+    end
+  end
+
+
+  private
+
+  def config_file
+    File.join(@root_dir, "etc", "bixby.yml")
+  end
+
+end
+
+end
+end

data/test/test_agent_exec.rb

@@ -0,0 +1,53 @@
+
+require 'helper'
+
+module Bixby
+module Test
+
+class AgentExec < TestCase
+
+  def setup
+    super
+    setup_test_bundle("support", "test_bundle", "echo",
+      "c1af0e59a74367e83492a7501f6bdd7ed33de005c3f727a302c5ddfafa8c6f70")
+    @agent = create_new_agent()
+  end
+
+  def test_exec_error
+    # throws the first time
+    assert_throws(BundleNotFound) do
+      @agent.shell_exec(@c.to_hash)
+    end
+  end
+
+  def test_exec_pass
+    setup_root()
+    @c.args = "foo bar baz"
+    @c.user = `whoami`.strip
+    @c.group = `groups`.strip.split(/ /).first
+    ret = @agent.shell_exec(@c.to_hash)
+    assert ret
+    assert_kind_of CommandResponse, ret
+    assert_kind_of Fixnum, ret.status
+    assert_equal 0, ret.status
+    assert ret.stdout
+    assert ret.stderr
+    assert_equal "foo bar baz\n", ret.stdout
+    assert_equal "", ret.stderr
+  end
+
+  def test_execute_stdin
+    setup_root()
+    @c.command = "cat"
+    @c.stdin = "hi"
+    ret = @agent.shell_exec(@c.to_hash)
+    assert_equal 0, ret.status
+    assert_equal "hi", ret.stdout
+    assert_equal "", ret.stderr
+  end
+
+
+end
+
+end
+end

data/test/test_agent_handler.rb

@@ -0,0 +1,72 @@
+
+require 'helper'
+
+module Bixby
+module Test
+class TestAgentHandler < TestCase
+
+  def setup
+    super
+    ENV["BIXBY_HOME"] = nil
+
+    setup_existing_agent()
+    @agent = create_new_agent()
+    assert(!@agent.new?)
+    assert ENV["BIXBY_HOME"]
+    assert_equal ENV["BIXBY_HOME"], @root_dir
+
+    # more setup
+    setup_test_bundle("support", "test_bundle", "echo",
+      "c1af0e59a74367e83492a7501f6bdd7ed33de005c3f727a302c5ddfafa8c6f70")
+    setup_root()
+  end
+
+  def teardown
+    ENV["BIXBY_HOME"] = nil
+  end
+
+  def test_handle
+    @c.args = "foobar"
+    req = JsonRequest.new("shell_exec", @c.to_hash)
+
+    res = AgentHandler.new(nil).handle(req) # this is the actual test call
+    assert res
+    assert_kind_of JsonResponse, res
+    assert res.success? # shell exec succeeded
+
+    res = CommandResponse.from_json_response(res)
+    assert_kind_of CommandResponse, res
+    assert res.success? # actual command succeeded
+    assert_equal 0, res.status
+    assert_equal "foobar\n", res.stdout
+    assert_equal "", res.stderr
+  end
+
+  def test_handle_bundle_not_found
+    @c.bundle = "foo"
+    req = JsonRequest.new("shell_exec", @c.to_hash)
+
+    res = AgentHandler.new(nil).handle(req)
+    puts res.to_s
+    refute res.success?
+    assert_equal 404, res.code
+    assert res.message =~ /bundle not found/
+    assert_equal nil, res.data
+  end
+
+  def test_handle_command_not_found
+    @c.command = "foo"
+    req = JsonRequest.new("shell_exec", @c.to_hash)
+
+    res = AgentHandler.new(nil).handle(req)
+    puts res.to_s
+    refute res.success?
+    assert_equal 404, res.code
+    assert res.message =~ /command not found/
+    assert_equal nil, res.data
+  end
+
+
+end
+end
+end

data/test/test_app.rb

@@ -0,0 +1,138 @@
+
+require 'helper'
+require 'timeout'
+
+module Bixby
+module Test
+
+class TestApp < TestCase
+
+  def teardown
+    ENV["BIXBY_LOG"] = nil
+  end
+
+  def test_load_agent
+    ARGV.clear
+    ARGV << "--register" << @manager_uri
+    ARGV << "-d" << @root_dir
+    ARGV << "--tenant" << "mytenant"
+    ARGV << "--password" << "mypass"
+
+    response_str = MultiJson.dump({
+                      :data => {:server_key => "-----BEGIN RSA PUBLIC KEY-----",
+                                :access_key => "foo",
+                                :secret_key => "bar"},
+                      :code    => nil,
+                      :status  => "success",
+                      :message => nil })
+
+    stub_request(:post, "http://localhost:3000/api").
+      to_return(:status => 200, :body => response_str)
+
+    # stub out daemons & sinatra server
+    Daemons.expects(:run_proc).once().with{ |name,opts|
+      name == "bixby-agent" && opts.kind_of?(Hash) &&
+        opts[:dir] == File.join(@root_dir, "var")
+    }
+
+    app = App.new.run!
+
+    assert_requested(:post, @manager_uri + "/api", :times => 1)
+    conf_file = File.join(@root_dir, "etc", "bixby.yml")
+    assert File.exists? conf_file
+    assert File.exists? File.join(@root_dir, "etc", "server.pub")
+
+    # verify yaml config file
+    conf = File.read(conf_file)
+    assert conf
+    assert_includes conf, "uuid"
+    assert_includes conf, "mac_address"
+    assert_includes conf, "access_key"
+    assert_includes conf, "secret_key"
+    assert_includes conf, "log_level"
+    Bixby::Agent::Config::KEYS.each{ |k| assert_includes conf, k }
+
+    assert_includes conf, "foo"
+    assert_includes conf, "bar"
+    assert_kind_of Hash, YAML.load(conf)
+  end
+
+  def test_run_agent
+
+    ARGV.clear
+    ARGV << "--register" # uri should default to bixby.io
+    ARGV << "--debug"
+    ARGV << "--directory" << @root_dir
+    ARGV << "--tenant" << "mytenant"
+    ARGV << "--password" << "mypass"
+
+    Bixby::WebSocket::Client.any_instance.expects(:start).once()
+
+    response_str = MultiJson.dump({
+                      :data => {:server_key => "-----BEGIN RSA PUBLIC KEY-----",
+                                :access_key => "foo",
+                                :secret_key => "bar"},
+                      :code    => nil,
+                      :status  => "success",
+                      :message => nil })
+
+    stub_request(:post, "https://bixby.io/api").
+      to_return(:status => 200, :body => response_str)
+
+    App.new.run!
+
+    assert_equal "debug", ENV["BIXBY_LOG"]
+  end
+
+  def test_run_with_bad_manager_uri
+    ARGV.clear
+    ARGV << "--register" << "asdf"
+    ARGV << "-d" << @root_dir
+
+    assert_throws(ConfigException) do
+      App.new.run!
+    end
+  end
+
+  def test_register_failed
+
+    ARGV.clear
+    ARGV << "-d"
+    ARGV << @root_dir
+    ARGV << @manager_uri
+
+    stub_request(:post, "http://localhost:3000/api").to_return(:status => 200, :body => MultiJson.dump({:status => "fail"}))
+
+    app = App.new
+    assert_throws(SystemExit) do
+      app.load_agent()
+    end
+
+  end
+
+  def test_register_agent_with_old_date
+
+    ARGV.clear
+    ARGV << "--register" # uri should default to bixby.io
+    ARGV << "--debug"
+    ARGV << "--directory" << @root_dir
+    ARGV << "--tenant" << "mytenant"
+    ARGV << "--password" << "mypass"
+
+    stub_request(:get, "http://google.com/").to_return(:status => 200, :body => "", :headers => {"Date" => (Time.new+1).utc.to_s})
+
+    ret = Bixby::JsonResponse.new("fail", "request is more than 900 seconds old", nil, 401).to_json
+    stub_request(:post, "https://bixby.io/api").
+      to_return(:status => 200, :body => ret)
+
+    assert_output(nil, /current time.*sudo ntpdate/m) do
+      assert_throws(SystemExit) do
+        App.new.run!
+      end
+    end
+  end
+
+end
+
+end
+end