libsql 0.1.0

data/spec/aggregate_spec.rb

@@ -0,0 +1,158 @@
+require 'spec_helper'
+
+class AggregateTest1 < ::Libsql::Aggregate
+  def initialize
+    super
+    @name = 'atest1'
+    @arity = -1
+    @count = 0
+  end
+  def step( *args )
+    @count += 1
+  end
+  def finalize
+    return @count
+  end
+end
+
+
+describe "Aggregate SQL Functions" do
+
+  it "must have a finalize method implemented" do
+    ag = ::Libsql::Aggregate.new
+    lambda { ag.finalize }.should raise_error( NotImplementedError, /Aggregate#finalize must be implemented/ )
+  end
+
+  it "can define a custom SQL aggregate as a class with N params" do
+    @iso_db.define_aggregate("atest1", AggregateTest1 )
+    r = @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country")
+    r.first['a'].should eql(r.first['c'])
+    r.first['a'].should eql(242)
+  end
+
+  it "can remove a custom SQL aggregate by class" do
+    @iso_db.define_aggregate("atest1", AggregateTest1 )
+    @iso_db.aggregates.size.should eql(1)
+    r = @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country")
+    r.first['a'].should eql(r.first['c'])
+    r.first['a'].should eql(242)
+    @iso_db.remove_aggregate( "atest1", AggregateTest1 )
+    @iso_db.aggregates.size.should eql(0)
+    lambda{ @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country") }.should raise_error(::Libsql::SQLite3::Error, /no such function: atest1/ )
+  end
+
+  it "can remove a custom SQL aggregate by arity" do
+    @iso_db.define_aggregate("atest1", AggregateTest1 )
+    @iso_db.aggregates.size.should eql(1)
+    r = @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country")
+    r.first['a'].should eql(r.first['c'])
+    r.first['a'].should eql(242)
+    @iso_db.remove_aggregate( "atest1", -1)
+    @iso_db.aggregates.size.should eql(0)
+    lambda{ @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country") }.should raise_error(::Libsql::SQLite3::Error, /no such function: atest1/ )
+  end
+
+  it "can remove all custom SQL aggregates with the same name" do
+    class AT2 < AggregateTest1
+      def arity() 1; end
+    end
+    @iso_db.define_aggregate("atest1", AggregateTest1 )
+    @iso_db.define_aggregate("atest1", AT2)
+    @iso_db.aggregates.size.should eql(2)
+    r = @iso_db.execute("SELECT atest1(id,name) as a, atest1(id), count(*) as c FROM country")
+    r.first['a'].should eql(r.first['c'])
+    r.first['a'].should eql(242)
+    @iso_db.remove_aggregate( "atest1" )
+    @iso_db.aggregates.size.should eql(0)
+    lambda{ @iso_db.execute("SELECT atest1(id,name) as a, count(*) as c FROM country") }.should raise_error(::Libsql::SQLite3::Error, /no such function: atest1/ )
+  end
+
+
+
+  it "does not allow mixing of arbitrary and mandatory arguments to an SQL function" do
+    class AggregateTest2 < AggregateTest1
+      def name() "atest2"; end
+      def arity() -2; end
+    end
+    lambda { @iso_db.define_aggregate("atest2", AggregateTest2 ) }.should raise_error( ::Libsql::Database::AggregateError, 
+                                                    /Use only mandatory or arbitrary parameters in an SQL Aggregate, not both/ )
+  end
+
+  it "does not allow outrageous arity" do
+    class AggregateTest3 < AggregateTest1
+      def name() "atest3"; end
+      def arity() 128; end
+    end
+    lambda { @iso_db.define_aggregate("atest3", AggregateTest3 ) }.should raise_error( ::Libsql::SQLite3::Error, /SQLITE_ERROR .* Library used incorrectly/ )
+  end
+
+  it "does not allow registering a function which does not match the defined name " do
+    class AggregateTest4 < AggregateTest1
+      def name() "name_mismatch"; end
+    end
+    lambda { @iso_db.define_aggregate("atest4", AggregateTest4 ) }.should raise_error( ::Libsql::Database::AggregateError,
+                                           /Aggregate implementation name 'name_mismatch' does not match defined name 'atest4'/)
+  end
+
+  it "handles an error being thrown during the step function" do
+    class AggregateTest5 < AggregateTest1
+      def initialize
+        super
+        @name = "atest5"
+        @arity = -1
+        @count = 0
+      end
+
+      def step( *args )
+        @count += 1
+        if @count > 50 then
+          raise "Stepwise error!" if @count > 50
+        end
+      end
+
+    end
+
+    @iso_db.define_aggregate( "atest5", AggregateTest5 )
+    lambda { @iso_db.execute( "SELECT atest5(*) AS a FROM country" ) }.should raise_error( ::Libsql::SQLite3::Error, /Stepwise error!/ )
+  end
+
+  it "handles an error being thrown during the finalize function" do
+    class AggregateTest6 < AggregateTest1
+      def initialize
+        super
+        @name = "atest6"
+        @count = 0
+        @arity = -1
+      end
+      def finalize
+        raise "Finalize error!"
+      end
+    end
+    @iso_db.define_aggregate( "atest6", AggregateTest6 )
+    lambda { @iso_db.execute( "SELECT atest6(*) AS a FROM country" ) }.should raise_error( ::Libsql::SQLite3::Error, /Finalize error!/ )
+  end
+
+  it "handles an error being thrown during initialization in the C extension" do
+    class AggregateTest7 < AggregateTest1
+      @called = false
+      def self.called?
+        if @called then
+          raise "Initialization error!"
+        else
+          @called = true
+        end
+      end
+
+      def initialize
+        super
+        @name = "atest7"
+        @count = 0
+        @arity = -1
+        self.class.called?
+      end
+    end
+    @iso_db.define_aggregate( "atest7", AggregateTest7 )
+    lambda { @iso_db.execute( "SELECT atest7(*) AS a FROM country" ) }.should raise_error( ::Libsql::SQLite3::Error, /Initialization error!/ )
+  end
+end
+

data/spec/blob_spec.rb

@@ -0,0 +1,78 @@
+require 'spec_helper'
+
+describe ::Libsql::Blob do
+
+  before(:each) do
+    @blob_db_name = File.join(File.dirname( __FILE__ ), "blob.db")
+    File.unlink @blob_db_name if File.exist?( @blob_db_name )
+    @db = ::Libsql::Database.new( @blob_db_name )
+    @schema_sql = <<-SQL
+      CREATE TABLE blobs(
+        id      INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
+        name    VARCHAR(128) NOT NULL UNIQUE,
+        data    TEXT ); 
+    SQL
+    @db.execute( @schema_sql )
+    @country_data_file = ::Libsql::Iso3166Database.country_data_file
+    @junk_file = File.join( File.dirname(__FILE__), "test_output")
+  end
+
+  after(:each) do
+    @db.close
+    File.unlink @blob_db_name if File.exist?( @blob_db_name )
+    File.unlink @junk_file    if File.exist?( @junk_file )
+  end
+
+  { :file   => ::Libsql::Iso3166Database.country_data_file,
+    :string => IO.read( ::Libsql::Iso3166Database.country_data_file),
+    :io     => StringIO.new( IO.read( ::Libsql::Iso3166Database.country_data_file) ) }.each_pair do |style, data |
+    describe "inserts a blob from a #{style}" do
+      before(:each) do
+        column = @db.schema.tables['blobs'].columns['data']
+        @db.execute("INSERT INTO blobs(name, data) VALUES ($name, $data)",
+                    { "$name" => @country_data_file,
+                      "$data" => ::Libsql::Blob.new( style => data,
+                                                      :column => column ) } )
+        @db.execute("VACUUM")
+      end
+
+      after(:each) do
+        @db.execute("DELETE FROM blobs")
+        data.rewind if data.respond_to?( :rewind )
+      end
+
+      it "and retrieves the data as a single value" do
+        all_rows = @db.execute("SELECT name,data FROM blobs")
+        all_rows.size.should eql(1)
+        all_rows.first['name'].should eql(@country_data_file)
+        all_rows.first['data'].should_not be_incremental
+        all_rows.first['data'].to_string_io.string.should eql(IO.read( @country_data_file ))
+      end
+
+      it "and retrieves the data using incremental IO" do
+        all_rows = @db.execute("SELECT * FROM blobs")
+        all_rows.size.should eql(1)
+        all_rows.first['name'].should eql(@country_data_file)
+        all_rows.first['data'].should be_incremental
+        all_rows.first['data'].to_string_io.string.should eql(IO.read( @country_data_file ))
+      end
+
+      it "writes the data to a file " do
+        all_rows = @db.execute("SELECT * FROM blobs")
+        all_rows.size.should eql(1)
+        all_rows.first['name'].should eql(@country_data_file)
+        all_rows.first['data'].should be_incremental
+        all_rows.first['data'].write_to_file( @junk_file )
+        IO.read( @junk_file).should eql(IO.read( @country_data_file ))
+      end
+    end
+  end
+
+
+
+  it "raises an error if initialized incorrectly" do
+    lambda{ ::Libsql::Blob.new( :file => "/dev/null", :string => "foo" ) }.should raise_error( ::Libsql::Blob::Error )
+  end
+end
+
+

data/spec/boolean_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+require 'libsql'
+require 'libsql/boolean'
+
+describe ::Libsql::Boolean do
+  %w[ True Y Yes T 1 ].each do |v|
+    it "converts #{v} to true" do
+      ::Libsql::Boolean.to_bool(v).should == true
+    end
+  end
+
+  %w[ False F f No n 0 ].each do |v|
+    it "converts #{v} to false " do
+      ::Libsql::Boolean.to_bool(v).should == false
+    end
+  end
+
+  %w[ other things nil ].each do |v|
+    it "converts #{v} to nil" do
+      ::Libsql::Boolean.to_bool(v).should == nil
+    end
+  end
+end

data/spec/busy_handler.rb

@@ -0,0 +1,157 @@
+require 'spec_helper'
+
+class BusyHandlerTest < ::Libsql::BusyHandler
+  attr_accessor :call_count
+  def initialize( max = 5 )
+    @max = max
+    @call_count = 0
+  end
+
+  def call( c )
+    @call_count += 1
+    if call_count >= @max then
+      return false
+    end
+    return true
+  end
+end
+
+describe "Busy Handlers" do
+  before(:each) do
+    @read_db = ::Libsql::Database.new( @iso_db_path )
+    @write_db = ::Libsql::Database.new( @iso_db_path )
+  end
+
+  after(:each) do
+    @write_db.close
+    @read_db.close
+  end
+
+  it "raises NotImplemented if #call is not overwritten" do
+    bh = ::Libsql::BusyHandler.new
+    lambda { bh.call( 42 ) }.should raise_error( ::NotImplementedError, /The busy handler call\(N\) method must be implemented/ )
+  end
+
+  it "can be registered as block" do
+    call_count = 0
+    @write_db.busy_handler do |x|
+      call_count = x
+      if call_count >= 20 then
+        false
+      else
+        true
+      end
+    end
+
+    # put a read lock on the database
+    @read_db.transaction( "DEFERRED" )
+
+    # put a read lock on the database, but want to go to an exclusive
+    @write_db.transaction( "IMMEDIATE" )
+
+    # do a read operation
+    @read_db.execute("SELECT count(*) FROM subcountry")
+
+    # attempt to do a write operation and commit it
+    @write_db.execute("DELETE FROM subcountry")
+    lambda { @write_db.execute("COMMIT"); }.should raise_error( ::Libsql::SQLite3::Error, /database is locked/ )
+    call_count.should == 20
+  end
+
+  it "can be registered as lambda" do
+    call_count = 0
+    callable = lambda do |x|
+      call_count = x
+      if call_count >= 40 then
+        false
+      else
+        true
+      end
+    end
+
+    @write_db.busy_handler( callable )
+
+    # put a read lock on the database
+    @read_db.transaction( "DEFERRED" )
+
+    # put a read lock on the database, but want to go to an exclusive
+    @write_db.transaction( "IMMEDIATE" )
+
+    # do a read operation
+    @read_db.execute("SELECT count(*) FROM subcountry")
+
+    # attempt to do a write operation and commit it
+    @write_db.execute("DELETE FROM subcountry")
+    lambda { @write_db.execute("COMMIT"); }.should raise_error( ::Libsql::SQLite3::Error, /database is locked/ )
+    call_count.should == 40
+  end
+
+  it "can be registered as a class" do
+    h = BusyHandlerTest.new( 10 )
+    @write_db.busy_handler( h )
+
+    # put a read lock on the database
+    @read_db.transaction( "DEFERRED" )
+
+    # put a read lock on the database, but want to go to an exclusive
+    @write_db.transaction( "IMMEDIATE" )
+
+    # do a read operation
+    @read_db.execute("SELECT count(*) FROM subcountry")
+
+    # attempt to do a write operation and commit it
+    @write_db.execute("DELETE FROM subcountry")
+    lambda { @write_db.execute("COMMIT"); }.should raise_error( ::Libsql::SQLite3::Error, /database is locked/ )
+    h.call_count.should == 10
+  end
+
+  it "has a default timeout class available " do
+    to = ::Libsql::BusyTimeout.new( 5, 10 ) 
+    @write_db.busy_handler( to )
+
+    # put a read lock on the database
+    @read_db.transaction( "DEFERRED" )
+
+    # put a read lock on the database, but want to go to an exclusive
+    @write_db.transaction( "IMMEDIATE" )
+
+    # do a read operation
+    @read_db.execute("SELECT count(*) FROM subcountry")
+
+    # attempt to do a write operation and commit it
+    @write_db.execute("DELETE FROM subcountry")
+    before = Time.now
+    lambda { @write_db.execute("COMMIT"); }.should raise_error( ::Libsql::SQLite3::Error, /database is locked/ )
+    after = Time.now
+    to.call_count.should > 5
+    (after - before).should > 0.05
+  end
+
+  it "cannot register a block with the wrong arity" do
+    lambda do 
+      @write_db.define_busy_handler { |x,y| puts "What!" }
+    end.should raise_error( ::Libsql::Database::BusyHandlerError, /A busy handler expects 1 and only 1 argument/ )
+  end
+
+  it "can remove a busy handler" do
+    bht = BusyHandlerTest.new
+
+    @write_db.busy_handler( bht )
+
+    # put a read lock on the database
+    @read_db.transaction( "DEFERRED" )
+
+    # put a read lock on the database, but want to go to an exclusive
+    @write_db.transaction( "IMMEDIATE" )
+
+    # do a read operation
+    @read_db.execute("SELECT count(*) FROM subcountry")
+
+    # attempt to do a write operation and commit it
+    @write_db.execute("DELETE FROM subcountry")
+    @write_db.remove_busy_handler
+    lambda { @write_db.execute("COMMIT"); }.should raise_error( ::Libsql::SQLite3::Error, /database is locked/ )
+    bht.call_count.should == 0
+  end
+
+end

data/spec/data/iso-3166-country.txt

@@ -0,0 +1,242 @@
+Afghanistan|AF|4
+Albania|AL|8
+Algeria|DZ|12
+American Samoa|AS|16
+Andorra|AD|20
+Angola|AO|24
+Anguilla|AI|660
+Antarctica|AQ|10
+Antigua and Barbuda|AG|28
+Argentina|AR|32
+Armenia|AM|51
+Aruba|AW|533
+Australia|AU|36
+Austria|AT|40
+Azerbaijan|AZ|31
+Bahamas|BS|44
+Bahrain|BH|48
+Bangladesh|BD|50
+Barbados|BB|52
+Belarus|BY|112
+Belgium|BE|56
+Belize|BZ|84
+Benin|BJ|204
+Bermuda|BM|60
+Bhutan|BT|64
+Bolivia|BO|68
+Bosnia and Herzegowina|BA|70
+Botswana|BW|72
+Bouvet Island|BV|74
+Brazil|BR|76
+British Indian Ocean Territory|IO|86
+Brunei Darussalam|BN|96
+Bulgaria|BG|100
+Burkina Faso|BF|854
+Burundi|BI|108
+Cambodia|KH|116
+Cameroon|CM|120
+Cape Verde|CV|132
+Cayman Islands|KY|136
+Central African Republic|CF|140
+Chad|TD|148
+Chile|CL|152
+China|CN|156
+Christmas Island|CX|162
+Cocos (Keeling) Islands|CC|166
+Colombia|CO|170
+Comoros|KM|174
+Congo|CG|178
+Congo, The Democratic Republic of the|CD|180
+Cook Islands|CK|184
+Costa Rica|CR|188
+Cote D'Ivoire|CI|384
+Croatia (local name: Hrvatska)|HR|191
+Cuba|CU|192
+Cyprus|CY|196
+Czech Republic|CZ|203
+Denmark|DK|208
+Djibouti|DJ|262
+Dominica|DM|212
+Dominican Republic|DO|214
+East Timor|TP|626
+Ecuador|EC|218
+Egypt|EG|818
+El Salvador|SV|222
+Equatorial Guinea|GQ|226
+Eritrea|ER|232
+Estonia|EE|233
+Ethiopia|ET|231
+Falkland Islands (Malvinas)|FK|238
+Faroe Islands|FO|234
+Fiji|FJ|242
+Finland|FI|246
+France|FR|250
+France, Metropolitan|FX|249
+French Guiana|GF|254
+French Polynesia|PF|258
+French Southern Territories|TF|260
+Gabon|GA|266
+Gambia|GM|270
+Georgia|GE|268
+Germany|DE|276
+Ghana|GH|288
+Gibraltar|GI|292
+Greece|GR|300
+Greenland|GL|304
+Grenada|GD|308
+Guadeloupe|GP|312
+Guam|GU|316
+Guatemala|GT|320
+Guinea|GN|324
+Guinea-Bissau|GW|624
+Guyana|GY|328
+Haiti|HT|332
+Heard and McDonald Islands|HM|334
+Holy See (Vatican City State)|VA|336
+Honduras|HN|340
+Hong Kong|HK|344
+Hungary|HU|348
+Iceland|IS|352
+India|IN|356
+Indonesia|ID|360
+Iran (Islamic Republic of)|IR|364
+Iraq|IQ|368
+Ireland|IE|372
+Israel|IL|376
+Italy|IT|380
+Jamaica|JM|388
+Japan|JP|392
+Jordan|JO|400
+Kazakhstan|KZ|398
+Kenya|KE|404
+Kiribati|KI|296
+Korea, Democratic People's Republic of|KP|408
+Korea, Republic of|KR|410
+Kuwait|KW|414
+Kyrgyzstan|KG|417
+Lao People's Democratic Republic|LA|418
+Latvia|LV|428
+Lebanon|LB|422
+Lesotho|LS|426
+Liberia|LR|430
+Libyan Arab Jamahiriya|LY|434
+Liechtenstein|LI|438
+Lithuania|LT|440
+Luxembourg|LU|442
+Macau|MO|446
+Macedonia, the Former Yugoslav Republic of|MK|807
+Madagascar|MG|450
+Malawi|MW|454
+Malaysia|MY|458
+Maldives|MV|462
+Mali|ML|466
+Malta|MT|470
+Marshall Islands|MH|584
+Martinique|MQ|474
+Mauritania|MR|478
+Mauritius|MU|480
+Mayotte|YT|175
+Micronesia, Federated States of|FM|583
+Moldova, Republic of|MD|498
+Monaco|MC|492
+Mongolia|MN|496
+Montserrat|MS|500
+Morocco|MA|504
+Mozambique|MZ|508
+Myanmar|MM|104
+Namibia|NA|516
+Nauru|NR|520
+Nepal|NP|524
+Netherlands|NL|528
+Netherlands Antilles|AN|530
+New Caledonia|NC|540
+New Zealand|NZ|554
+Nicaragua|NI|558
+Niger|NE|562
+Nigeria|NG|566
+Niue|NU|570
+Norfolk Island|NF|574
+Northern Mariana Islands|MP|580
+Norway|NO|578
+Oman|OM|512
+Pakistan|PK|586
+Palau|PW|585
+Palestinian Territory, Occupied|PS|275
+Panama|PA|591
+Papua New Guinea|PG|598
+Paraguay|PY|600
+Peru|PE|604
+Philippines|PH|608
+Pitcairn|PN|612
+Poland|PL|616
+Portugal|PT|620
+Puerto Rico|PR|630
+Qatar|QA|634
+Reunion|RE|638
+Romania|RO|642
+Russian Federation|RU|643
+Rwanda|RW|646
+Saint Kitts and Nevis|KN|659
+Saint Lucia|LC|662
+Saint Vincent and the Grenadines|VC|670
+Samoa|WS|882
+San Marino|SM|674
+Sao Tome and Principe|ST|678
+Saudi Arabia|SA|682
+Senegal|SN|686
+Serbia and Montenegro|CS|891
+Seychelles|SC|690
+Sierra Leone|SL|694
+Singapore|SG|702
+Slovakia (Slovak Republic)|SK|703
+Slovenia|SI|705
+Solomon Islands|SB|90
+Somalia|SO|706
+South Africa|ZA|710
+South Georgia and the South Sandwich Islands|GS|239
+Spain|ES|724
+Sri Lanka|LK|144
+St. Helena|SH|654
+St. Pierre and Miquelon|PM|666
+Sudan|SD|736
+Suriname|SR|740
+Svalbard and Jan Mayen Islands|SJ|744
+Swaziland|SZ|748
+Sweden|SE|752
+Switzerland|CH|756
+Syrian Arab Republic|SY|760
+Taiwan, Province of China|TW|158
+Tajikistan|TJ|762
+Tanzania, United Republic of|TZ|834
+Thailand|TH|764
+Timor-Leste|TL|626
+Togo|TG|768
+Tokelau|TK|772
+Tonga|TO|776
+Trinidad and Tobago|TT|780
+Tunisia|TN|788
+Turkey|TR|792
+Turkmenistan|TM|795
+Turks and Caicos Islands|TC|796
+Tuvalu|TV|798
+Uganda|UG|800
+Ukraine|UA|804
+United Arab Emirates|AE|784
+United States Minor Outlying Islands|UM|581
+Uruguay|UY|858
+Uzbekistan|UZ|860
+Vanuatu|VU|548
+Venezuela|VE|862
+Viet Nam|VN|704
+Virgin Islands (British)|VG|92
+Virgin Islands (U.S.)|VI|850
+Wallis and Futuna Islands|WF|876
+Western Sahara|EH|732
+Yemen|YE|887
+Yugoslavia|YU|891
+Zambia|ZM|894
+Zimbabwe|ZW|716
+Canada|CA|124
+Mexico|MX|484
+United Kingdom|GB|826
+United States|US|840

data/spec/data/iso-3166-schema.sql

@@ -0,0 +1,22 @@
+
+-- SET client_encoding TO 'UNICODE';
+
+CREATE TABLE country (
+    name TEXT NOT NULL,
+    two_letter TEXT PRIMARY KEY,
+    id integer NOT NULL
+);
+CREATE INDEX country_name  ON country(name);
+
+
+CREATE TABLE subcountry (
+    country TEXT NOT NULL REFERENCES country(two_letter),
+    name TEXT,
+    subdivision TEXT,
+    level TEXT,
+    UNIQUE(country, name)
+);
+
+CREATE INDEX subcountry_country ON subcountry(country);
+CREATE INDEX subcountry_name ON subcountry(name);
+