Add support for 'include_all_branches' flag when creating repository from a template

Author: leowebb

src/main/java/org/kohsuke/github/GHCreateRepositoryBuilder.java

@@ -98,6 +98,18 @@ public GHCreateRepositoryBuilder gitignoreTemplate(String language) throws IOExc
         return with("gitignore_template", language);
     }
 
+    /**
+     * Include all branches when creating from a template repository
+     *
+     * @param includeAllBranches
+     *            whether or not to include all branches from the template repository
+     * @return a builder to continue with building
+     * @throws IOException
+     */
+    public GHCreateRepositoryBuilder includeAllBranches(boolean includeAllBranches) throws IOException {
+        return with("include_all_branches", includeAllBranches);
+    }
+
     /**
      * Desired license template to apply.
      *

src/test/java/org/kohsuke/github/GHOrganizationTest.java

@@ -44,21 +44,21 @@ public GHOrganizationTest() {
      * @throws IOException
      *             Signals that an I/O exception has occurred.
      */
-    @Before
-    @After
-    public void cleanUpTeam() throws IOException {
-        // Cleanup is only needed when proxying
-        if (!mockGitHub.isUseProxy()) {
-            return;
-        }
+     @Before
+     @After
+     public void cleanUpTeam() throws IOException {
+     // Cleanup is only needed when proxying
+     if (!mockGitHub.isUseProxy()) {
+     return;
+     }
 
-        GHTeam team = getNonRecordingGitHub().getOrganization(GITHUB_API_TEST_ORG).getTeamByName(TEAM_NAME_CREATE);
-        if (team != null) {
-            team.delete();
-        }
+     GHTeam team = getNonRecordingGitHub().getOrganization(GITHUB_API_TEST_ORG).getTeamByName(TEAM_NAME_CREATE);
+     if (team != null) {
+     team.delete();
+     }
 
-        getNonRecordingGitHub().getOrganization(GITHUB_API_TEST_ORG).enableOrganizationProjects(true);
-    }
+     getNonRecordingGitHub().getOrganization(GITHUB_API_TEST_ORG).enableOrganizationProjects(true);
+     }
 
     /**
      * Test are organization projects enabled.
@@ -264,6 +264,37 @@ public void testCreateRepositoryWithTemplateAndGHRepository() throws IOException
 
     }
 
+    /**
+     * Test create a repository from a template with all branches included
+     *
+     * @throws IOException
+     *             Signals that an I/O exception has occurred.
+     * @throws InterruptedException
+     *             Signals that Thread.sleep() was interrupted
+     */
+
+    @Test
+    public void testCreateRepositoryWithTemplateAndIncludeAllBranches() throws IOException, InterruptedException {
+        cleanupRepository(GITHUB_API_TEST_ORG + '/' + GITHUB_API_TEST);
+
+        GHOrganization org = gitHub.getOrganization(GITHUB_API_TEST_ORG);
+        GHRepository templateRepository = org.getRepository(GITHUB_API_TEMPLATE_TEST);
+
+        GHRepository repository = gitHub.createRepository(GITHUB_API_TEST)
+                .fromTemplateRepository(templateRepository)
+                .includeAllBranches(true)
+                .owner(GITHUB_API_TEST_ORG)
+                .create();
+
+        assertThat(repository, notNullValue());
+
+        // give it a moment for branches to be created
+        Thread.sleep(1500);
+
+        assertThat(repository.getBranches().keySet(), equalTo(templateRepository.getBranches().keySet()));
+
+    }
+
     /**
      * Test create team.
      *