spec/db_spec.rb in chatterbot-0.6.3 vs spec/db_spec.rb in chatterbot-0.6.5

@@ -2,40 +2,46 @@
 
 require 'sequel'
 
 describe "Chatterbot::DB" do
   before(:each) do
-    @db_tmp = Tempfile.new("config.db")
-    @db_uri = "sqlite://#{@db_tmp.path}"
-
-    @bot = Chatterbot::Bot.new    
+    @db_uri = "sqlite:/tmp/chatterbot.db"
+    File.delete("/tmp/chatterbot.db") if File.exist?("/tmp/chatterbot.db")
+    
+    @bot = Chatterbot::Bot.new
+    @bot.stub!(:update_config_at_exit)
     @bot.config[:db_uri] = @db_uri
   end
 
   describe "get_connection" do
     it "should make sure sequel is actually installed" do
       @bot.stub!(:has_sequel?).and_return(false)
       @bot.should_receive(:display_db_config_notice)
       @bot.db
     end
   end
+
+  context "db interactions" do
+    after(:each) do
+      @bot.db.disconnect unless @bot.db.nil?
+    end
   
-  describe "table creation" do
-    [:blacklist, :tweets, :config].each do |table|
-      it "should create table #{table}" do
+    describe "table creation" do
+      [:blacklist, :tweets, :config].each do |table|
+        it "should create table #{table}" do
+          @tmp_conn = @bot.db
+          @tmp_conn.tables.include?(table).should == true
+        end
+      end      
+    end
+    
+    describe "store_database_config" do
+      it "doesn't fail" do
+        @bot = Chatterbot::Bot.new    
+        @bot.config[:db_uri] = @db_uri
+        
         @bot.db
-        @tmp_conn = Sequel.connect(@db_uri)
-        @tmp_conn.tables.include?(table).should == true
+        @bot.store_database_config.should == true
       end
-    end      
-  end
-  
-  describe "store_database_config" do
-    it "doesn't fail" do
-      @bot = Chatterbot::Bot.new    
-      @bot.config[:db_uri] = @db_uri
-
-      @bot.db      
-      @bot.store_database_config.should == true
     end
   end
 end