lib/tasks/fauna.rake in fauna-0.2.6 vs lib/tasks/fauna.rake in fauna-1.1.0

@@ -1,11 +1,71 @@
 
 task :environment
 
 namespace :fauna do
-  desc "Migrate your fauna database to the latest version"
+  desc "Migrate your Fauna database to the latest version of your schema"
   task :migrate => :environment do
+    puts "Migrating #{Rails.env} Fauna database schema"
     Fauna::Client.context(Fauna.connection) do
       Fauna.migrate_schema!
     end
   end
+
+  desc "Completely reset your Fauna database"
+  task :reset => :environment do
+    if Rails.env.production?
+      puts "Won't reset #{Rails.env} Fauna database"
+    else
+      Fauna::Client.context(Fauna.root_connection) do
+        puts "Resetting #{Rails.env} Fauna database"
+        Fauna::Client.delete("everything")
+      end
+      Fauna.auth!
+    end
+  end
+
+  desc "Dump the contents of your Fauna database to 'test/fixtures/fauna'"
+  task :dump => :environment do
+    puts "Dumping #{Rails.env} Fauna database contents to #{Fauna::FIXTURES_DIR}"
+    Fauna::Client.context(Fauna.connection) do
+      FileUtils.mkdir_p(Fauna::FIXTURES_DIR)
+
+      class_configs = Fauna.connection.get("/classes")["references"] || {}
+      class_configs["users"] = nil
+
+      Dir.chdir(Fauna::FIXTURES_DIR) do
+        class_configs.each do |ref, value|
+          class_name = ref[0..-8]
+          FileUtils.mkdir_p(class_name)
+
+          Dir.chdir(class_name) do
+            # FIXME shouldn't round trip JSON
+            File.open("config.json", "w") { |f| f.write(value.to_json)} if value
+            (Fauna.connection.get(class_name)["references"] || {}).each do |ref, value|
+              File.open("#{ref.split("/").last}.json", "w") { |f| f.write(value.to_json) }
+            end
+          end
+        end
+      end
+    end
+  end
+
+  desc "Load the contents of your Fauna database"
+  task :load => :environment do
+    puts "Loading #{Rails.env} Fauna database contents from #{Fauna::FIXTURES_DIR}"
+    Fauna::Client.context(Fauna.connection) do
+      Dir.chdir(Fauna::FIXTURES_DIR) do
+        Dir["**/*.json"].map do |filename|
+          value = JSON.parse(File.open(filename) { |f| f.read })
+          begin
+            Fauna.connection.put(value["ref"], value)
+          rescue Fauna::Connection::NotFound
+            Fauna.connection.post(value["ref"].split("/")[0..-2].join("/"), value)
+          end
+        end
+      end
+    end
+  end
+
 end
+
+task :test => ["fauna:reset", "fauna:migrate", "fauna:load"]