adams-visiona 1.0.0

data/spec/io/buffered_file_spec.rb

@@ -0,0 +1,146 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'spec_helper'
+require 'cosmos/io/buffered_file'
+
+module Cosmos
+
+  describe BufferedFile do
+    DATA = "RyanSaysHelloToU"
+
+    before(:all) do
+      @filename = File.join(System.paths['LOGS'], 'test.bin')
+      File.open(@filename, 'wb') do |file|
+        file.write DATA * (2 * BufferedFile::BUFFER_SIZE / DATA.length)
+      end
+    end
+
+    describe "read" do
+      it "reads less than the buffer size" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(8)).to eql DATA[0..7]
+          expect(file.pos).to eql 8
+          expect(file.read(4)).to eql DATA[8..11]
+          expect(file.pos).to eql 12
+          expect(file.read(1)).to eql DATA[12..12]
+          expect(file.pos).to eql 13
+          expect(file.read(3)).to eql DATA[13..15]
+          expect(file.pos).to eql 16
+          expect(file.read(16)).to eql DATA
+          expect(file.pos).to eql 32
+          expected_pos = 32
+          while (expected_pos < (2 * BufferedFile::BUFFER_SIZE))
+            expect(file.read(16)).to eql DATA
+            expected_pos += 16
+            expect(file.pos).to eql expected_pos
+          end
+          expect(file.read(1)).to be_nil
+        end
+      end
+
+      it "handles trying to read past the end" do
+        BufferedFile.open(@filename, "rb") do |file|
+          file.seek(-16, IO::SEEK_END)
+          expect(file.pos).to eql ((2 * BufferedFile::BUFFER_SIZE) - 16)
+          expect(file.read(DATA.length * 10)).to eql(DATA)
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE * 2
+          expect(file.read(DATA.length)).to be_nil
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE * 2
+        end
+      end
+
+      it "reads equal to the buffer size" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(BufferedFile::BUFFER_SIZE)).to eql(DATA * (BufferedFile::BUFFER_SIZE / DATA.length))
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE
+          expect(file.read(BufferedFile::BUFFER_SIZE)).to eql(DATA * (BufferedFile::BUFFER_SIZE / DATA.length))
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE * 2
+          expect(file.read(BufferedFile::BUFFER_SIZE)).to be_nil
+        end
+      end
+
+      it "reads greater than the buffer size" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(BufferedFile::BUFFER_SIZE + 1)).to eql(DATA * (BufferedFile::BUFFER_SIZE / DATA.length) << DATA[0..0])
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE + 1
+          expect(file.read(BufferedFile::BUFFER_SIZE + 1)).to eql((DATA * (BufferedFile::BUFFER_SIZE / DATA.length))[1..-1])
+          expect(file.pos).to eql BufferedFile::BUFFER_SIZE * 2
+          expect(file.read(BufferedFile::BUFFER_SIZE + 1)).to be_nil
+        end
+      end
+
+      it "reads greater than the buffer size after a previous read" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(DATA.length * 10)).to eql(DATA * 10)
+          expect(file.pos).to eql DATA.length * 10
+          expect(file.read(BufferedFile::BUFFER_SIZE + 1)).to eql(DATA * ((BufferedFile::BUFFER_SIZE + 10) / DATA.length) << DATA[0])
+          expect(file.pos).to eql(DATA.length * 10 + BufferedFile::BUFFER_SIZE + 1)
+        end
+      end
+    end
+
+    describe "seek" do
+      it "raises if given more than 2 arguments" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect { file.seek(0, 4, IO::SEEK_CUR) }.to raise_error(ArgumentError)
+        end
+      end
+
+      it "implies SEEK_SET with 1 argument" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(8)).to eql DATA[0..7]
+          expect(file.pos).to eql 8
+          file.seek(4, IO::SEEK_CUR)
+          expect(file.pos).to eql 12
+          file.seek(0)
+          expect(file.pos).to eql 0
+        end
+      end
+
+      it "has reads still work afterwards" do
+        BufferedFile.open(@filename, "rb") do |file|
+          expect(file.read(8)).to eql DATA[0..7]
+          expect(file.pos).to eql 8
+          file.seek(4, IO::SEEK_CUR)
+          expect(file.pos).to eql 12
+          expect(file.read(1)).to eql DATA[12..12]
+          expect(file.pos).to eql 13
+          expect(file.read(3)).to eql DATA[13..15]
+          expect(file.pos).to eql 16
+          file.seek(0, IO::SEEK_SET)
+          expect(file.pos).to eql 0
+          expect(file.read(8)).to eql DATA[0..7]
+          expect(file.pos).to eql 8
+          file.seek(0, IO::SEEK_END)
+          expect(file.pos).to eql (2 * BufferedFile::BUFFER_SIZE)
+          file.seek(-16, IO::SEEK_END)
+          expect(file.pos).to eql ((2 * BufferedFile::BUFFER_SIZE) - 16)
+          expect(file.read(4)).to eql DATA[0..3]
+          expect(file.pos).to eql ((2 * BufferedFile::BUFFER_SIZE) - 12)
+          file.seek(4, IO::SEEK_CUR)
+          expect(file.pos).to eql ((2 * BufferedFile::BUFFER_SIZE) - 8)
+          file.seek(0, IO::SEEK_SET)
+          expect(file.pos).to eql 0
+          expect(file.read(8)).to eql DATA[0..7]
+          expect(file.pos).to eql 8
+          file.seek(4, IO::SEEK_CUR)
+          expect(file.pos).to eql 12
+          expect(file.read(1)).to eql DATA[12..12]
+          expect(file.pos).to eql 13
+          expect(file.read(3)).to eql DATA[13..15]
+          expect(file.pos).to eql 16
+        end
+      end
+    end
+
+  end
+end
+

