stockboy 0.5.0

data/spec/stockboy/providers_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+require 'stockboy/providers'
+
+module Stockboy
+  describe Providers do
+
+    let(:provider) { double("provider") }
+
+    describe ".register" do
+      it "registers a key and class" do
+        Providers.register(:snailmail, provider).should === provider
+      end
+    end
+
+    describe ".find" do
+      it "returns a provider class" do
+        Providers.register(:snailmail, provider)
+        Providers.find(:snailmail).should === provider
+      end
+    end
+
+    describe ".[]" do
+      it "returns a provider class" do
+        Providers.register(:snailmail, provider)
+        Providers[:snailmail].should === provider
+      end
+    end
+
+    describe ".all" do
+      it "returns all registered providers" do
+        Providers.register(:snailmail, provider)
+        Providers.register(:slugmail, provider)
+        Providers.all.should include(snailmail: provider, slugmail: provider)
+      end
+    end
+
+  end
+end

data/spec/stockboy/readers/csv_spec.rb

@@ -0,0 +1,68 @@
+require 'spec_helper'
+require 'stockboy/readers/csv'
+
+module Stockboy
+  describe Readers::CSV do
+
+    subject(:reader) { Stockboy::Readers::CSV.new }
+
+    describe "default options" do
+      its(:row_sep)          { should be_nil }
+      its(:col_sep)          { should be_nil }
+      its(:quote_char)       { should be_nil }
+      its(:headers)          { should be_true }
+      its(:skip_header_rows) { should == 0 }
+      its(:skip_footer_rows) { should == 0 }
+    end
+
+    describe "initialize" do
+      it "configures options with argument hash" do
+        reader = Readers::CSV.new(col_sep: '|')
+        reader.col_sep.should == '|'
+      end
+
+      it "configures options with a block" do
+        reader = Readers::CSV.new do
+          encoding 'ISO-8859-1'
+          col_sep "|"
+          skip_header_rows 2
+          skip_footer_rows 1
+        end
+
+        reader.col_sep.should == '|'
+        reader.skip_header_rows.should == 2
+        reader.skip_footer_rows.should == 1
+      end
+    end
+
+    describe "#parse" do
+      it "returns an array of records" do
+        records = reader.parse "id,name\n42,Arthur Dent"
+
+        records[0].should == {"id" => "42", "name" => "Arthur Dent"}
+      end
+
+      it "strips null bytes from empty fields (MSSQL BCP exports)" do
+        reader.col_sep = '|'
+        reader.headers = %w[city state country]
+        records = reader.parse "Vancouver|\x00|Canada"
+
+        records.should ==
+          [{"city" => "Vancouver", "state" => nil, "country" => "Canada"}]
+      end
+
+      it "strips preamble header rows" do
+        reader.skip_header_rows = 2
+        data = "IGNORE\r\nCOMMENTS\r\nid,name\r\n42,Arthur Dent"
+        records = reader.parse data
+
+        records[0].should == {"id" => "42", "name" => "Arthur Dent"}
+      end
+
+      it "shares hash key instances between records" do
+        records = reader.parse "id,name\n42,Arthur Dent\n999,Zaphod"
+        records[0].keys[0].should be records[1].keys[0]
+      end
+    end
+  end
+end

data/spec/stockboy/readers/fixed_width_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+require 'stockboy/readers/fixed_width'
+
+module Stockboy
+  describe Readers::FixedWidth do
+
+    let(:headers_hash) { {first: 10, last: 10, age: 4, sex: 2, planet: 10} }
+    let(:blank_row)    { "                                    \r\n" }
+    let(:data) {
+      "Arthur    Dent      42  M Human     \r\n" +
+      "Ford      Prefect   44  M Betelgeuse\r\n"
+    }
+
+    it "parses based on column widths" do
+      reader = described_class.new(headers: [10, 10, 4, 2, 10])
+      reader.parse(data).should == [
+        {0 => "Arthur", 1 => "Dent", 2 => "42", 3 => "M", 4 => "Human"},
+        {0 => "Ford", 1 => "Prefect", 2 => "44", 3 => "M", 4 => "Betelgeuse"}
+      ]
+    end
+
+    it "parses based on a hash" do
+      reader = described_class.new(headers: headers_hash)
+      reader.parse(data).should == [
+        {first: "Arthur", last: "Dent", age: "42", sex: "M", planet: "Human"},
+        {first: "Ford", last: "Prefect", age: "44", sex: "M", planet: "Betelgeuse"}
+      ]
+    end
+
+    it "skips blank rows" do
+      reader = described_class.new(headers: headers_hash)
+      records = reader.parse(blank_row + data + blank_row)
+      records.first[:age].should == '42'
+      records.last[:age].should == '44'
+    end
+
+    it "skips number of specified header rows" do
+      reader = described_class.new(headers: headers_hash)
+      reader.skip_header_rows = 1
+      records = reader.parse("REPORT\r\n" + data)
+      records.first[:age].should == '42'
+    end
+
+    it "skips number of specified footer rows" do
+      reader = described_class.new(headers: headers_hash)
+      reader.skip_footer_rows = 2
+      records = reader.parse(data + "SUBTOTAL\r\nTOTAL\r\n")
+      records.last[:age].should == '44'
+    end
+
+  end
+end

