Add `get_spec` to `PackageOverrides`

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/test_config/test_assets.py b/tests/test_config/test_assets.py
index 5c535ec6c..ea2fff741 100644
--- a/tests/test_config/test_assets.py
+++ b/tests/test_config/test_assets.py
@@ -638,6 +638,28 @@ class TestPackageOverrides(unittest.TestCase):
         po.overrides = overrides
         self.assertEqual(list(po.filtered_sources('whatever')), ['foo'])
 
+    def test_get_spec(self):
+        source = DummyAssetSource(spec='test:foo.pt')
+        overrides = [DummyOverride(None), DummyOverride((source, ''))]
+        package = DummyPackage('package')
+        po = self._makeOne(package)
+        po.overrides = overrides
+        result = po.get_spec('whatever')
+        self.assertEqual(result, 'test:foo.pt')
+        self.assertEqual(source.resource_name, '')
+
+    def test_get_spec_file_doesnt_exist(self):
+        source = DummyAssetSource(spec=None)
+        overrides = [
+            DummyOverride(None),
+            DummyOverride((source, 'wont_exist')),
+        ]
+        package = DummyPackage('package')
+        po = self._makeOne(package)
+        po.overrides = overrides
+        self.assertEqual(po.get_spec('whatever'), None)
+        self.assertEqual(source.resource_name, 'wont_exist')
+
     def test_get_filename(self):
         source = DummyAssetSource(filename='foo.pt')
         overrides = [DummyOverride(None), DummyOverride((source, ''))]
@@ -1053,6 +1075,10 @@ class DummyAssetSource:
     def __init__(self, **kw):
         self.kw = kw
 
+    def get_spec(self, resource_name):
+        self.resource_name = resource_name
+        return self.kw['spec']
+
     def get_filename(self, resource_name):
         self.resource_name = resource_name
         return self.kw['filename']