data/spec/io/io_multiplexer_spec.rb

@@ -0,0 +1,102 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'spec_helper'
+require 'cosmos/io/io_multiplexer'
+
+module Cosmos
+
+  describe IoMultiplexer do
+    before(:each) do
+      @io = IoMultiplexer.new
+    end
+
+    describe "stream_operator" do
+      it "supports the << operator" do
+        @io.add_stream(STDOUT)
+        expect($stdout).to receive(:<<).with("TEST").and_return($stdout)
+        result = (@io << "TEST")
+        expect(result).to eql(@io)
+      end
+    end
+
+    describe "add_stream" do
+      it "adds a single stream" do
+        @io.add_stream(STDOUT)
+        expect($stdout).to receive(:puts).with("TEST")
+        @io.puts "TEST"
+      end
+
+      it "adds multiple streams" do
+        @io.add_stream(STDOUT)
+        @io.add_stream(STDERR)
+        expect($stdout).to receive(:puts).with("TEST")
+        expect($stderr).to receive(:puts).with("TEST")
+        @io.puts "TEST"
+      end
+    end
+
+    describe "remove_stream" do
+      it "removes the stream from output" do
+        @io.add_stream(STDOUT)
+        @io.add_stream(STDERR)
+        @io.remove_stream(STDOUT)
+        expect($stdout).not_to receive(:puts).with("TEST")
+        expect($stderr).to receive(:puts).with("TEST")
+        @io.puts "TEST"
+      end
+    end
+
+    describe "print, printf, putc, puts, flush" do
+      it "defers to the stream" do
+        @io.add_stream(STDOUT)
+        expect($stdout).to receive(:print).with("TEST")
+        @io.print "TEST"
+        expect($stdout).to receive(:printf).with("TEST")
+        @io.printf "TEST"
+        expect($stdout).to receive(:putc).with("TEST")
+        @io.putc "TEST"
+        expect($stdout).to receive(:puts).with("TEST")
+        @io.puts "TEST"
+        expect($stdout).to receive(:flush)
+        @io.flush
+      end
+    end
+
+    describe "write write_nonblock" do
+      it "defers to the stream" do
+        @io.add_stream(STDOUT)
+        expect($stdout).to receive(:write).with("TEST").and_return(4)
+        len = @io.write "TEST"
+        expect(len).to eql 4
+        expect($stdout).to receive(:write_nonblock).with("TEST").and_return(4)
+        len = @io.write_nonblock "TEST"
+        expect(len).to eql 4
+      end
+    end
+
+    describe "remove_default_io" do
+      it "removes STDOUT and STDERR from the streams" do
+        f = File.open("unittest.txt",'w')
+        @io.add_stream(STDOUT)
+        @io.add_stream(STDERR)
+        @io.add_stream(f)
+        @io.remove_default_io
+        @io.puts "TEST"
+        f.close
+        expect($stdout).not_to receive(:puts).with("TEST")
+        expect($stderr).not_to receive(:puts).with("TEST")
+        expect(File.read("unittest.txt")).to eql "TEST\n"
+        File.delete("unittest.txt")
+      end
+    end
+  end
+end
+

data/spec/io/json_drb_object_spec.rb