data/spec/stockboy/readers/spreadsheet_spec.rb

@@ -0,0 +1,121 @@
+require 'spec_helper'
+require 'stockboy/readers/spreadsheet'
+
+module Stockboy
+  describe Readers::Spreadsheet do
+
+    subject(:reader) { Stockboy::Readers::Spreadsheet.new }
+
+    describe "default options" do
+      its(:format) { should == :xls }
+      its(:sheet)  { should == :first }
+    end
+
+
+    describe "#initialize" do
+      it "configures with hash parameter" do
+        reader = described_class.new(
+          format:  :xlsx,
+          sheet:   'Sheet 42',
+          header_row: 5,
+          headers: %w(X Y Z)
+        )
+
+        reader.format.should == :xlsx
+        reader.sheet.should  == 'Sheet 42'
+        reader.header_row.should == 5
+        reader.headers.should == %w(X Y Z)
+      end
+
+      it "configures with a block" do
+        reader = described_class.new do
+          format :xlsx
+          sheet 'Sheet 42'
+          header_row 5
+          headers %w(X Y Z)
+        end
+
+        reader.format.should == :xlsx
+        reader.sheet.should  == 'Sheet 42'
+        reader.header_row.should == 5
+        reader.headers.should == %w(X Y Z)
+      end
+    end
+
+    describe "#parse" do
+      let(:content) { File.read(RSpec.configuration.fixture_path.join(fixture_file)) }
+
+      context "with an XLS file" do
+        let(:fixture_file) { 'spreadsheets/test_data.xls' }
+
+        it "returns an array of hashes" do
+          reader = described_class.new(format: :xls)
+          data = reader.parse(content)
+
+          data.should_not be_empty
+          data.each { |i| i.should be_a Hash }
+        end
+      end
+
+      context "with blank line options" do
+        let(:fixture_file) { 'spreadsheets/test_row_options.xls' }
+
+        it "starts on the given first row" do
+          reader = described_class.new(format: :xls, first_row: 6)
+          data = reader.parse(content)
+
+          data.first.values.should == ["Arthur Dent", 42]
+        end
+
+        it "ends on the given last row counting backwards" do
+          reader = described_class.new(format: :xls, last_row: -3)
+          data = reader.parse(content)
+
+          data.last.values.should == ["Marvin", 9999999]
+        end
+
+        it "ends on the given last row counting upwards" do
+          reader = described_class.new(format: :xls, last_row: 9)
+          data = reader.parse(content)
+
+          data.last.values.should == ["Ford", 40]
+        end
+      end
+    end
+
+    describe "#sheet", pending: "Hard to test this due to roo. Needs a test case with fixtures" do
+      let(:sheets)      { ['Towels', 'Lemons'] }
+      let(:expectation) { mock(:spreadsheet, sheets: sheets).should_receive(:default_sheet=) }
+      let(:spreadsheet) { expectation }
+      before { subject.stub!(:with_spreadsheet).and_yield(spreadsheet) }
+
+      context "with :first" do
+        let(:spreadsheet) { expectation.with('Towels') }
+
+        it "calls on first sheet by name" do
+          subject.sheet = :first
+          subject.parse ""
+        end
+      end
+
+      context "with a string" do
+        let(:spreadsheet) { expectation.with('Towels') }
+
+        it "passes unchanged" do
+          subject.sheet = 'Towels'
+          subject.parse ""
+        end
+      end
+
+      context "with an integer" do
+        let(:spreadsheet) { expectation.with('Towels') }
+
+        it "gets sheet name" do
+          subject.sheet = 1
+          subject.parse ""
+        end
+      end
+    end
+
+  end
+end

data/spec/stockboy/readers/xml_spec.rb

