[#4184] Got the most basic objects ported from Budget to Contracts.

test/integration/budget_plugin_migration_test.rb

@@ -12,6 +12,18 @@ class BudgetPluginMigrationTest < ActionController::IntegrationTest
   context "migrate" do
     setup do
       @data = load_test_fixture
+      @project_one = Project.generate!
+      @project_two = Project.generate!
+      # Stub out the Project finders because we need to match the ids but can't
+      # be sure what ids object_daddy will give us
+      Project.stubs(:find_by_id).with(1).returns(@project_one)
+      Project.stubs(:find_by_id).with(2).returns(@project_one)
+
+      @manager = User.generate!
+      @role = Role.generate!
+      User.add_to_project(@manager, @project_one, @role)
+      User.add_to_project(@manager, @project_two, @role)
+
     end
 
     should "load a YAML dump of the old budget data" do
@@ -20,9 +32,32 @@ class BudgetPluginMigrationTest < ActionController::IntegrationTest
       assert_equal @data, RedmineContracts::BudgetPluginMigration.data
     end
     
-    should "create a new Deliverable for each old Deliverable"
+    should "create a new Deliverable for each old Deliverable" do
+
+      assert_difference("Deliverable.count", 3) do
+        assert_difference("HourlyDeliverable.count", 2) do
+          assert_difference("FixedDeliverable.count", 1) do
+            RedmineContracts::BudgetPluginMigration.migrate(@data)
+          end
+        end
+      end
+
+    end
     
-    should "create a new Contract for each project that had an old deliverable"
+    should "create a new Contract for each project that had an old deliverable" do
+      assert_difference("Contract.count", 2) do
+        RedmineContracts::BudgetPluginMigration.migrate(@data)
+      end
+
+      assert_equal 1, @project_one.reload.contracts.first.deliverables.count
+      assert_equal 2, @project_two.reload.contracts.first.deliverables.count
+    end
+
+    should "pick the first project member for the deliverable manager" do
+      RedmineContracts::BudgetPluginMigration.migrate(@data)
+
+      assert_equal [@manager, @manager, @manager], Deliverable.all.collect(&:manager)
+    end
   end
 end