lib/tasks/app_base_dir.rake in daddy-0.3.15 vs lib/tasks/app_base_dir.rake in daddy-0.3.16

@@ -1,14 +1,17 @@
-require 'rake'
-require 'rails'
+require_relative 'task_helper'
 
 namespace :dad do
-  task :app_base_dir do
-    app_base_dir = File.join('/var/lib', Daddy.config.application)
-    fail unless system("sudo mkdir -p #{app_base_dir}")
-    fail unless system("sudo chown root:root #{app_base_dir}")
 
-    env_dir = File.join(app_base_dir, Rails.env)
-    fail unless system("sudo mkdir -p #{env_dir}")
-    fail unless system("sudo chown #{ENV['USER']}:#{ENV['USER']} #{env_dir}")
+  task :app_base_dir do
+    case Rails.env
+    when 'production', 'development'
+      app_base_dir = File.join('/var/lib', Daddy.config.application, Rails.env)
+      run "sudo mkdir -p #{app_base_dir}",
+          "sudo chown #{ENV['USER']}:#{ENV['USER']} #{app_base_dir}"
+    when 'test'
+      app_base_dir = File.join('/tmp', Daddy.config.application, Rails.env)
+      run "mkdir -p #{app_base_dir}"
+    end
   end
+
 end