ticker_fetcher 0.2.0 → 0.3.0

data/lib/ticker_fetcher.rb

@@ -7,17 +7,17 @@ class TickerFetcher
   Name = 1
   LastSale = 2
   MarketCap = 3
+  IPOYear = 4
   Sector = 5
   Industry = 6
-  attr_reader :exchanges
+  attr_reader :exchanges, :exchange_urls
 
-  # Passing 1 or more exchange names will retrieve only those exchanges. Default
-  # is to retrieve all.
+  # Passing an exchange name will retrieve only that exchange. Default is to retrieve all.
   def initialize
     @exchanges = {}
     @exchange_urls = {
-      'NASD' => 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=NASDAQ&render=download',
       'AMEX' => 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=AMEX&render=download',
+      'NASD' => 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=NASDAQ&render=download',
       'NYSE' => 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=NYSE&render=download'
     }
   end
@@ -42,15 +42,39 @@ class TickerFetcher
   private
 
   def parse_csv(data)
-    tickers = []
-    CSV.parse(data) { |row|
-      name = row[Name]
-      ticker = row[Symbol].strip.gsub('"', '')
-      unless(ticker.nil? || ticker.empty? || ticker == 'Symbol')
-        ticker = ticker.gsub('/', '.').gsub('^', '-') if ticker =~ /\W/
-        tickers << [ticker, name]
-      end
-    }
-    tickers
+    CSV.parse(data, :headers => true).inject([]) do |tickers, row|
+      tickers << [ticker(row), name(row), last_sale(row), market_cap(row), ipo_year(row), sector(row), industry(row)]
+    end
+  end
+
+  def name(row)
+    row[Name]
+  end
+
+  def ticker(row)
+    ticker = row[Symbol].strip.gsub('"', '')
+    unless(ticker.nil? || ticker.empty?)
+      ticker = ticker =~ /\W/ ? ticker.gsub('/', '.').gsub('^', '-') : ticker
+    end
+  end
+
+  def last_sale(row)
+    row[LastSale]
+  end
+
+  def market_cap(row)
+    row[MarketCap]
+  end
+
+  def ipo_year(row)
+    row[IPOYear]
+  end
+
+  def sector(row)
+    row[Sector]
+  end
+
+  def industry(row)
+    row[Industry]
   end
 end

data/test/helper.rb

@@ -1,10 +1,7 @@
 require 'rubygems'
 require 'test/unit'
-require 'flexmock/test_unit'
+require 'fakeweb'
 
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 require 'ticker_fetcher'
-
-class Test::Unit::TestCase
-end

data/test/test_ticker_fetcher.rb

@@ -2,38 +2,76 @@ require 'helper'
 
 class TestTickerFetcher < Test::Unit::TestCase
   def setup
