ethel-sequel 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+bin

data/Gemfile

@@ -0,0 +1,12 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ethel-sequel.gemspec
+gemspec
+
+group :development do
+  gem 'test-unit'
+  gem 'mocha'
+  gem 'guard-test'
+  gem 'rb-inotify', '~> 0.8.8'
+  gem 'sqlite3'
+end

data/Guardfile

@@ -0,0 +1,8 @@
+guard 'test' do
+  watch(%r{^lib/ethel/([^/]+/)*([^/]+)\.rb$}) do |m|
+    "test/unit/#{m[1]}test_#{m[2]}.rb"
+  end
+  watch(%r{^test/unit/([^/]+/)*test_.+\.rb$})
+  watch(%r{^test/integration/test_.+\.rb$})
+  watch('test/helper.rb') { 'test' }
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Jeremy Stephens
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# Ethel::Sequel
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'ethel-sequel'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ethel-sequel
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,9 @@
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+task :default => :test

data/ethel-sequel.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ethel/sequel/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "ethel-sequel"
+  gem.version       = Ethel::Sequel::VERSION
+  gem.authors       = ["Jeremy Stephens"]
+  gem.email         = ["jeremy.f.stephens@vanderbilt.edu"]
+  gem.description   = %q{Adds Sequel source and target for Ethel}
+  gem.summary       = %q{Sequel plugin for Ethel}
+  gem.homepage      = "https://github.com/coupler/ethel-sequel"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency 'sequel'
+  gem.add_dependency 'ethel'
+end

data/lib/ethel-sequel.rb

@@ -0,0 +1 @@
+require 'ethel/sequel'

data/lib/ethel/sequel.rb

@@ -0,0 +1,6 @@
+require 'ethel'
+require 'sequel'
+
+require 'ethel/sequel/version'
+require 'ethel/sources/sequel'
+require 'ethel/targets/sequel'

data/lib/ethel/sequel/version.rb

@@ -0,0 +1,5 @@
+module Ethel
+  module Sequel
+    VERSION = "0.0.1"
+  end
+end

data/lib/ethel/sources/sequel.rb

@@ -0,0 +1,31 @@
+module Ethel
+  module Sources
+    class Sequel < Source
+      def initialize(*args)
+        if args.length == 1
+          @dataset = args[0]
+          @database = @dataset.db
+        else
+          @database = ::Sequel.connect(*args[1..-1])
+          @dataset = @database[args[0]]
+          @dataset_arg = args[0]
+        end
+      end
+
+      def schema
+        table =
+          if @dataset_arg.nil?
+            @dataset.first_source_table
+          else
+            @dataset_arg
+          end
+
+        @database.schema(table)
+      end
+
+      def each(&block)
+        @dataset.each(&block)
+      end
+    end
+  end
+end

data/lib/ethel/targets/sequel.rb

@@ -0,0 +1,66 @@
+module Ethel
+  module Targets
+    class Sequel < Target
+      attr_accessor :force, :import_limit
+
+      def initialize(table_name, *args)
+        super
+
+        @database =
+          if args.length == 1 && args[0].kind_of?(::Sequel::Database)
+            args[0]
+          else
+            ::Sequel.connect(*args)
+          end
+
+        @table_name = table_name
+        @fields = []
+        @rows = []
+
+        @force = false
+        @import_limit = 10_000
+      end
+
+      def add_field(field)
+        @fields << field
+      end
+
+      def prepare
+        exists = @database.tables.include?(@table_name)
+        if exists && !@force
+          raise "Table #{@table_name} already exists"
+        else
+          generator = @database.create_table_generator
+          @fields.each do |field|
+            type =
+              case field.type
+              when :integer
+                Integer
+              when :string
+                String
+              end
+            generator.column(field.name, type)
+          end
+          if exists
+            @database.create_table!(@table_name, generator)
+          else
+            @database.create_table(@table_name, generator)
+          end
+        end
+      end
+
+      def add_row(row)
+        @rows << row
+
+        flush if @rows.length >= @import_limit
+      end
+
+      def flush
+        dataset = @database[@table_name]
+        keys = @fields.collect { |field| field.name.to_sym }
+        dataset.import(keys, @rows.collect { |r| r.values_at(*keys) })
+        @rows.clear
+      end
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,27 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+
+require 'test/unit'
+require 'mocha/setup'
+require 'tempfile'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'ethel/sequel'
+
+class SequenceHelper
+  def initialize(name)
+    @seq = Mocha::Sequence.new(name)
+  end
+
+  def <<(expectation)
+    expectation.in_sequence(@seq)
+  end
+end

