python-gitlab
diff --git a/‎tests/functional/conftest.py
Lines changed: 27 additions & 1 deletion b/‎tests/functional/conftest.py
Lines changed: 27 additions & 1 deletion
diff --git a/‎tests/functional/fixtures/.env
Lines changed: 5 additions & 2 deletions b/‎tests/functional/fixtures/.env
Lines changed: 5 additions & 2 deletions
diff --git a/‎tests/functional/fixtures/Dockerfile
Lines changed: 8 additions & 0 deletions b/‎tests/functional/fixtures/Dockerfile
Lines changed: 8 additions & 0 deletions
diff --git a/‎tests/functional/fixtures/build-dockerfile.sh
Lines changed: 17 additions & 0 deletions b/‎tests/functional/fixtures/build-dockerfile.sh
Lines changed: 17 additions & 0 deletions
diff --git a/‎tests/functional/fixtures/create_license.rb
Lines changed: 51 additions & 0 deletions b/‎tests/functional/fixtures/create_license.rb
Lines changed: 51 additions & 0 deletions
diff --git a/‎tests/functional/fixtures/docker-compose.yml
Lines changed: 3 additions & 2 deletions b/‎tests/functional/fixtures/docker-compose.yml
Lines changed: 3 additions & 2 deletions
diff --git a/‎tox.ini
Lines changed: 2 additions & 0 deletions b/‎tox.ini
Lines changed: 2 additions & 0 deletions
@@ -15,6 +15,13 @@ def fixture_dir(test_dir):
 
 
 def reset_gitlab(gl):
+    if _is_gitlab_ee(gl):
+        # NOTE(jlvillal): By default in GitLab EE it will wait 7 days before
+        # deleting a group. Change it to 0 days.
+        settings = gl.settings.get()
+        if settings.deletion_adjourned_period != 0:
+            settings.deletion_adjourned_period = 0
+            settings.save()
     # previously tools/reset_gitlab.py
     for project in gl.projects.list():
         for deploy_token in project.deploytokens.list():