@@ -0,0 +1,106 @@
+# encoding: ascii-8bit
+
+# Copyright 2014 Ball Aerospace & Technologies Corp.
+# All Rights Reserved.
+#
+# This program is free software; you can modify and/or redistribute it
+# under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 3 with
+# attribution addendums as found in the LICENSE.txt
+
+require 'spec_helper'
+require 'cosmos/io/json_drb_object'
+
+module Cosmos
+
+  describe JsonDRbObject do
+    describe "initialize" do
+      it "rescues bad hosts" do
+        expect { JsonDRbObject.new("blah", 7777) }.to raise_error("Invalid hostname: blah")
+      end
+    end
+
+    describe "method_missing" do
+      it "calls the method on the remote object" do
+        class JsonDRbObjectServer
+          def my_method(param)
+            param * 2
+          end
+        end
+
+        json = JsonDRb.new
+        json.start_service('127.0.0.1', 7777, JsonDRbObjectServer.new)
+        obj = JsonDRbObject.new("localhost", 7777)
+        expect(obj.my_method(10)).to eql 20
+        obj.disconnect
+        json.stop_service
+        sleep(0.1)
+      end
+
+      it "raises an exception if the remote connection can't be made" do
+        obj = JsonDRbObject.new("localhost", 7777)
+        expect { obj.my_method(10) }.to raise_error(DRb::DRbConnError)
+        obj.disconnect
+        sleep(0.1)
+      end
+
+      it "retries the request and then raise an exception" do
+        allow_any_instance_of(HTTPClient).to receive(:post) { raise "Error" }
+        obj = JsonDRbObject.new("localhost", 7777)
+        expect { obj.my_method(10) }.to raise_error(DRb::DRbConnError)
+        obj.disconnect
+        sleep(0.1)
+      end
+
+      it "raises an exception if the remote method returns an error" do
+        json = JsonDRb.new
+        json.start_service('127.0.0.1', 7777, self)
+        obj = JsonDRbObject.new("localhost", 7777)
+        expect { obj.no_such_method() }.to raise_error(NoMethodError)
+        expect { obj.method_missing(:send) }.to raise_error(/Cannot call unauthorized methods/)
+        obj.disconnect
+        json.stop_service
+        sleep(0.1)
+      end
+
+      it "handles the remote not returning a response" do
+        class JsonDRbObjectServer
+          def my_method(param)
+            param * 2
+          end
+        end
+
+        json = JsonDRb.new
+        json.start_service('127.0.0.1', 7777, JsonDRbObjectServer.new)
+        obj = JsonDRbObject.new("localhost", 7777)
+        allow_any_instance_of(JsonDrbRack).to receive(:call) { sleep(5.0); nil }
+        expect { obj.my_method(10) }.to raise_error(DRb::DRbConnError)
+        obj.disconnect
+        json.stop_service
+        sleep(0.1)
+      end
+
+      it "handles the remote going away and coming back" do
+        class JsonDRbObjectServer
+          def my_method(param)
+            param * 2
+          end
+        end
+
+        json = JsonDRb.new
+        json.start_service('127.0.0.1', 7777, JsonDRbObjectServer.new)
+        obj = JsonDRbObject.new("localhost", 7777)
+        expect(obj.my_method(10)).to eql 20
+        json.stop_service
+        json = JsonDRb.new
+        json.start_service('127.0.0.1', 7777, JsonDRbObjectServer.new)
+        expect(obj.my_method(10)).to eql 20
+        obj.disconnect
+        json.stop_service
+        sleep(0.1)
+      end
+
+    end
+  end
+end
+

data/spec/io/json_drb_rack_spec.rb