-    @t = TickerFetcher.new
-    @t_mock = flexmock(@t)
-    @t_mock.should_receive(:parse_exchange).with('NYSE', 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=NYSE&render=download').and_return(@t_mock.send(:parse_csv, File.open("test/nyse_test_data.csv").read))
-    @t_mock.should_receive(:parse_exchange).with('NASD', 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=NASDAQ&render=download').and_return(@t_mock.send(:parse_csv, File.open("test/nasd_test_data.csv").read))
-    @t_mock.should_receive(:parse_exchange).with('AMEX', 'http://www.nasdaq.com/screening/companies-by-industry.aspx?exchange=AMEX&render=download').and_return(@t_mock.send(:parse_csv, File.open("test/amex_test_data.csv").read))
-#    @t_mock.should_receive(:retrieve).with_no_args.and_return({'NYSE' => File.open("test/nyse_test_data.csv").read, 'NASD' => File.open("test/nasd_test_data.csv").read, 'AMEX' => File.open("test/amex_test_data.csv").read})
-#    @t_mock.should_receive(:retrieve).with('NYSE').and_return({'NYSE' => File.open("test/nyse_test_data.csv").read})
-#    @t_mock.should_receive(:retrieve).with('NYSE', 'NASD').and_return({'NYSE' => File.open("test/nyse_test_data.csv").read, 'NASD' => File.open("test/nasd_test_data.csv").read})
-  end
-
-  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_the_correct_data
-    @t_mock.retrieve('NYSE')
-    assert_not_nil(@t_mock.exchanges['NYSE'])
-    assert_nil(@t_mock.exchanges['NASD'])
-    assert_nil(@t_mock.exchanges['AMEX'])
-    assert_equal('SC', @t_mock.exchanges['NYSE'][0][0])
-    assert_equal('AC-B', @t_mock.exchanges['NYSE'][1][0])
-    assert_equal('C.A', @t_mock.exchanges['NYSE'][2][0])
-  end
-
-  def test_retrieve_with_two_exchanges_returns_a_ticker_fetcher_with_the_correct_data
-    @t_mock.retrieve('NYSE')
-    @t_mock.retrieve('NASD')
-    assert_not_nil(@t_mock.exchanges['NYSE'])
-    assert_not_nil(@t_mock.exchanges['NASD'])
-    assert_nil(@t_mock.exchanges['AMEX'])
-  end
-
-  def test_retrieve_with_all_exchanges_returns_a_ticker_fetcher_with_the_correct_data
-    @t_mock.retrieve
-    assert_not_nil(@t_mock.exchanges['NYSE'])
-    assert_not_nil(@t_mock.exchanges['NASD'])
-    assert_not_nil(@t_mock.exchanges['AMEX'])
+    @tf = TickerFetcher.new
+    @tf.exchange_urls.each do |exchange, url|
+      FakeWeb.register_uri(
+        :get,
+        url,
+        :body => File.open("test/#{exchange.downcase}_test_data.csv").read)
+    end
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_one_exchange
+    @tf.retrieve('NYSE')
+    assert_not_nil(@tf.exchanges['NYSE'])
+    assert_nil(@tf.exchanges['NASD'])
+    assert_nil(@tf.exchanges['AMEX'])
+  end
+
+  def test_retrieve_with_two_exchanges_returns_a_ticker_fetcher_with_two_exchanges
+    @tf.retrieve('NYSE')
+    @tf.retrieve('NASD')
+    assert_not_nil(@tf.exchanges['NYSE'])
+    assert_not_nil(@tf.exchanges['NASD'])
+    assert_nil(@tf.exchanges['AMEX'])
+  end
+
+  def test_retrieve_with_all_exchanges_returns_a_ticker_fetcher_with_all_exchanges
+    @tf.retrieve
+    @tf.exchanges.keys.each do |exchange|
+      assert_not_nil(@tf.exchanges[exchange])
+    end
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_tickers
+    @tf.retrieve('NYSE')
+    assert_equal('SC', @tf.exchanges['NYSE'].first[TickerFetcher::Symbol])
+    assert_equal('ZGEN', @tf.exchanges['NYSE'].last[TickerFetcher::Symbol])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_company_names
+    @tf.retrieve('NYSE')
+    assert_equal('StupidCompany, Inc.', @tf.exchanges['NYSE'].first[TickerFetcher::Name])
+    assert_equal('ZymoGenetics, Inc.', @tf.exchanges['NYSE'].last[TickerFetcher::Name])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_last_sale_price
+    @tf.retrieve('NYSE')
+    assert_equal('2.71', @tf.exchanges['NYSE'].first[TickerFetcher::LastSale])
+    assert_equal('25', @tf.exchanges['NYSE'].last[TickerFetcher::LastSale])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_market_cap
+    @tf.retrieve('NYSE')
+    assert_equal('72858350', @tf.exchanges['NYSE'].first[TickerFetcher::MarketCap])
+    assert_equal('85473000', @tf.exchanges['NYSE'].last[TickerFetcher::MarketCap])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_ipo_year
+    @tf.retrieve('NYSE')
+    assert_equal('1999', @tf.exchanges['NYSE'].first[TickerFetcher::IPOYear])
+    assert_equal('2015', @tf.exchanges['NYSE'].last[TickerFetcher::IPOYear])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_sector
+    @tf.retrieve('NYSE')
+    assert_equal('Consumer Services', @tf.exchanges['NYSE'].first[TickerFetcher::Sector])
+    assert_equal('Technology', @tf.exchanges['NYSE'].last[TickerFetcher::Sector])
+  end
+
+  def test_retrieve_with_one_exchange_returns_a_ticker_fetcher_with_correct_industry
+    @tf.retrieve('NYSE')
+    assert_equal('Other Specialty Stores', @tf.exchanges['NYSE'].first[TickerFetcher::Industry])
+    assert_equal('Semiconductors', @tf.exchanges['NYSE'].last[TickerFetcher::Industry])
   end
 end

data/ticker_fetcher.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name = "ticker_fetcher"
-  s.version = "0.2.0"
+  s.version = "0.3.0"
   s.authors = ["Matt White"]
   s.email = %q{mattw922@gmail.com}
   s.homepage = %q{http://github.com/whitethunder/ticker_fetcher}
@@ -12,6 +12,6 @@ Gem::Specification.new do |s|
   s.test_files    = `git ls-files -- {test}/*`.split("\n")
   s.require_paths = ["lib"]
 
-  s.add_development_dependency(%q<flexmock>, [">= 0.8.2"])
+  s.add_development_dependency(%q{fakeweb}, [">= 1.3.0"])
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ticker_fetcher
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,19 +9,19 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-08-05 00:00:00.000000000Z
+date: 2011-08-09 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: flexmock
-  requirement: &81809110 !ruby/object:Gem::Requirement
+  name: fakeweb
+  requirement: &71675170 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
       - !ruby/object:Gem::Version
-        version: 0.8.2
+        version: 1.3.0
   type: :development
   prerelease: false
-  version_requirements: *81809110
+  version_requirements: *71675170
 description: Retrieves tickers, and names for all securities listed on the 3 major
   US exchanges (NYSE, NASDAQ, AMEX).
 email: mattw922@gmail.com