[BitBucket] Extend repo create method with is_private and fork_policy. (#776)

Author: Spacetown

atlassian/bitbucket/cloud/repositories/__init__.py

@@ -63,16 +63,28 @@ def each(self, after=None, role=None, q=None, sort=None):
 
 
 class WorkspaceRepositories(RepositoriesBase):
+    ALLOW_FORKS = "allow_forks"
+    NO_PUBLIC_FORKS = "no_public_forks"
+    NO_FORKS = "no_forks"
+    FORK_POLICIES = [
+        ALLOW_FORKS,
+        NO_PUBLIC_FORKS,
+        NO_FORKS,
+    ]
+
     def __init__(self, url, *args, **kwargs):
         super(WorkspaceRepositories, self).__init__(url, *args, **kwargs)
 
-    def create(self, repo_slug, project_key=None):
+    def create(self, repo_slug, project_key=None, is_private=None, fork_policy=None):
         """
         Creates a new repository with the given repo_slug.
 
         :param repo_slug: string: The repo_slug of the project.
         :param project_key: string: The key of the project. If the project is not provided, the repository
                                     is automatically assigned to the oldest project in the workspace.
+        :param is_private: boolean: Set to false if the repository shall be public.
+        :param fork_policy: string: The fork policy (one of WorkspaceRepositories.ALLOW_FORKS,
+                                    WorkspaceRepositories.NO_PUBLIC_FORKS, WorkspaceRepositories.NO_FORKS).
 
         :return: The created project object
 
@@ -82,6 +94,12 @@ def create(self, repo_slug, project_key=None):
         data = {"scm": "git"}
         if project_key is not None:
             data["project"] = {"key": project_key}
+        if is_private is not None:
+            data["is_private"] = is_private
+        if fork_policy is not None:
+            if fork_policy not in self.FORK_POLICIES:
+                raise ValueError("fork_policy must be one of {}".format(self.FORK_POLICIES))
+            data["fork_policy"] = fork_policy
         return self._get_object(self.post(repo_slug, data=data))
 
     def each(self, role=None, q=None, sort=None):
@@ -277,6 +295,11 @@ def is_private(self, is_private):
         """Setter for the repository private flag"""
         return self.update(is_private=is_private)
 
+    @property
+    def fork_policy(self):
+        """Getter for the repository fork policy"""
+        return self.get_data("fork_policy")
+
     @property
     def uuid(self):
         """The repository uuid"""

tests/responses/bitbucket/cloud/2.0/repositories/TestWorkspace1/TestRepositoryCreated/POST

@@ -80,10 +80,10 @@ responses['{"scm": "git"}'] = {
         "uuid": "{TestWorkspace1}",
     },
 }
-responses['{"scm": "git", "project": {"key": "TEST1"}}'] = {
+responses['{"scm": "git", "project": {"key": "TEST1"}, "fork_policy": "allow_forks"}'] = {
     "created_on": "2020-11-03T12:09:39.418592+00:00",
     "description": "Test newly created repository in project TEST1",
-    "fork_policy": "no_forks",
+    "fork_policy": "allow_forks",
     "full_name": "TestWorkspace1/TestRepositoryCreated",
     "has_issues": True,
     "has_wiki": False,

tests/test_bitbucket_cloud.py

@@ -1,4 +1,5 @@
 # coding: utf8
+from atlassian.bitbucket.cloud.repositories import WorkspaceRepositories
 import pytest
 import sys
 from datetime import datetime
@@ -48,10 +49,10 @@ def test_not_exists_repository(self):
     def test_create_repositories(self):
         result = CLOUD.workspaces.get("TestWorkspace1").repositories.create("TestRepositoryCreated")
         assert result.description == "Test newly created repository", "Result of repositories [create(...)]"
-        result = CLOUD.workspaces.get("TestWorkspace1").repositories.create("TestRepositoryCreated", "TEST1")
-        assert (
-            result.description == "Test newly created repository in project TEST1"
-        ), "Result of repositories [create(...)] with project"
+        result = CLOUD.workspaces.get("TestWorkspace1").repositories.create(
+            "TestRepositoryCreated", "TEST1", fork_policy=WorkspaceRepositories.ALLOW_FORKS
+        )
+        assert result.fork_policy == "allow_forks", "Result of repositories [create(...)] with project and fork policy"
 
     def test_get_repositories(self):
         result = [x["name"] for x in BITBUCKET.get_repositories("TestWorkspace1")]