@@ -0,0 +1,166 @@
+  # encoding: ascii-8bit
+
+  # Copyright 2014 Ball Aerospace & Technologies Corp.
+  # All Rights Reserved.
+  #
+  # This program is free software; you can modify and/or redistribute it
+  # under the terms of the GNU General Public License
+  # as published by the Free Software Foundation; version 3 with
+  # attribution addendums as found in the LICENSE.txt
+
+  require 'spec_helper'
+  require 'cosmos/io/json_drb_rack'
+
+  module Cosmos
+
+    describe JsonDrbRack do
+      before(:each) do
+        @env = {
+          "GATEWAY_INTERFACE" => "CGI/1.1",
+          "PATH_INFO" => "/index.html",
+          "QUERY_STRING" => "",
+          "REMOTE_ADDR" => "::1",
+          "REMOTE_HOST" => "localhost",
+          "REQUEST_METHOD" => "POST",
+          "REQUEST_URI" => "http://localhost:3000/index.html",
+          "SCRIPT_NAME" => "",
+          "SERVER_NAME" => "localhost",
+          "SERVER_PORT" => "3000",
+          "SERVER_PROTOCOL" => "HTTP/1.1",
+          "SERVER_SOFTWARE" => "WEBrick/1.3.1 (Ruby/2.0.0/2013-11-22)",
+          "HTTP_HOST" => "127.0.0.1:7777",
+          "HTTP_USER_AGENT" => "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.9; rv:26.0) Gecko/20100101 Firefox/26.0",
+          "HTTP_ACCEPT" => "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
+          "HTTP_ACCEPT_LANGUAGE" => "zh-tw,zh;q=0.8,en-us;q=0.5,en;q=0.3",
+          "HTTP_ACCEPT_ENCODING" => "gzip, deflate",
+          "HTTP_COOKIE" => "jsonrpc.session=3iqp3ydRwFyqjcfO0GT2bzUh.bacc2786c7a81df0d0e950bec8fa1a9b1ba0bb61",
+          "HTTP_CONNECTION" => "keep-alive",
+          "HTTP_CACHE_CONTROL" => "max-age=0",
+          "rack.version" => [1, 2],
+          "rack.input" => StringIO.new(""),
+          "rack.errors" => nil,
+          "rack.multithread" => true,
+          "rack.multiprocess" => false,
+          "rack.run_once" => false,
+          "rack.url_scheme" => "http",
+          "HTTP_VERSION" => "HTTP/1.1",
+          "REQUEST_PATH" => "/index.html"
+        }
+      end
+
+      class MockDRb
+        class MockDRbAcl
+          def initialize(val = true)
+            @val = val
+          end
+          def allow_addr?(val)
+            return @val
+          end
+        end
+
+        def initialize(val, response_data = "response", error_code = nil)
+          @acl = MockDRbAcl.new(val)
+          @response_data = response_data
+          @error_code = error_code
+        end
+
+        def acl
+          @acl
+        end
+
+        def process_request(request_data, start_time)
+          return @response_data, @error_code
+        end
+      end
+
+      class MockSystem
+        def initialize(token = 'SuperSecret', hosts = ['127.0.0.1:7777'], origins = [])
+          @token = token
+          @hosts = hosts
+          @origins = origins
+        end
+        def x_csrf_token
+          @token
+        end
+        def allowed_hosts
+          @hosts
+        end
+        def allowed_origins
+          @origins
+        end
+      end
+
+      describe "call" do
+        it "supports the drb acl" do
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(false), MockSystem.new)
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 403
+          expect(type).to eql({'Content-Type' => "text/plain"})
+          expect(body_array).to eql ["Forbidden"]
+        end
+
+        it "handles X-Csrf-Token" do
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess'))
+          @env['HTTP_X_CSRF_TOKEN'] = 'TryToGuess'
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 403
+          expect(type).to eql({'Content-Type' => "text/plain"})
+          expect(body_array).to eql ["Forbidden: Bad X-Csrf-Token: #{@env['HTTP_X_CSRF_TOKEN']}"]
+
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess'))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 200
+          expect(type).to eql({'Content-Type' => "application/json-rpc"})
+          expect(body_array).to eql ["response"]
+        end
+
+        it "Handles Allowed Hosts" do
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess'))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          @env['HTTP_HOST'] = "5.6.7.8:7777"
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 403
+          expect(type).to eql({'Content-Type' => "text/plain"})
+          expect(body_array).to eql ["Forbidden: #{@env['HTTP_HOST']} not in allowed hosts"]
+
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess', ['5.6.7.8:7777']))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          @env['HTTP_HOST'] = "5.6.7.8:7777"
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 200
+          expect(type).to eql({'Content-Type' => "application/json-rpc"})
+          expect(body_array).to eql ["response"]
+        end
+
+        it "Handles Allowed Origins" do
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess'))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          @env['HTTP_ORIGIN'] = "5.6.7.8:7777"
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 403
+          expect(type).to eql({'Content-Type' => "text/plain"})
+          expect(body_array).to eql ["Forbidden: #{@env['HTTP_ORIGIN']} not in allowed origins"]
+
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess', ['127.0.0.1:7777'], ['3.6.7.8:7777']))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          @env['HTTP_ORIGIN'] = "3.6.7.8:7777"
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 200
+          expect(type).to eql({'Content-Type' => "application/json-rpc"})
+          expect(body_array).to eql ["response"]
+        end
+
+        it "Only accepts posts" do
+          json_drb_rack = JsonDrbRack.new(MockDRb.new(true), MockSystem.new('NeverGuess'))
+          @env['HTTP_X_CSRF_TOKEN'] = 'NeverGuess'
+          @env["REQUEST_METHOD"] = "GET"
+          status, type, body_array = json_drb_rack.call(@env)
+          expect(status).to eql 405
+          expect(type).to eql({'Content-Type' => "text/plain"})
+          expect(body_array).to eql ["Request not allowed"]
+        end
+
+      end
+    end
+  end