Fixed wrong imports

sdelements · Feb 14, 2020 · 13bd181 · 13bd181
1 parent aa040a8
commit 13bd181
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/tests/fake/test_fake_sync.py b/tests/fake/test_fake_sync.py
@@ -1,4 +1,9 @@
-from pubtools.pulplib import FakeController, YumRepository, SyncOptions, PulpException
+from pubtools.pulplib import (
+    FakeController,
+    YumRepository,
+    YumSyncOptions,
+    PulpException,
+)
 
 
 def test_can_sync():
@@ -12,7 +17,7 @@ def test_can_sync():
     repo1 = client.get_repository("repo1")
 
     # Call to sync should succeed
-    sync_f = repo1.sync(SyncOptions(feed="mock://feed/"))
+    sync_f = repo1.sync(YumSyncOptions(feed="mock://feed/"))
 
     # The future should resolve successfully
     tasks = sync_f.result()
@@ -44,6 +49,6 @@ def test_sync_absent_raises():
     assert repo_copy1.delete().result()
 
     # ...then sync through the other handle becomes impossible
-    exception = repo_copy2.sync(SyncOptions(feed="mock://feed/")).exception()
+    exception = repo_copy2.sync(YumSyncOptions(feed="mock://feed/")).exception()
     assert isinstance(exception, PulpException)
     assert "repo1 not found" in str(exception)
diff --git a/tests/repository/test_sync.py b/tests/repository/test_sync.py
@@ -7,7 +7,7 @@
     Task,
     Distributor,
     DetachedException,
-    SyncOptions,
+    YumSyncOptions,
     TaskFailedException,
 )
 
@@ -60,7 +60,7 @@ def test_sync_with_options(
     repo = YumRepository(id="some-repo")
     repo.__dict__["_client"] = client
 
-    options = SyncOptions(ssl_validation=False, feed="mock://example.com/")
+    options = YumSyncOptions(ssl_validation=False, feed="mock://example.com/")
 
     # It should have succeeded, with the tasks as retrieved from Pulp
     assert repo.sync(options).result() == [