rally_workspace_utils 0.0.2 → 0.0.3

data/Manifest.txt

@@ -14,4 +14,6 @@ lib/translate/copy/test_case_result_copy.rb
 lib/translate/copy/defect_copy.rb
 lib/translate/copy/object_copy.rb
 lib/translate/copy/test_case_copy.rb
+lib/translate/translation_store.rb
+test/translation_store_spec.rb
 lib/rally_workspace_utils.rb

data/bin/translate_workspace

@@ -40,6 +40,7 @@ args = {:base_url => "https://rally1.rallydev.com/slm"}
 debug = false
 audit = false
 logfile = STDOUT
+$VERBOSE = nil
 
 opts.each do |opt, arg|
   case opt

data/lib/translate/copy/object_copy.rb

@@ -4,6 +4,7 @@ class ObjectCopy
   end
 
   def copy(new_workspace, additional_values = {})
+    @object.rally_rest.logger.info "Copying #{@object.type} -- #{@object.name}"
     values = shallow_copy(@object, new_workspace)
     values.merge! additional_values
 
@@ -18,8 +19,7 @@ class ObjectCopy
   end
 
   def remember old, new
-    $TRANSLATED ||= {}
-    $TRANSLATED[old.oid] = new.oid
+    TranslationStore[old.oid] = new.oid
   end
 
   def create_object(rally_rest, values)
@@ -85,7 +85,7 @@ class ObjectCopy
 
   def fetch_object(type, oid, workspace)
     @@cached_objects ||= {}
-    new_oid = $TRANSLATED[oid]
+    new_oid = TranslationStore[oid]
     return @@cached_objects[new_oid] if @@cached_objects.key? oid
     
     object = @object.rally_rest.find(type, :workspace => workspace) { equal :object_i_d, new_oid }.first

data/lib/translate/translate.rb

@@ -1,5 +1,6 @@
 require File.dirname(__FILE__) + '/copy'
 require File.dirname(__FILE__) + '/translation_audit'
+require File.dirname(__FILE__) + '/translation_store'
 
 class Translate
   include TranslationAudit
@@ -13,7 +14,6 @@ class Translate
     @from_workspace_name = args[:from_workspace]
     @to_workspace_name = args[:to_workspace]
     @logger = args[:logger] if args[:logger]
-    $TRANSLATED ||= {}
   end
 
   def run
@@ -25,41 +25,30 @@ class Translate
     raise "No such workspace #{@from_workspace_name}" unless @from_workspace
     raise "No such workspace #{@to_workspace_name}" unless @to_workspace
 
-
-    if File::exists? "TRANSLATED"
-      File.open("TRANSLATED", "r") do |f|
-	$TRANSLATED = Marshal.load(f)
-      end
-
-      if $TRANSLATED[:from] != @from_workspace.oid || $TRANSLATED[:to] != @to_workspace.oid
-	$TRANSLATED = {}
-	$TRANSLATED[:from] = @from_workspace.oid
-	$TRANSLATED[:to] = @to_workspace.oid
-      end
+    store = TranslationStore.instance(@from_workspace, @to_workspace)
+
+    begin 
+      # now do the copy
+      [:iteration, :release, :story, :defect, :test_case].each do |type|
+#	@slm.find_all(type, :workspace => @from_workspace, :order => [:creation_date, :desc]).each do |o|
+	@slm.find_all(type, :workspace => @from_workspace).each do |o|
+	  o.copy(@to_workspace) unless TranslationStore[o.oid]
+	end
+      end unless store.copy_finished?
+      store.copy_finished!
+
+      # Now tangle the objects
+      [:test_case, :defect].each do |type|
+	@slm.find_all(type, :workspace => @from_workspace).each do |o|
+	  o.tangle(@to_workspace)
+	end
+      end unless store.tangle_finished?
+      store.tangle_finished!
+    rescue Exception => e
+      @logger.info "Caught an exception #{e}. Remembering where we were."
+      # Remember where we were if there was a failure
+      store.dump
     end
