From 98ed3abc72a85ee6cd1f0686bd6fdb449fe7f2f7 Mon Sep 17 00:00:00 2001
From: jobrauer <jobrauer@redhat.com>
Date: Tue, 7 Apr 2020 10:16:11 +0200
Subject: [PATCH] Add test_import_module

---
 tests/integration/example.test.env.yaml |  2 +
 tests/integration/test_import_module.py | 14 ++++++
 tests/integration/utils.py              | 58 ++++++++++++++++++-------
 tox.ini                                 |  1 +
 4 files changed, 60 insertions(+), 15 deletions(-)
 create mode 100644 tests/integration/test_import_module.py

diff --git a/tests/integration/example.test.env.yaml b/tests/integration/example.test.env.yaml
index b994a9e6..a0e334b6 100644
--- a/tests/integration/example.test.env.yaml
+++ b/tests/integration/example.test.env.yaml
@@ -96,3 +96,5 @@ testdata:
     # with koji_tag_with_modules from testmodule's platform (stream 8.1.0.z)
     module: testmodule
     branch: test-reuse-tagged-module
+  import_module:
+    scmurl: "git://src.stg.fedoraproject.org/modules/testmodule?#test-import-module"
diff --git a/tests/integration/test_import_module.py b/tests/integration/test_import_module.py
new file mode 100644
index 00000000..481a56bf
--- /dev/null
+++ b/tests/integration/test_import_module.py
@@ -0,0 +1,14 @@
+
+def test_import_module(scenario, mbs):
+    """
+    Test import module functionality.
+
+    Steps:
+      * Request module import with scmurl provided by test.env.yaml.
+    Checks:
+      * Non-error response.
+    """
+    scmurl = scenario.get("scmurl")
+    assert scmurl, "No SCM URL specified in test.env.yaml file."
+
+    mbs.import_module(scmurl)
diff --git a/tests/integration/utils.py b/tests/integration/utils.py
index 66a0106c..43e1337e 100644
--- a/tests/integration/utils.py
+++ b/tests/integration/utils.py
@@ -1,17 +1,20 @@
 # -*- coding: utf-8 -*-
 # SPDX-License-Identifier: MIT
 
+import os
 import re
 import sys
 import time
 
 from kobo import rpmlib
+import json
 import koji
-import yaml
+import pytest
 import requests
-import tempfile
+import requests_kerberos
 import sh
-import os
+import tempfile
+import yaml
 
 our_sh = sh(_out=sys.stdout, _err=sys.stderr, _tee=True)
 from our_sh import Command, git, pushd  # noqa
@@ -85,10 +88,9 @@ class Koji:
         return self._session.getTag(tag_info)
 
     def get_macro_specfile(self, build):
-        """
-        Download macro src.rpm and extract spec file .
+        """Download macro src.rpm and extract spec file .
 
-        :param build: build object
+        :param Build build: build object
         :return: content of module-build-macros.spec
         :rtype: str
         """
@@ -142,8 +144,7 @@ class Repo:
 
     @property
     def platform(self):
-        """
-        List of platforms in the modulemd file, obtaining values differs on version
+        """List of platforms in the modulemd file, obtaining values differs on version
 
         :return: List of platforms in the modulemd file
         :rtype: list of strings
@@ -334,8 +335,7 @@ class Build:
         return {comp["package"]: comp["task_id"] for comp in self.components()}
 
     def batches(self):
-        """
-        Components of the module build separated in sets according to batches
+        """Components of the module build separated in sets according to batches
 
         :return: list of components according to batches
         :rtype: list of sets
@@ -401,8 +401,7 @@ class Build:
         return False
 
     def get_modulemd(self):
-        """
-        Get module's metadata (from MBS API)
+        """Get module's metadata (from MBS API)
 
         :return: module's metadata
         :rtype: dict
@@ -469,9 +468,9 @@ class MBS:
     def get_builds(self, module, stream, order_desc_by=None):
         """Get list of Builds objects via mbs api.
 
-        :attribute string module: Module name
-        :attribute string stream: Stream name
-        :attribute string order_desc_by: Optional sorting parameter e.g. "version"
+        :param str module: Module name
+        :param str stream: Stream name
+        :param str order_desc_by: Optional sorting parameter e.g. "version"
         :return: list of Build objects
         :rtype: list
         """
@@ -483,6 +482,35 @@ class MBS:
         r.raise_for_status()
         return [Build(self._mbs_api, build["id"]) for build in r.json()["items"]]
 
+    def import_module(self, scmurl):
+        """Import module from SCM URL (modulemd).
+
+        :param str scmurl:
+        :return: requests response
+        :rtype: requests response object
+        """
+        url = f"{self._mbs_api}import-module/"
+        headers = {"Content-Type": "application/json"}
+        data = json.dumps({'scmurl': scmurl})
+
+        # (!) User executing this request must be allowed to do so on the target MBS instance.
+        # MBS server does not support mutual auth, so make it optional (inspired by mbs-cli).
+        auth = requests_kerberos.HTTPKerberosAuth(mutual_authentication=requests_kerberos.OPTIONAL)
+
+        response = requests.post(
+            url,
+            auth=auth,
+            headers=headers,
+            verify=False,
+            data=data
+        )
+        try:
+            response.raise_for_status()
+            return response
+        except requests.exceptions.HTTPError:
+            # response message contains useful information, which requests module omits
+            pytest.fail(response.text)
+
     def get_module_builds(self, **kwargs):
         """
         Query MBS API on module-builds endpoint
diff --git a/tox.ini b/tox.ini
index eaaa8b59..7fce33a7 100644
--- a/tox.ini
+++ b/tox.ini
@@ -80,6 +80,7 @@ deps =
     # rpm is an optional dependency for kobo, rpm-py-installer makes it work
     rpm-py-installer
     sh
+    requests_kerberos
 # Set this to /etc/pki/tls/certs/ca-bundle.crt, for example,
 # if the instance tested has a self-signed certificate.
 passenv = REQUESTS_CA_BUNDLE MBS_TEST_CONFIG MBS_TEST_WORKERS HOME