data/test/integration/test_migration.rb

@@ -0,0 +1,22 @@
+require 'helper'
+
+class TestMigration < Test::Unit::TestCase
+  test "basic migration" do
+    db = Sequel.sqlite
+    db.create_table(:foo) do
+      primary_key :id
+      String :bar
+    end
+    db[:foo].import([:bar], [%w{123}, %w{456}])
+    source = Ethel::Sources::Sequel.new(db[:foo])
+    target = Ethel::Targets::Sequel.new(:bar, db)
+    migration = Ethel::Migration.new(source, target)
+    migration.copy(source.fields[:id])
+    migration.cast(source.fields[:bar], :integer)
+    migration.run
+
+    assert_include db.tables, :bar
+    assert_equal([{:id => 1, :bar => 123}, {:id => 2, :bar => 456}],
+      db[:bar].all)
+  end
+end

data/test/unit/sources/test_sequel.rb

@@ -0,0 +1,83 @@
+require 'helper'
+
+module TestSources
+  class TestSequel < Test::Unit::TestCase
+    def self.const_missing(name)
+      if Ethel.const_defined?(name)
+        Ethel.const_get(name)
+      else
+        super
+      end
+    end
+
+    def setup
+      @database = stub('database')
+      Sequel.stubs(:connect).returns(@database)
+      @dataset = stub('dataset', :db => @database)
+      @database.stubs(:[]).returns(@dataset)
+    end
+
+    test "subclass of Source" do
+      assert_equal Source, Sources::Sequel.superclass
+    end
+
+    test "initialize with table name, uri, and options" do
+      Sequel.expects(:connect).with('sqlite:/', :timeout => 1234).
+        returns(@database)
+      @database.expects(:[]).with(:foo).returns(@dataset)
+      assert_nothing_raised do
+        Sources::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+      end
+    end
+
+    test "initialize with Sequel dataset" do
+      @dataset.expects(:db).returns(@database)
+      assert_nothing_raised { Sources::Sequel.new(@dataset) }
+    end
+
+    test "schema when initialized with Sequel dataset" do
+      source = Sources::Sequel.new(@dataset)
+
+      schema = [
+        [:id, {
+          :allow_null=>false, :default=>nil, :primary_key=>true,
+          :db_type=>"integer", :type=>:integer, :ruby_default=>nil
+        }],
+        [:foo, {
+          :allow_null=>true, :default=>nil, :primary_key=>false,
+          :db_type=>"varchar(255)", :type=>:string, :ruby_default=>nil
+        }]
+      ]
+      @dataset.expects(:first_source_table).returns(:foo)
+      @database.expects(:schema).with(:foo).returns(schema)
+      assert_equal schema, source.schema
+    end
+
+    test "schema when initialized with table_name, uri, and options" do
+      source = Sources::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+
+      schema = [
+        [:id, {
+          :allow_null=>false, :default=>nil, :primary_key=>true,
+          :db_type=>"integer", :type=>:integer, :ruby_default=>nil
+        }],
+        [:foo, {
+          :allow_null=>true, :default=>nil, :primary_key=>false,
+          :db_type=>"varchar(255)", :type=>:string, :ruby_default=>nil
+        }]
+      ]
+      @database.expects(:schema).with(:foo).returns(schema)
+      assert_equal schema, source.schema
+    end
+
+    test "each" do
+      source = Sources::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+
+      row = {:id => 1, :foo => 123}
+      @dataset.expects(:each).yields(row)
+      source.each do |actual_row|
+        assert_equal row, actual_row
+      end
+    end
+  end
+end

data/test/unit/targets/test_sequel.rb

