bixby-agent 0.3.0

data/test/test_bixby_common.rb

@@ -0,0 +1,18 @@
+require 'helper'
+
+module Bixby
+module Test
+
+  class BixbyCommon < TestCase
+
+    def test_common_loaded
+      assert_equal(JsonRequest, JsonRequest.new(nil, nil).class)
+      assert_equal(BundleNotFound, BundleNotFound.new.class)
+      assert_equal(CommandNotFound, CommandNotFound.new.class)
+      assert_equal(CommandSpec, CommandSpec.new.class)
+    end
+
+  end
+
+end
+end

data/test/test_crypto.rb

@@ -0,0 +1,78 @@
+
+require 'helper'
+
+module Bixby
+module Test
+
+class Crypto < TestCase
+
+  def test_keygen
+    create_new_agent()
+
+    assert File.exists? @agent.private_key_file
+    assert @agent.private_key
+    assert @agent.public_key
+  end
+
+  def test_server_key
+    setup_existing_agent()
+
+    assert @agent.have_server_key?
+    assert @agent.server_key
+    assert @agent.encrypt_for_server("foobar")
+  end
+
+  def test_decrypt
+    setup_existing_agent()
+    input = encrypt_for_agent("foobar")
+    assert_equal "foobar", @agent.decrypt_from_server(input)
+  end
+
+  def test_decrypt_invalid_hmac
+    setup_existing_agent()
+    input = encrypt_for_agent("foobar")
+
+    # mangle the hmac
+    s = StringIO.new(input)
+    test = s.readline
+    test += "32" + s.readline
+    test += s.read
+
+    assert_throws Bixby::EncryptionError, "hmac verification failed" do
+      @agent.decrypt_from_server(test)
+    end
+  end
+
+  # This test is the same as Bixby::Test::Provisioning.test_list_files except
+  # that crypto routines are enabled.
+  def test_api_call_with_crypto
+
+    setup_test_bundle("vendor", "system/provisioning", "get_bundle.rb")
+    begin
+      require @c.command_file
+    rescue Bixby::ConfigException
+    rescue SystemExit
+      # bundle is executed by default so we can ignore an expected exit
+    end
+    ENV["BIXBY_NOCRYPTO"] = "0"
+    setup_existing_agent()
+
+    stub_request(:post, @api_url).with { |req|
+      req.headers.include?("Authorization") &&
+        req.headers.include?("Date") &&
+        req.headers.include?("Content-Md5") &&
+        req.body =~ /operation/ && req.body =~ /test_bundle/
+    }.to_return(:status => 200, :body => JsonResponse.new(200, nil, {:foo => "bar"}).to_json)
+    Agent.stubs(:create).returns(@agent)
+
+    cmd = CommandSpec.new({ :repo => "support", :bundle => "test_bundle", :command => "echo" })
+    ret = Bixby::Repository.list_files(cmd)
+    assert ret
+    assert_kind_of Hash, ret
+    assert_equal "bar", ret["foo"]
+  end
+
+end
+
+end
+end

data/test/websocket/test_client.rb

