ip_to_country 0.0.1 → 0.0.2

data/test/fixtures/ips/test_ips.csv

@@ -0,0 +1,14 @@
+"2.22.160.0","2.22.226.255","35037184","35054335","FR","France"
+"2.22.227.0","2.22.227.255","35054336","35054591","GB","United Kingdom"
+"2.22.228.0","2.22.229.255","35054592","35055103","EU","Europe"
+"2.22.230.0","2.22.230.255","35055104","35055359","ES","Spain"
+"2.22.231.0","2.22.232.255","35055360","35055871","EU","Europe"
+"2.22.233.0","2.22.233.255","35055872","35056127","NL","Netherlands"
+"2.22.234.0","2.22.245.255","35056128","35059199","EU","Europe"
+"2.22.246.0","2.22.247.255","35059200","35059711","GB","United Kingdom"
+"2.22.248.0","2.22.251.255","35059712","35060735","EU","Europe"
+"2.22.252.0","2.22.253.255","35060736","35061247","GB","United Kingdom"
+"2.22.254.0","2.22.255.255","35061248","35061759","FR","France"
+"2.23.0.0","2.23.111.255","35061760","35090431","EU","Europe"
+"2.23.112.0","2.23.127.255","35090432","35094527","GR","Greece"
+"2.23.128.0","2.23.255.255","35094528","35127295","EU","Europe"

data/test/loader_test.rb

@@ -0,0 +1,95 @@
+# -*- encoding : utf-8 -*-
+require 'test_helper'
+
+class LoaderTest < ActiveSupport::TestCase
+  
+  test "ip populate workflow" do
+    IpToCountry::Loader.expects(download_and_unzip_ips: true, create_tempo_table: true, 
+      load_csv_in_tempo_table: true, rename_tempo_into_geoip: true, 
+      clean_tmp_files: true)
+    assert(IpToCountry::Loader.populate!)
+  end
+
+  test "it download and unzip csv with ip address" do
+    setup_download_and_extract
+    
+    IpToCountry::Loader.expects(:system).with("wget -N #{IpToCountry::Loader::ARCHIVE_SOURCE}")
+    IpToCountry::Loader.expects(:system).with("unzip -j /tmp/test_ips.zip -d /tmp")
+    IpToCountry::Loader.download_and_unzip_ips
+    assert File.exists?('/tmp/test_ips.csv')
+    
+    teardown_download_and_extract
+  end
+
+  test "it creates a tempo table for ips" do
+    assert(IpToCountry::Loader.create_tempo_table)
+    assert ActiveRecord::Base.connection.table_exists?('geoips_tempo')
+  end
+
+  test "it loads csv ips into tempo table" do
+    setup_load_csv
+    IpToCountry::Loader.create_tempo_table
+    IpToCountry::Loader.load_csv_in_tempo_table
+    IpToCountry::Loader.rename_tempo_into_geoip
+    assert_equal 'France', IpToCountry::Geoip.by_ip('2.22.160.0').country_name
+    teardown_load_csv
+  end
+
+  test "it renames tempo table into geoips" do
+    IpToCountry::Loader.create_tempo_table
+    IpToCountry::Loader.rename_tempo_into_geoip
+    refute ActiveRecord::Base.connection.table_exists?('geoips_tempo')
+  end
+
+  test "it remove temp files" do
+    setup_file_remove
+    
+    IpToCountry::Loader.clean_tmp_files
+    refute File.exists?('/tmp/test_ips.csv')
+    refute File.exists?('/tmp/test_ips.zip')
+    
+    teardown_file_remove
+  end
+
+  private
+
+  def redefine_const(name, value)
+    IpToCountry::Loader.send(:remove_const, name) if IpToCountry::Loader.const_defined?(name)
+    IpToCountry::Loader.const_set(name, value)
+  end
+
+  def setup_download_and_extract
+    redefine_const('ARCHIVE_NAME', 'test_ips.zip')
+    FileUtils.cp('test/fixtures/ips/test_ips.zip', '.')
+    FileUtils.cp('test/fixtures/ips/test_ips.csv', '/tmp')
+  end
+
+  def teardown_download_and_extract
+    redefine_const('ARCHIVE_NAME', 'GeoIPCountryCSV.zip')
+    FileUtils.rm('/tmp/test_ips.csv')
+    FileUtils.rm('/tmp/test_ips.zip')
+  end
+
+  def setup_load_csv
+    FileUtils.cp('test/fixtures/ips/test_ips.csv', '/tmp')
+    redefine_const('EXTRACTED_IP_FILE', '/tmp/test_ips.csv')
+  end
+
+  def teardown_load_csv
+    FileUtils.rm('/tmp/test_ips.csv')
+    redefine_const('EXTRACTED_IP_FILE', '/tmp/GeoIPCountryWhois.csv')
+  end
+
+  def setup_file_remove
+    FileUtils.cp('test/fixtures/ips/test_ips.zip', "/tmp")
+    FileUtils.cp('test/fixtures/ips/test_ips.csv', "/tmp")
+    redefine_const('EXTRACTED_IP_FILE', '/tmp/test_ips.csv')
+    redefine_const('ARCHIVE_NAME', 'test_ips.zip')
+  end
+
+  def teardown_file_remove
+    redefine_const('EXTRACTED_IP_FILE', '/tmp/GeoIPCountryWhois.csv')
+    redefine_const('ARCHIVE_NAME', 'GeoIPCountryCSV.zip')
+  end
+
+end

