[registry] improved tests, sqlite/tc bugfixes

3 files changed, 25 insertions, 6 deletions

diff --git a/lib/rbot/registry.rb b/lib/rbot/registry.rb
index 70b450ac..c4e50faf 100644
--- a/lib/rbot/registry.rb
+++ b/lib/rbot/registry.rb
@@ -222,8 +222,8 @@ class Registry
     # like Hash#each
     def each(&block)
       return nil unless dbexists?
-      registry.each_key do |key|
-        block.call(key, self[key])
+      registry.each do |key, value|
+        block.call(key, restore(value))
       end
     end
 
@@ -268,6 +268,7 @@ class Registry
     end
 
     # delete a key from the registry
+    # returns the value in success, nil otherwise
     def delete(key)
       return default unless dbexists?
       value = registry.delete(key.to_s)
diff --git a/lib/rbot/registry/sqlite.rb b/lib/rbot/registry/sqlite.rb
index 1af5d242..6a4a2312 100644
--- a/lib/rbot/registry/sqlite.rb
+++ b/lib/rbot/registry/sqlite.rb
@@ -23,7 +23,7 @@ class Registry
         begin
           @registry.execute('SELECT COUNT(*) FROM data')
         rescue
-          @registry.execute('CREATE TABLE data (key string, value blob)')
+          @registry.execute('CREATE TABLE data (key PRIMARY KEY, value)')
         end
       end
       @registry
@@ -37,7 +37,7 @@ class Registry
     def [](key)
       if dbexists?
         begin
-          value = @registry.get_first_row('SELECT value FROM data WHERE key = ?', key.to_s)
+          value = registry.get_first_row('SELECT value FROM data WHERE key = ?', key.to_s)
           return restore(value.first)
         rescue
           return default
@@ -65,12 +65,18 @@ class Registry
       end
     end
 
+    alias each_pair each
+
     def has_key?(key)
       return nil unless dbexists?
       res = registry.get_first_row('SELECT COUNT(*) FROM data WHERE key = ?', key.to_s)
       return res.first > 0
     end
 
+    alias include? has_key?
+    alias member? has_key?
+    alias key? has_key?
+
     def has_value?(value)
       return nil unless dbexists?
       value = SQLite3::Blob.new(store(value))
@@ -81,10 +87,11 @@ class Registry
     def delete(key)
       return default unless dbexists?
       begin
+        value = self[key]
         registry.execute('DELETE FROM data WHERE key = ?', key.to_s)
-        registry.changes > 0
+        value if registry.changes > 0
       rescue
-        false
+        nil
       end
     end
 
@@ -100,6 +107,8 @@ class Registry
       registry.execute('DELETE FROM data')
     end
 
+    alias truncate clear
+
     # returns the number of keys in your registry namespace
     def length
       return 0 unless dbexists?
@@ -107,6 +116,8 @@ class Registry
       res.first
     end
 
+    alias size length
+
   end
 
 end # Registry
diff --git a/lib/rbot/registry/tc.rb b/lib/rbot/registry/tc.rb
index 4e248f42..f1835c8a 100644
--- a/lib/rbot/registry/tc.rb
+++ b/lib/rbot/registry/tc.rb
@@ -41,6 +41,13 @@ class Registry
       @registry.optimize
     end
 
+    def delete(key)
+      return default unless dbexists?
+      value = self[key]
+      registry.delete(key.to_s)
+      value # return deleted value if success
+    end
+
   end
 
 end # Registry