lib/scenarios/scenario_db.rb in scenario_server-0.0.4 vs lib/scenarios/scenario_db.rb in scenario_server-0.0.6

@@ -141,9 +141,13 @@
     self.db ||= Sequel.sqlite(self.db_file)
     self.scenarios ||= self.db[:scenarios]
     self.routes ||= self.db[:routes]
     self.testdata ||= self.db[:testdata]
   end
+  
+  def close_database
+      self.db.disconnect
+  end
 
   #scenarios
 
   def create_scenarios_table
     # Create Scenarios table