@@ -0,0 +1,127 @@
+require 'helper'
+
+module TestTargets
+  class TestSequel < Test::Unit::TestCase
+    def self.const_missing(name)
+      if Ethel.const_defined?(name)
+        Ethel.const_get(name)
+      else
+        super
+      end
+    end
+
+    def setup
+      @database = stub('database')
+      ::Sequel.stubs(:connect).returns(@database)
+      @dataset = stub('dataset', :db => @database)
+      @database.stubs(:[]).returns(@dataset)
+    end
+
+    test "subclass of Target" do
+      assert_equal Target, Targets::Sequel.superclass
+    end
+
+    test "initialize with table name, uri, and options" do
+      ::Sequel.expects(:connect).with('sqlite:/', :timeout => 1234).
+        returns(@database)
+      assert_nothing_raised do
+        Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+      end
+    end
+
+    test "using Sequel::Database" do
+      ::Sequel.expects(:connect).never
+      @database.expects(:kind_of?).with(::Sequel::Database).returns(true)
+      Targets::Sequel.new(:foo, @database)
+    end
+
+    test "#prepare creates table if it doesn't exist" do
+      target = Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+
+      field_1 = stub('id field', :name => 'id', :type => :integer)
+      target.add_field(field_1)
+
+      field_2 = stub('foo field', :name => 'foo', :type => :string)
+      target.add_field(field_2)
+
+      seq = SequenceHelper.new('create table sequence')
+      generator = stub('table generator')
+      seq << @database.expects(:tables).returns([])
+      seq << @database.expects(:create_table_generator).returns(generator)
+      seq << generator.expects(:column).with('id', Integer)
+      seq << generator.expects(:column).with('foo', String)
+      seq << @database.expects(:create_table).with(:foo, generator)
+      target.prepare
+    end
+
+    test "#prepare raises error if table exists" do
+      target = Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+
+      field_1 = stub('id field', :name => 'id', :type => :integer)
+      target.add_field(field_1)
+
+      field_2 = stub('foo field', :name => 'foo', :type => :string)
+      target.add_field(field_2)
+
+      @database.expects(:tables).returns([:foo])
+      assert_raises { target.prepare }
+    end
+
+    test "#prepare forces table creation when force option is true" do
+      target = Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234}, )
+      target.force = true
+
+      field_1 = stub('id field', :name => 'id', :type => :integer)
+      target.add_field(field_1)
+
+      field_2 = stub('foo field', :name => 'foo', :type => :string)
+      target.add_field(field_2)
+
+      seq = SequenceHelper.new('create table sequence')
+      generator = stub('table generator')
+      seq << @database.expects(:tables).returns([:foo])
+      seq << @database.expects(:create_table_generator).returns(generator)
+      seq << generator.expects(:column).with('id', Integer)
+      seq << generator.expects(:column).with('foo', String)
+      seq << @database.expects(:create_table!).with(:foo, generator)
+      target.prepare
+    end
+
+    test "#flush" do
+      target = Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+
+      field_1 = stub('id field', :name => 'id', :type => :integer)
+      target.add_field(field_1)
+
+      field_2 = stub('foo field', :name => 'foo', :type => :string)
+      target.add_field(field_2)
+
+      target.add_row({:id => 1, :foo => 'bar'})
+      target.add_row({:id => 2, :foo => 'baz'})
+      @database.expects(:[]).with(:foo).returns(@dataset)
+      @dataset.expects(:import).with([:id, :foo], [[1, 'bar'], [2, 'baz']])
+      target.flush
+    end
+
+    test "#add_row automatically flushes when limit reached" do
+      target = Targets::Sequel.new(:foo, 'sqlite:/', {:timeout => 1234})
+      target.import_limit = 2
+
+      field_1 = stub('id field', :name => 'id', :type => :integer)
+      target.add_field(field_1)
+
+      field_2 = stub('foo field', :name => 'foo', :type => :string)
+      target.add_field(field_2)
+
+      target.add_row({:id => 1, :foo => 'bar'})
+      @database.expects(:[]).with(:foo).returns(@dataset)
+      @dataset.expects(:import).with([:id, :foo], [[1, 'bar'], [2, 'baz']])
+      target.add_row({:id => 2, :foo => 'baz'})
+
+      target.add_row({:id => 3, :foo => 'test'})
+      @database.expects(:[]).with(:foo).returns(@dataset)
+      @dataset.expects(:import).with([:id, :foo], [[3, 'test'], [4, 'junk']])
+      target.add_row({:id => 4, :foo => 'junk'})
+    end
+  end
+end

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: ethel-sequel
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Jeremy Stephens
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ethel
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Adds Sequel source and target for Ethel
+email:
+- jeremy.f.stephens@vanderbilt.edu
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Guardfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- ethel-sequel.gemspec
+- lib/ethel-sequel.rb
+- lib/ethel/sequel.rb
+- lib/ethel/sequel/version.rb
+- lib/ethel/sources/sequel.rb
+- lib/ethel/targets/sequel.rb
+- test/helper.rb
+- test/integration/test_migration.rb
+- test/unit/sources/test_sequel.rb
+- test/unit/targets/test_sequel.rb
+homepage: https://github.com/coupler/ethel-sequel
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Sequel plugin for Ethel
+test_files:
+- test/helper.rb
+- test/integration/test_migration.rb
+- test/unit/sources/test_sequel.rb
+- test/unit/targets/test_sequel.rb