data/test/test_helper.rb

@@ -5,6 +5,7 @@ ENV["RAILS_ENV"] = "test"
 require File.expand_path("../dummy/config/environment.rb",  __FILE__)
 require "rails/test_help"
 require 'coveralls'
+require "mocha/setup"
 
 Coveralls.wear!
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ip_to_country
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2013-04-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
-  requirement: &15605500 !ruby/object:Gem::Requirement
+  requirement: &13549180 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,18 +21,7 @@ dependencies:
         version: 3.2.12
   type: :runtime
   prerelease: false
-  version_requirements: *15605500
-- !ruby/object:Gem::Dependency
-  name: pg
-  requirement: &15632440 !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: *15632440
+  version_requirements: *13549180
 description: Simple tool to find the country of an ip address.
 email:
 - vincent.pochet@gmail.com
@@ -42,18 +31,22 @@ extra_rdoc_files: []
 files:
 - app/models/ip_to_country/geoip.rb
 - lib/ip_to_country/engine.rb
+- lib/ip_to_country/loader.rb
 - lib/ip_to_country/version.rb
 - lib/ip_to_country.rb
 - lib/generators/geoip/geoip_generator.rb
 - lib/generators/geoip/templates/migration.rb
-- lib/tasks/ip_to_country_tasks.rake
+- lib/tasks/ip_to_country.rake
 - MIT-LICENSE
 - Rakefile
 - README.md
 - test/ip_to_country_test.rb
+- test/loader_test.rb
 - test/geoip_test.rb
 - test/test_helper.rb
 - test/fixtures/geoips.yml
+- test/fixtures/ips/test_ips.csv
+- test/fixtures/ips/test_ips.zip
 - test/dummy/log/test.log
 - test/dummy/script/rails
 - test/dummy/public/favicon.ico
@@ -110,9 +103,12 @@ specification_version: 3
 summary: Simple tool to find the country of an ip address
 test_files:
 - test/ip_to_country_test.rb
+- test/loader_test.rb
 - test/geoip_test.rb
 - test/test_helper.rb
 - test/fixtures/geoips.yml
+- test/fixtures/ips/test_ips.csv
+- test/fixtures/ips/test_ips.zip
 - test/dummy/log/test.log
 - test/dummy/script/rails
 - test/dummy/public/favicon.ico

data/lib/tasks/ip_to_country_tasks.rake

@@ -1,4 +0,0 @@
-# desc "Explaining what the task does"
-# task :ip_to_country do
-#   # Task goes here
-# end