Add validations. Fixes specs

1 files changed, 8 insertions, 2 deletions

diff --git a/app/models/referential.rb b/app/models/referential.rb
index b08f6be4c..d0fdbd676 100644
--- a/app/models/referential.rb
+++ b/app/models/referential.rb
@@ -17,14 +17,20 @@ class Referential < ActiveRecord::Base
     Chouette::Company.scoped
   end
 
+  def switch
+    raise "Referential not created" if new_record?
+    Apartment::Database.switch(slug)
+    self
+  end
+
   private 
 
   def create_schema
-    Apartment::Database.create self.slug
+    Apartment::Database.create slug
   end
 
   def destroy_schema
-    Apartment::Database.drop self.slug
+    Apartment::Database.drop slug
   end
   
 end