From d2d1205ece0dac80e4b09ec3f664fcdd81bdac9d Mon Sep 17 00:00:00 2001
From: Fedor Yudanov <fedwiz@academ.org>
Date: Thu, 4 Apr 2013 20:31:46 +0700
Subject: [PATCH] #50 - advanced tests (test 13)

---
 rbejdb/src/rbejdb.c |  9 +++++++--
 rbejdb/test/t2.rb   | 15 +++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/rbejdb/src/rbejdb.c b/rbejdb/src/rbejdb.c
index d70ab72..b70c15d 100644
--- a/rbejdb/src/rbejdb.c
+++ b/rbejdb/src/rbejdb.c
@@ -116,7 +116,12 @@ void EJDB_close(VALUE self) {
     ejdbclose(ejdb);
 }
 
-void EJDB_drop_collection(VALUE self, VALUE collName, VALUE prune) {
+void EJDB_drop_collection(int argc, VALUE* argv, VALUE self) {
+    VALUE collName;
+    VALUE prune;
+
+    rb_scan_args(argc, argv, "11", &collName, &prune);
+
     SafeStringValue(collName);
 
     EJDB* ejdb = getEJDB(self);
@@ -588,7 +593,7 @@ Init_rbejdb() {
     rb_define_method(ejdbClass, "update", RUBY_METHOD_FUNC(EJDB_update), -1);
     rb_define_method(ejdbClass, "remove", RUBY_METHOD_FUNC(EJDB_remove), 2);
 
-    rb_define_method(ejdbClass, "drop_collection", RUBY_METHOD_FUNC(EJDB_drop_collection), 2);
+    rb_define_method(ejdbClass, "drop_collection", RUBY_METHOD_FUNC(EJDB_drop_collection), -1);
     rb_define_method(ejdbClass, "ensure_collection", RUBY_METHOD_FUNC(EJDB_ensure_collection), -1);
 
     rb_define_method(ejdbClass, "drop_indexes", RUBY_METHOD_FUNC(EJDB_drop_indexes), 2);
diff --git a/rbejdb/test/t2.rb b/rbejdb/test/t2.rb
index 0aa4a36..42cefd7 100644
--- a/rbejdb/test/t2.rb
+++ b/rbejdb/test/t2.rb
@@ -338,5 +338,20 @@ class EJDBTestUnit < Test::Unit::TestCase
     assert_not_nil $jb
     assert $jb.is_open?
     $jb.sync
+    puts "test_ejdbc_sync has passed successfull"
   end
+
+  def test_ejdbd_remove_colls
+    assert_not_nil $jb
+    assert $jb.is_open?
+
+    $jb.drop_collection("birds")
+
+    results = $jb.find("birds", {})
+    assert_equal(0, results.count)
+    assert_nil results.find { true }
+
+    puts "test_ejdbd_remove_colls has passed successfull"
+  end
+
 end
-- 
2.7.4