@@ -0,0 +1,94 @@
+require 'spec_helper'
+require 'stockboy/readers/xml'
+
+module Stockboy
+  describe Readers::XML do
+    subject(:reader) { Stockboy::Readers::XML.new }
+
+    describe "initialize" do
+      it "initializes hash options" do
+        reader = Readers::XML.new(
+          strip_namespaces:     true,
+          advanced_typecasting: true,
+          convert_tags_to:      ->(t) { t.snakecase },
+          parser:               :nokogiri,
+          elements:             ['SomeNested', 'Record']
+        )
+
+        reader.options[:strip_namespaces].should be_true
+        reader.options[:advanced_typecasting].should be_true
+        reader.options[:convert_tags_to].should be_a Proc
+        reader.options[:parser].should == :nokogiri
+        reader.elements.should == ['some_nested', 'record']
+      end
+
+      it "configures with a block" do
+        reader = Readers::XML.new do
+          encoding 'UTF-8'
+          strip_namespaces true
+          advanced_typecasting true
+          convert_tags_to ->(t) { t.snakecase }
+          parser :nokogiri
+          elements ['SomeNested', 'Record']
+        end
+
+        reader.options[:strip_namespaces].should be_true
+        reader.options[:advanced_typecasting].should be_true
+        reader.options[:convert_tags_to].should be_a Proc
+        reader.options[:parser].should == :nokogiri
+        reader.elements.should == ['some_nested', 'record']
+      end
+    end
+
+    describe "#parse" do
+
+      let(:xml_file)    { RSpec.configuration.fixture_path.join "xml/body.xml" }
+      let(:xml_string)  { File.read(xml_file) }
+      let(:elements)    { ['MultiNamespacedEntryResponse', 'history', 'case'] }
+      let(:output_keys) { ['logTime', 'logType', 'logText'] }
+      subject(:reader)  { Readers::XML.new(elements: elements, strip_namespaces: true) }
+
+      shared_examples :parse_data do
+        it "returns element hashes" do
+          items = reader.parse data
+          output_keys.each do |k|
+            items[0].keys.should include k
+          end
+        end
+
+        it "shares key string instances between records" do
+          items = reader.parse data
+          output_keys.each do |key|
+            i = items[0].keys.index(key)
+            items[0].keys[i].should be items[1].keys[i]
+          end
+        end
+      end
+
+      shared_examples :parse_data_with_tag_conversion do
+        context "with tag conversion" do
+          before            { reader.options[:convert_tags_to] = ->(t) { t.snakecase } }
+          let(:output_keys) { ['log_time', 'log_type', 'log_text'] }
+
+          include_examples :parse_data
+        end
+      end
+
+      context "an XML string" do
+        let(:data) { xml_string }
+
+        include_examples :parse_data
+        # include_examples :parse_data_with_tag_conversion
+      end
+
+
+      context "a SOAP response" do
+        let(:data) { Nori.new(strip_namespaces: true).parse(xml_string) }
+
+        include_examples :parse_data
+        include_examples :parse_data_with_tag_conversion
+      end
+
+    end
+  end
+end

data/spec/stockboy/readers_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+require 'stockboy/readers'
+
+module Stockboy
+  describe Readers do
+
+    let(:reader_class) { Class.new }
+
+    describe ".register" do
+      it "registers a key for a reader class" do
+        Readers.register(:markup, reader_class).should == reader_class
+      end
+    end
+
+    describe ".find" do
+      it "returns a reader class" do
+        Readers.register(:markup, reader_class)
+        Readers.find(:markup).should == reader_class
+      end
+    end
+
+    describe ".[]" do
+      it "returns a reader class" do
+        Readers.register(:markup, reader_class)
+        Readers[:markup].should == reader_class
+      end
+    end
+
+  end
+end

data/spec/stockboy/source_record_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+require 'stockboy/source_record'
+
+module Stockboy
+  describe SourceRecord do
+    subject(:record) do
+      SourceRecord.new({:full_name => 'Arthur Dent'},
+                       {3 => 'Arthur Dent'})
+    end
+
+    it "accesses initialized fields from hash" do
+      record.full_name.should == 'Arthur Dent'
+    end
+
+    it "accesses source field names" do
+      record[3].should == 'Arthur Dent'
+    end
+  end
+end

data/spec/stockboy/template_file_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+require 'stockboy/template_file'
+
+module Stockboy
+  describe TemplateFile do
+
+    let(:template_path) { File.expand_path("../../fixtures/jobs", __FILE__) }
+
+    before do
+      Stockboy.configuration.template_load_paths = [template_path]
+    end
+
+    describe ".read" do
+      it "returns the template string" do
+        TemplateFile.read("test_job").should match "# file exists!"
+      end
+    end
+
+    describe ".template_path" do
+      it "returns nil when missing" do
+        TemplateFile.find("not_here").should be_nil
+      end
+
+      it "returns a file path when found" do
+        TemplateFile.find("test_job").should == "#{template_path}/test_job.rb"
+      end
+    end
+
+  end
+end