@@ -134,7 +141,7 @@ def gitlab_config(check_is_alive, docker_ip, docker_services, temp_dir, fixture_
     port = docker_services.port_for("gitlab", 80)
 
     docker_services.wait_until_responsive(
-        timeout=200, pause=5, check=lambda: check_is_alive("gitlab-test")
+        timeout=300, pause=5, check=lambda: check_is_alive("gitlab-test")
     )
 
     token = set_token("gitlab-test", fixture_dir=fixture_dir)
@@ -164,6 +171,25 @@ def gl(gitlab_config):
     return instance
 
 
+@pytest.fixture(scope="session")
+def is_gitlab_ee(gl) -> bool:
+    """Fixture to determine if we are running with GitLab EE as opposed to
+    GitLab CE"""
+    return _is_gitlab_ee(gl=gl)
+
+
+def _is_gitlab_ee(gl: gitlab.Gitlab) -> bool:
+    """Determine if we are running with GitLab EE as opposed to GitLab CE"""
+    try:
+        license = gl.get_license()
+    except gitlab.exceptions.GitlabLicenseError:
+        license = None
+    # If we have a license then we assume we are running on GitLab EE
+    if license:
+        return True
+    return False
+
+
 @pytest.fixture(scope="session")
 def gitlab_runner(gl):
     container = "gitlab-runner-test"
 
@@ -1,2 +1,5 @@
-GITLAB_IMAGE=gitlab/gitlab-ce
-GITLAB_TAG=14.5.2-ce.0
+GITLAB_IMAGE=gitlab/gitlab-ee
+GITLAB_TAG=14.5.2-ee.0
+
+GITLAB_CI_IMAGE=localhost/gitlab-ci
+GITLAB_CI_TAG=latest
@@ -0,0 +1,8 @@
+ARG GITLAB_IMAGE=get_this_from_env
+ARG GITLAB_TAG=get_this_from_env
+FROM ${GITLAB_IMAGE}:${GITLAB_TAG}
+
+COPY create_license.rb /
+RUN ruby /create_license.rb
+
+CMD ["/assets/wrapper"]
@@ -0,0 +1,17 @@
+#!/bin/bash
+
+set -u
+set -e
+
+TOP_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+
+cd "${TOP_DIR}" || exit 1
+
+source .env
+
+docker build \
+    -t "${GITLAB_CI_IMAGE}:${GITLAB_CI_TAG}" \
+    --build-arg GITLAB_IMAGE="${GITLAB_IMAGE}" \
+    --build-arg GITLAB_TAG="${GITLAB_TAG}" \
+    --no-cache \
+    "${TOP_DIR}"
@@ -0,0 +1,51 @@
+# NOTE: As of 2021-12-26 the GitLab Enterprise Edition License has the following
+# section:
+#   Notwithstanding the foregoing, you may copy and modify the Software for development
+#   and testing purposes, without requiring a subscription.
+#
+# https://gitlab.com/gitlab-org/gitlab/-/blob/master/ee/LICENSE
+#
+# This code is strictly intended for use in the testing framework of python-gitlab
+
+# Code inspired by MIT licensed code at: https://github.com/CONIGUERO/gitlab-license.git
+
+require 'openssl'
+require 'gitlab/license'
+
+# Generate a 2048 bit key pair.
+license_encryption_key = OpenSSL::PKey::RSA.generate(2048)
+
+# Save the private key
+File.open("/.license_encryption_key", "w") { |f| f.write(license_encryption_key.to_pem) }
+# Save the public key
+public_key = license_encryption_key.public_key
+File.open("/.license_encryption_key.pub", "w") { |f| f.write(public_key.to_pem) }
+File.open("/opt/gitlab/embedded/service/gitlab-rails/.license_encryption_key.pub", "w") { |f| f.write(public_key.to_pem) }
+
+Gitlab::License.encryption_key = license_encryption_key
+
+# Build a new license.
+license = Gitlab::License.new
+
+license.licensee = {
+  "Name"    => "python-gitlab-ci",
+  "Company" => "python-gitlab-ci",
+  "Email"   => "python-gitlab-ci@example.com",
+}
+
+# The date the license starts.
F438
 
+license.starts_at         = Date.today
+# Want to make sure we get at least 1 day of usage. Do two days after because if CI
+# started at 23:59 we could be expired in one minute if we only did one next_day.
+license.expires_at         = Date.today.next_day.next_day
+
+# Use 'ultimate' plan so that we can test all features in the CI
+license.restrictions = {
+  :plan => "ultimate", 
+  :id   => rand(1000..99999999)
+}
+
+# Export the license, which encrypts and encodes it.
+data = license.export
+
+File.open("/python-gitlab-ci.gitlab-license", 'w') { |file| file.write(data) }
@@ -1,12 +1,12 @@
-version: '3'
+version: '3.5'
 
 networks:
   gitlab-network:
     name: gitlab-network
 
 services:
   gitlab:
-    image: '${GITLAB_IMAGE}:${GITLAB_TAG}'
+    image: '${GITLAB_CI_IMAGE}:${GITLAB_CI_TAG}'
     container_name: 'gitlab-test'
     hostname: 'gitlab.test'
     privileged: true # Just in case https://gitlab.com/gitlab-org/omnibus-gitlab/-/issues/1350
@@ -31,6 +31,7 @@ services:
         gitlab_exporter['enable'] = false
         grafana['enable'] = false
         letsencrypt['enable'] = false
+        gitlab_rails['initial_license_file'] = "/python-gitlab-ci.gitlab-license"
     ports:
       - '8080:80'
       - '2222:22'
 
@@ -95,11 +95,13 @@ exclude_lines =
 [testenv:cli_func_v4]
 deps = -r{toxinidir}/requirements-docker.txt
 commands =
+  {toxinidir}/tests/functional/fixtures/build-dockerfile.sh
   pytest --script-launch-mode=subprocess --cov --cov-report xml tests/functional/cli {posargs}
 
 [testenv:py_func_v4]
 deps = -r{toxinidir}/requirements-docker.txt
 commands =
+  {toxinidir}/tests/functional/fixtures/build-dockerfile.sh
   pytest --cov --cov-report xml tests/functional/api {posargs}
 
 [testenv:smoke]