-
-    # now do the copy
-    [:iteration, :release, :story, :defect, :test_case].each do |type|
-      @slm.find_all(type, :workspace => @from_workspace).each do |o|
-	@logger.info "Copying #{o.type} -- #{o.name}"
-	o.copy(@to_workspace)
-      end
-    end unless $TRANSLATED[:copy]
-    # mark these workspaces as copied
-    $TRANSLATED[:copy] = true
-    # dump the oids to disk
-    dump_oids
-
-    # Now tangle the objects
-    [:test_case, :defect].each do |type|
-      @slm.find_all(type, :workspace => @from_workspace).each do |o|
-	@logger.info "Tangle #{o.type} -- #{o.name}"
-	o.tangle(@to_workspace)
-      end
-    end unless $TRANSLATED[:tangle]
-    
-    $TRANSLATED[:tangle] = true
-    dump_oids
   end
 
   def dump_oids

data/lib/translate/translation_audit.rb

@@ -55,7 +55,7 @@ module TranslationAudit
 
   def compare_each(type)
     @slm.find_all(type, :workspace => @from_workspace).each do |old_object|
-      new_object = @slm.find(type, :workspace => @to_workspace) { equal :object_i_d, $TRANSLATED[old_object.oid] }.first
+      new_object = @slm.find(type, :workspace => @to_workspace) { equal :object_i_d, TranslationStore[old_object.oid] }.first
       if new_object.nil?
 	self.logger.info "Cound not find a copy for object #{old_object.type} -- #{old_object.name} -- #{old_object.oid}"
 	next

data/lib/translate/translation_store.rb

@@ -0,0 +1,76 @@
+
+class TranslationStore
+  private_class_method :new
+
+  def TranslationStore.instance(from_workspace = nil, to_workspace = nil)
+    unless defined? @@instance
+      raise "TranslationStory needs to be initialized with two workspaces" unless from_workspace && to_workspace
+      @@instance = new(from_workspace, to_workspace)
+    end
+    @@instance
+  end
+
+  def TranslationStore.[](key)
+    instance[key]
+  end
+
+  def TranslationStore.[]=(key, value)
+    instance[key] = value
+  end
+
+  def initialize(w1, w2)
+    @w1 = w1.oid
+    @w2 = w2.oid
+    @store = {}
+    @store[:copy] = false
+    @store[:tangle] = false
+    load
+  end
+
+  def [](key)
+    @store[key]
+  end
+
+  def []=(key, value)
+    @store[key] = value
+  end
+
+  def copy_finished!
+    @store[:copy] = true
+    dump
+  end
+
+  def copy_finished?
+      @store[:copy] == true
+  end
+
+  def tangle_finished!
+    @store[:tangle] = true
+    dump
+  end
+
+  def tangle_finished?
+      @store[:tangle] == true
+  end
+
+  def dump
+    File.open(filename, "w+") do |f|
+      Marshal.dump(@store, f)
+    end
+  end
+
+  def load
+    File.open(filename, "r") do |f|
+      @store = Marshal.load(f)
+    end if File.exists? filename
+  end
+
+  def filename
+    "translate-#{@w1}-#{@w2}"
+  end
+
+  def cleanup
+    File.delete(filename) if File.exists? filename
+  end
+  
+end

data/lib/translate/version.rb

@@ -2,7 +2,7 @@ module Translate #:nodoc:
   module VERSION #:nodoc:
     MAJOR = 0
     MINOR = 0
-    TINY  = 2
+    TINY  = 3
 
     STRING = [MAJOR, MINOR, TINY].join('.')
   end

data/test/translation_store_spec.rb

