disposable_db 0.0.2 → 0.0.3

data/disposable_db.gemspec

@@ -10,7 +10,7 @@ Gem::Specification.new do |s|
   s.authors     = ['Keith Larrimore']
   s.homepage    = 'https://github.com/klarrimore/disposable_db'
   s.date        = Date.today.to_s
-  s.summary = 'A utility that makes it easy to create "on-the-fly databases"'
+  s.summary     = 'A utility that makes it easy to create "on-the-fly databases"'
   s.description = 'A utility that makes it easy to create "on-the-fly databases"'
 
   #s.rubyforge_project = "disposable_db"
@@ -18,7 +18,7 @@ Gem::Specification.new do |s|
   s.files         = `git ls-files`.split("\n")
   s.test_files    = `git ls-files -- {test,spec,bench,features}/*`.split("\n")
   s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
-  s.require_paths = ["lib"]
+  s.require_paths = ['lib']
 
   # specify any dependencies here; for example:
   s.add_development_dependency 'rspec', ['>= 2.9.0']

data/lib/disposable_db/database.rb

@@ -8,17 +8,25 @@ module DisposableDB
 
     def create_table!(table_template)
       @connection.create_table! table_template.name.to_sym do
+        primary_key table_template.primary_key if table_template.primary_key 
         table_template.columns.each do |column|
           column column.name.to_sym, column.type
         end
+        table_template.indexes.each do |idx|
+          index idx.to_sym
+        end
       end
     end
 
     def create_table(table_template)
       @connection.create_table? table_template.name.to_sym do
+        primary_key table_template.primary_key if table_template.primary_key 
         table_template.columns.each do |column|
           column column.name.to_sym, column.type
         end
+        table_template.indexes.each do |idx|
+          index idx.to_sym
+        end
       end
     end
 
@@ -27,4 +35,4 @@ module DisposableDB
     end
 
   end
-end
+end

data/lib/disposable_db/table_template.rb

@@ -1,16 +1,25 @@
 module DisposableDB
   class TableTemplate
-    attr_accessor :name, :columns
+    attr_accessor :name, :columns, :primary_key
 
     def initialize(name, columns = [])
       @name = name
       @columns = columns
+      @indexes = indexes
     end
 
     def add_column(column)
       @columns << column
     end
 
+    def set_primary_key(pk)
+      @primary_key = pk
+    end
+
+    def indexes
+      @columns.select{ |c| c.indexed == true }.collect{ |c| c.name }
+    end
+
     class Column
       attr_accessor :name, :type, :indexed
 
@@ -31,4 +40,4 @@ module DisposableDB
     end
 
   end
-end
+end

data/lib/disposable_db/version.rb

@@ -1,3 +1,3 @@
 module DisposableDB
-  VERSION = '0.0.2' unless defined? DisposableDB::VERSION
+  VERSION = '0.0.3' unless defined? DisposableDB::VERSION
 end

data/spec/database_spec.rb

@@ -20,5 +20,16 @@ describe Database do
     d.create_table tt
     d.table_exists?(tt.name).should be_true
   end
+  
+  it "can create table with primary_key" do
+    tt = TableTemplate.new('foo2', [TableTemplate::Column.new('t1', Integer)])
+    tt.set_primary_key :id
+
+    d = Database.new
+    d.table_exists?(tt.name).should be_false
+    d.create_table tt
+    d.table_exists?(tt.name).should be_true
+    d.connection[:foo2].columns.include?(:id).should be_true
+  end
 
 end

data/spec/table_template_spec.rb

@@ -19,5 +19,13 @@ describe TableTemplate do
 
     tt.columns.length.should == columns.length
   end
+  
+  it "can set primary key" do
+    tt = TableTemplate.new('foo')
+
+    tt.set_primary_key :id
+
+    tt.primary_key == :id
+  end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: disposable_db
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-26 00:00:00.000000000Z
+date: 2012-03-27 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &88875350 !ruby/object:Gem::Requirement
+  requirement: &7944580 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: 2.9.0
   type: :development
   prerelease: false
-  version_requirements: *88875350
+  version_requirements: *7944580
 - !ruby/object:Gem::Dependency
   name: bundler
-  requirement: &88875080 !ruby/object:Gem::Requirement
+  requirement: &7943920 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *88875080
+  version_requirements: *7943920
 - !ruby/object:Gem::Dependency
   name: ffaker
-  requirement: &88863790 !ruby/object:Gem::Requirement
+  requirement: &7943020 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: 1.13.0
   type: :development
   prerelease: false
-  version_requirements: *88863790
+  version_requirements: *7943020
 - !ruby/object:Gem::Dependency
   name: ruby-prof
-  requirement: &88863580 !ruby/object:Gem::Requirement
+  requirement: &7942140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *88863580
+  version_requirements: *7942140
 - !ruby/object:Gem::Dependency
   name: sequel
-  requirement: &88863280 !ruby/object:Gem::Requirement
+  requirement: &7940620 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - =
@@ -65,10 +65,10 @@ dependencies:
         version: 3.33.0
   type: :runtime
   prerelease: false
-  version_requirements: *88863280
+  version_requirements: *7940620
 - !ruby/object:Gem::Dependency
   name: sqlite3
-  requirement: &88863030 !ruby/object:Gem::Requirement
+  requirement: &7939500 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -76,7 +76,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *88863030
+  version_requirements: *7939500
 description: A utility that makes it easy to create "on-the-fly databases"
 email:
 - keithlarrimore@icehook.com
@@ -116,21 +116,15 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: 36076115
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: 36076115
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.10
+rubygems_version: 1.8.17
 signing_key: 
 specification_version: 3
 summary: A utility that makes it easy to create "on-the-fly databases"