From 6787f089de1c829509f7d9dfd59295aae71e5ff6 Mon Sep 17 00:00:00 2001
From: Pavlina Bortlova <pbortlov@redhat.com>
Date: Fri, 9 Oct 2020 12:59:36 +0200
Subject: [PATCH] Fix imports from iiblib

---
 pubtools/iib/utils.py | 8 +++-----
 requirements.txt      | 2 +-
 tests/test_iib_ops.py | 6 +++---
 tests/utils.py        | 3 ---
 4 files changed, 7 insertions(+), 12 deletions(-)

diff --git a/pubtools/iib/utils.py b/pubtools/iib/utils.py
index cf634a2..d96fd0b 100644
--- a/pubtools/iib/utils.py
+++ b/pubtools/iib/utils.py
@@ -4,14 +4,12 @@
 import sys
 import pkg_resources
 
-from iiblib import iibclient
+from iiblib import iib_client, iib_authentication
 from pubtools import pulplib
 
-from iiblib.iibclient import IIBBuildDetailsModel
-
 
 def setup_iib_client(parsed_args):
-    iib_auth = iibclient.IIBKrbAuth(
+    iib_auth = iib_authentication.IIBKrbAuth(
         parsed_args.iib_krb_principal,
         parsed_args.iib_server,
         ktfile=parsed_args.iib_krb_ktfile,
@@ -21,7 +19,7 @@ def setup_iib_client(parsed_args):
     }
     if parsed_args.iib_insecure:
         kwargs["ssl_verify"] = False
-    iibc = iibclient.IIBClient(parsed_args.iib_server, **kwargs)
+    iibc = iib_client.IIBClient(parsed_args.iib_server, **kwargs)
     return iibc
 
 
diff --git a/requirements.txt b/requirements.txt
index d9fcb57..bf3eba2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,5 +3,5 @@ setuptools
 pyrsistent<=0.16.0; python_version == '2.7'
 pushcollector
 pubtools-pulp
-iiblib>=0.9.0
+iiblib>=0.13.0
 pubtools-pulplib
diff --git a/tests/test_iib_ops.py b/tests/test_iib_ops.py
index d830a81..3a4e8f9 100644
--- a/tests/test_iib_ops.py
+++ b/tests/test_iib_ops.py
@@ -11,7 +11,7 @@
 import pushcollector
 
 
-from iiblib.iibclient import IIBBuildDetailsModel
+from iiblib.iib_build_details_model import IIBBuildDetailsModel
 from pubtools.pulplib import ContainerImageRepository
 from more_executors.futures import f_return
 
@@ -65,7 +65,7 @@ def setup_entry_point_py(entry_tuple, environ_vars):
 
 @pytest.fixture
 def fixture_iib_client():
-    with mock.patch("iiblib.iibclient.IIBClient") as iibc_patched:
+    with mock.patch("iiblib.iib_client.IIBClient") as iibc_patched:
         iibc_patched.return_value.add_bundles.side_effect = (
             lambda *args, **kwargs: IIBBuildDetailsModel.from_dict(
                 fake_tm.setup_task(*args, **kwargs)
@@ -97,7 +97,7 @@ def fixture_pushcollector():
 
 @pytest.fixture
 def fixture_iib_krb_auth():
-    with mock.patch("iiblib.iibclient.IIBKrbAuth") as iib_krbauth_patched:
+    with mock.patch("iiblib.iib_authentication.IIBKrbAuth") as iib_krbauth_patched:
         iib_krbauth_patched.return_value = mock.MagicMock(name="MockedIIBKrbAuth")
         yield iib_krbauth_patched
 
diff --git a/tests/utils.py b/tests/utils.py
index 12ea7cf..59af1e7 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -1,6 +1,3 @@
-from iiblib.iibclient import IIBBuildDetailsModel
-
-
 class FakeTaskManager(object):
     def __init__(self):
         self.tasks = {}