@@ -0,0 +1,123 @@
+
+require 'rubygems'
+require 'spec'
+
+require File.dirname(__FILE__) + '/../lib/translate/translation_store'
+
+context "A TranslationStore singleton" do
+  
+  setup do
+    @workspace1 = Object.new
+    @workspace1.stub!(:oid).and_return(12345)
+    @workspace2 = Object.new
+    @workspace2.stub!(:oid).and_return(54321)
+  end
+
+  teardown do
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+  end
+
+  specify "should only have a single instance" do
+    lambda { TranslationStore.new }.should_raise Exception
+    TranslationStore.should_respond_to :instance
+  end
+  
+  specify "singleton should be initialized with two workspaces" do
+
+    lambda { TranslationStore.instance }.should_raise Exception
+    lambda { TranslationStore.instance(nil, nil) }.should_raise Exception
+    lambda { TranslationStore.instance(nil, @workspace1) }.should_raise Exception
+    lambda { TranslationStore.instance(@workspace1, nil) }.should_raise Exception
+    lambda { TranslationStore[1] = 2 }.should_raise Exception
+    lambda { TranslationStore.instance(@workspace1, @workspace2) }.should_not_raise Exception
+  end
+
+  specify "singleton should return an instance of TranslationStory when initialized with two workspaces" do
+    TranslationStore.instance(@workspace1, @workspace2).should_be_instance_of TranslationStore
+  end
+end
+
+context "A TranslationStore" do
+  setup do
+    @workspace1 = Object.new
+    @workspace1.stub!(:oid).and_return(12345)
+    @workspace2 = Object.new
+    @workspace2.stub!(:oid).and_return(54321)
+    @store = TranslationStore.instance(@workspace1, @workspace2)
+  end
+
+  teardown do
+    @store.cleanup
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+  end
+
+  specify "should store mappings of old oids to new oids" do
+    @store[12345] = 54321
+    @store[12345].should_be 54321
+    
+    TranslationStore[1] = 2
+    TranslationStore[1].should_be 2
+  end
+
+  specify "should write all contents to a file" do
+    @store[1] = 2
+    @store.dump
+    File.exists?("translate-12345-54321").should_be true
+  end
+
+  specify "should cleanup file" do
+    @store[1] = 2
+    @store.dump
+    File.exists?("translate-12345-54321").should_be true
+    @store.cleanup
+    File.exists?("translate-12345-54321").should_be false
+  end
+
+  specify "should read contents from a file" do
+    @store[1] = 2
+    @store.dump
+    File.exists?("translate-12345-54321").should_be true
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+    @store = TranslationStore.instance(@workspace1, @workspace2)
+    @store[1].should_be 2
+  end
+
+  specify "should read from a file, only when the workspace oids match" do
+    @store[1] = 2
+    @store.dump
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+    
+    @workspace2 = Object.new
+    @workspace2.stub!(:oid).and_return(11111)
+    @store = TranslationStore.instance(@workspace1, @workspace2)
+    @store.load
+    @store[1].should_be nil
+  end
+
+  specify "should remember when the copy is finished" do
+    @store.copy_finished?.should_be false
+    @store.copy_finished!
+    @store.copy_finished?.should_be true
+
+    @store.dump
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+    @store = TranslationStore.instance(@workspace1, @workspace2)
+    @store.load
+
+    @store.copy_finished?.should_be true
+  end
+
+  specify "should remember when the tangle is finished" do
+    @store.tangle_finished?.should_be false
+    @store.tangle_finished!
+    @store.tangle_finished?.should_be true
+
+    @store.dump
+    TranslationStore.send(:remove_class_variable, :@@instance) rescue ""
+    @store = TranslationStore.instance(@workspace1, @workspace2)
+    @store.load
+
+    @store.tangle_finished?.should_be true
+  end
+
+end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.0
 specification_version: 1
 name: rally_workspace_utils
 version: !ruby/object:Gem::Version 
-  version: 0.0.2
-date: 2006-12-19 00:00:00 -07:00
+  version: 0.0.3
+date: 2006-12-20 00:00:00 -07:00
 summary: A utility to translate a UseCase workspace to a UserStory workspace
 require_paths: 
 - lib
@@ -45,6 +45,8 @@ files:
 - lib/translate/copy/defect_copy.rb
 - lib/translate/copy/object_copy.rb
 - lib/translate/copy/test_case_copy.rb
+- lib/translate/translation_store.rb
+- test/translation_store_spec.rb
 - lib/rally_workspace_utils.rb
 test_files: 
 - test/rally_workspace_utils_test.rb