diff --git a/tests/unit/plugins/lookup/test_apt_repositories.py b/tests/unit/plugins/lookup/test_apt_repositories.py
index 296902a24..23ca9c929 100644
--- a/tests/unit/plugins/lookup/test_apt_repositories.py
+++ b/tests/unit/plugins/lookup/test_apt_repositories.py
@@ -102,32 +102,48 @@ def test_invalid_state(self):
 
     def test_wanttype_name(self):
         self.assertListEqual([
+            {'name': 'corge', 'state': 'absent'},
             {'name': 'foo', 'uris': 'foo', 'suites': 'foo/', 'state': 'present'},
             {'name': 'bar', 'uris': 'bar', 'suites': 'bar/', 'state': 'present'},
-            {'name': 'baz', 'uris': 'baz', 'suites': 'baz/', 'state': 'absent'},
+            {'name': 'baz', 'uris': 'baz', 'suites': 'baz/', 'state': 'present', 'legacy_file': 'baz.list'},
+            {'name': 'qux', 'uris': 'qux', 'suites': 'qux/', 'state': 'present', 'legacy_file': 'qux.list'},
         ], self.lookup.run([
             [
                 {'name': 'foo', 'uris': 'foo', 'suites': 'foo/'},
-                {'name': 'bar', 'uris': 'bar', 'suites': 'bar/', 'state': 'present'},
-                {'name': 'baz', 'uris': 'baz', 'suites': 'baz/', 'state': 'absent'},
+                {'name': 'bar', 'uris': 'bar', 'suites': 'bar/'},
+                {'name': 'baz', 'uris': 'baz', 'suites': 'baz/', 'legacy_file': 'baz.list'},
+                {'name': 'qux', 'uris': 'qux', 'suites': 'qux/', 'legacy_file': 'qux.list'},
             ],
             {},
             {},
             [],
-            [],
-            "",
+            [
+                {'path': '/quux.list'},
+                {'path': '/corge.sources'},
+            ],
+            "/etc/file",
         ], {}, **{'wanttype': 'name'}))
 
     def test_wanttype_file(self):
         self.assertListEqual([
-            {'file': '/etc/file/foo.list', 'state': 'absent'},
+            {'file': '/quux.list', 'state': 'absent'},
+            {'file': '/etc/file/baz.list', 'state': 'absent'},
+            {'file': '/etc/file/qux.list', 'state': 'absent'},
         ], self.lookup.run([
-            [{'name': 'foo', 'uris': 'foo', 'suites': 'foo/', 'legacy_file': '/etc/file/foo.list'},],
+            [
+                {'name': 'foo', 'uris': 'foo', 'suites': 'foo/'},
+                {'name': 'bar', 'uris': 'bar', 'suites': 'bar/'},
+                {'name': 'baz', 'uris': 'baz', 'suites': 'baz/', 'legacy_file': 'baz.list'},
+                {'name': 'qux', 'uris': 'qux', 'suites': 'qux/', 'legacy_file': 'qux.list'},
+            ],
             {},
             {},
             [],
-            [],
-            "",
+            [
+                {'path': '/quux.list'},
+                {'path': '/corge.sources'},
+            ],
+            "/etc/file",
         ], {}, **{'wanttype': 'file'}))
 
     def test_invalid_wanttype(self):