wip: testing gitlab-ee

JohnVillalovos · JohnVillalovos · commit df76c27bebfd · 2021-12-27T00:39:12.000-08:00
diff --git a/tests/functional/api/test_merge_request_approvals.py b/tests/functional/api/test_merge_request_approvals.py
@@ -0,0 +1,10 @@
+import gitlab
+import gitlab.v4.objects
+
+
+def test_merge_request_approvals(gl: gitlab.Gitlab, project):
+    license = gl.get_license()
+    print(license)
+
+    config = project.approvals.get()
+    assert config is None
diff --git a/tests/functional/conftest.py b/tests/functional/conftest.py
@@ -15,6 +15,12 @@ def fixture_dir(test_dir):
 
 
 def reset_gitlab(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 +140,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)
diff --git a/tests/functional/fixtures/.env b/tests/functional/fixtures/.env
@@ -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
diff --git a/tests/functional/fixtures/Dockerfile b/tests/functional/fixtures/Dockerfile
@@ -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"]
diff --git a/tests/functional/fixtures/build-dockerfile.sh b/tests/functional/fixtures/build-dockerfile.sh
@@ -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}"
diff --git a/tests/functional/fixtures/create_license.rb b/tests/functional/fixtures/create_license.rb
@@ -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: 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. 
+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) }
diff --git a/tests/functional/fixtures/docker-compose.yml b/tests/functional/fixtures/docker-compose.yml
@@ -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'
diff --git a/tox.ini b/tox.ini
@@ -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]
