Merge pull request #3 from maselvaraj/master

[JENKINS-10544] Persist ShelveProject property whenever the job is updated
jenkinsci · Mar 21, 2014 · ea6afb8 · ea6afb8
2 parents 2df406a + 634a524
commit ea6afb8
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 5 deletions.
diff --git a/src/main/java/org/jvnet/hudson/plugins/shelveproject/ItemListenerImpl.java b/src/main/java/org/jvnet/hudson/plugins/shelveproject/ItemListenerImpl.java
@@ -31,13 +31,14 @@ public void onLoaded()
     @Override
     public void onCreated( Item item )
     {
-        if ( item instanceof AbstractProject )
-        {
-            AbstractProject project = (AbstractProject) item;
-            addShelveProjectProperty( project );
-        }
+		addShelveProjectPropertyIfAbstractProject(item);
     }
 
+	@Override
+	public void onUpdated(Item item) {
+		addShelveProjectPropertyIfAbstractProject(item);
+	}
+
     private void addShelveProjectProperty( AbstractProject<?, ?> project )
     {
         try
@@ -52,4 +53,11 @@ private void addShelveProjectProperty( AbstractProject<?, ?> project )
             LOGGER.log( Level.SEVERE, "Failed to persist " + project, e );
         }
     }
+
+    private void addShelveProjectPropertyIfAbstractProject(Item item) {
+		if (item instanceof AbstractProject) {
+			AbstractProject project = (AbstractProject) item;
+			addShelveProjectProperty(project);
+		}
+	}
 }
diff --git a/src/test/java/org/jvnet/hudson/plugins/shelveproject/ItemListenerImplTest.java b/src/test/java/org/jvnet/hudson/plugins/shelveproject/ItemListenerImplTest.java
@@ -84,4 +84,16 @@ public void testOnCreate_shouldAddShelveProjectActionForNewProjects()
         assertEquals( "Project should have had a ShelveProjectAction added.", 1,
                       freeStyleProject.getActions( ShelveProjectAction.class ).size() );
     }
+
+    public void testOnUpdate_shouldAddShelveProjectActionForUpdatedProjects()
+        throws IOException
+    {
+        FreeStyleProject freeStyleProject = createFreeStyleProject( "Goofy Project" );
+
+        itemListener.onUpdated( freeStyleProject );
+
+        assertEquals( "Project should have been added just once.", 1, freeStyleProject.getActions().size() );
+        assertEquals( "Project should have had a ShelveProjectAction added.", 1,
+                      freeStyleProject.getActions( ShelveProjectAction.class ).size() );
+    }
 }