@@ -0,0 +1,110 @@
+
+require 'helper'
+require 'stub_eventmachine'
+
+module Bixby
+
+  module WebSocket
+    class APIChannel < Bixby::APIChannel
+      attr_reader :responses # expose for testing
+    end
+  end
+
+module Test
+
+  class TestHandler < Bixby::RpcHandler
+  end
+
+module WebSocket
+class TestClient < TestCase
+
+  class FakeClient
+    attr_reader :types
+    def initialize
+      @types = {}
+    end
+    def on(type, &block)
+      @types[type] = block
+    end
+    def trigger(type, event)
+      @types[type].call(event)
+    end
+    def close
+    end
+  end
+
+  def setup
+    super
+    EM.stub!
+    setup_existing_agent()
+  end
+
+  def teardown
+    EM.disable_stub!
+  end
+
+  def test_connect_and_reconnect
+
+    @fake_ws = FakeClient.new
+    Faye::WebSocket::Client.expects(:new).with("http://localhost", nil, :ping => 55).returns(@fake_ws).times(2)
+    @client = Bixby::WebSocket::Client.new("http://localhost", TestHandler)
+    @client.start
+
+    assert @fake_ws.types[:open]
+    assert @fake_ws.types[:message]
+    assert @fake_ws.types[:close]
+
+    # connect
+    @fake_ws.expects(:send).with{ |r| c = Bixby::WebSocket::Message.from_wire(r); c.type == "connect" }.times(2)
+    @fake_ws.trigger(:open, nil)
+    @client.api.responses.values.first.response = JsonResponse.new("success")
+    assert @client.api.connected?
+
+    # reconnect
+    @fake_ws.trigger(:close, nil)
+    refute @client.api.connected?
+
+    # stop w/o backoff/reconnect
+    @fake_ws.trigger(:open, nil)
+    @client.api.responses.values.first.response = JsonResponse.new("success")
+    assert @client.api.connected?
+
+    EM.expects(:stop_event_loop).once()
+    @client.stop()
+    @fake_ws.trigger(:close, nil)
+    refute @client.api.connected?
+  end
+
+  def test_message
+    @fake_ws = FakeClient.new
+    Faye::WebSocket::Client.expects(:new).with("http://localhost", nil, :ping => 55).returns(@fake_ws)
+    @client = Bixby::WebSocket::Client.new("http://localhost", TestHandler)
+    @client.start
+
+    @client.api.expects(:message).once.with("foobar")
+    @fake_ws.trigger(:message, "foobar")
+  end
+
+  def test_auth_fails_causes_exit
+    Logging.logger[Bixby::WebSocket::Client].level = :error
+    @fake_ws = FakeClient.new
+    Faye::WebSocket::Client.expects(:new).with("http://localhost", nil, :ping => 55).returns(@fake_ws)
+    @client = Bixby::WebSocket::Client.new("http://localhost", TestHandler)
+    @client.start
+
+    @fake_ws.expects(:send).with{ |r| c = Bixby::WebSocket::Message.from_wire(r); c.type == "connect" }
+    @fake_ws.trigger(:open, nil)
+
+    stub_request(:get, "http://google.com/").to_return(:status => 200, :body => "", :headers => {"Date" => (Time.new+1).utc.to_s})
+
+    assert_output(/ntpdate.*exiting since we failed to auth/m) do
+      assert_throws(SystemExit) do
+        @client.api.responses.values.first.response = JsonResponse.new("fail", "request is more than 900 seconds old", nil, 401)
+      end
+    end
+  end
+
+end
+end
+end
+end

metadata

@@ -0,0 +1,557 @@
+--- !ruby/object:Gem::Specification
+name: bixby-agent
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- Chetan Sarva
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-11-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bixby-common
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bixby-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bixby-auth
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: god
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: daemons
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: httpi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: curb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: facter
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: mixlib-cli
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mixlib-shellout
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: uuidtools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: logging
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: growl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hirb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test_guard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: cane
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: coveralls
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: micron
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rb-inotify
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rb-fsevent
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rb-fchange
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Bixby Agent
+email: chetan@pixelcop.net
+executables:
+- bixby-agent
+extensions: []
+extra_rdoc_files:
+- LICENSE
+files:
+- ".document"
+- ".testguardrc"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- Rakefile
+- VERSION
+- bin/bixby-agent
+- bixby-agent.gemspec
+- etc/bixby-god.initd
+- etc/bixby.god
+- etc/god.d/bixby-agent.god
+- lib/bixby-agent.rb
+- lib/bixby-agent/agent.rb
+- lib/bixby-agent/agent/config.rb
+- lib/bixby-agent/agent/crypto.rb
+- lib/bixby-agent/agent/handshake.rb
+- lib/bixby-agent/agent/shell_exec.rb
+- lib/bixby-agent/agent_handler.rb
+- lib/bixby-agent/app.rb
+- lib/bixby-agent/app/cli.rb
+- lib/bixby-agent/config_exception.rb
+- lib/bixby-agent/help/system_time.rb
+- lib/bixby-agent/version.rb
+- lib/bixby-agent/websocket/client.rb
+- tasks/cane.rake
+- tasks/coverage.rake
+- tasks/coveralls.rake
+- tasks/jeweler.rake
+- tasks/test.rake
+- tasks/yard.rake
+- test/base.rb
+- test/helper.rb
+- test/stub_eventmachine.rb
+- test/support/root_dir/bixby.yml
+- test/support/root_dir/id_rsa
+- test/support/root_dir/server
+- test/support/root_dir/server.pub
+- test/support/test_bundle/bin/cat
+- test/support/test_bundle/bin/echo
+- test/support/test_bundle/digest
+- test/support/test_bundle/manifest.json
+- test/test_agent.rb
+- test/test_agent_exec.rb
+- test/test_agent_handler.rb
+- test/test_app.rb
+- test/test_bixby_common.rb
+- test/test_crypto.rb
+- test/websocket/test_client.rb
+homepage: http://github.com/chetan/bixby-agent
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.2
+signing_key: 
+specification_version: 4
+summary: Bixby Agent
+test_files: []