diff options
-rw-r--r-- | bitbake/lib/toaster/orm/tests.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/bitbake/lib/toaster/orm/tests.py b/bitbake/lib/toaster/orm/tests.py index c30f99a3af..719266e6d6 100644 --- a/bitbake/lib/toaster/orm/tests.py +++ b/bitbake/lib/toaster/orm/tests.py | |||
@@ -23,7 +23,7 @@ | |||
23 | 23 | ||
24 | from django.test import TestCase, TransactionTestCase | 24 | from django.test import TestCase, TransactionTestCase |
25 | from orm.models import LocalLayerSource, LayerIndexLayerSource, ImportedLayerSource, LayerSource | 25 | from orm.models import LocalLayerSource, LayerIndexLayerSource, ImportedLayerSource, LayerSource |
26 | from orm.models import Branch | 26 | from orm.models import Branch, LayerVersionDependency |
27 | 27 | ||
28 | from orm.models import Project, Layer, Layer_Version, Branch, ProjectLayer | 28 | from orm.models import Project, Layer, Layer_Version, Branch, ProjectLayer |
29 | from orm.models import Release, ReleaseLayerSourcePriority, BitbakeVersion | 29 | from orm.models import Release, ReleaseLayerSourcePriority, BitbakeVersion |
@@ -160,3 +160,21 @@ class LayerVersionEquivalenceTestCase(TestCase): | |||
160 | compat_lv = self.project.get_all_compatible_layer_versions() | 160 | compat_lv = self.project.get_all_compatible_layer_versions() |
161 | self.assertEqual(list(compat_lv), [self.lver, self.lver2]) | 161 | self.assertEqual(list(compat_lv), [self.lver, self.lver2]) |
162 | 162 | ||
163 | def test_layerversion_get_alldeps(self): | ||
164 | """Test Layer_Version.get_alldeps API.""" | ||
165 | lvers = {} | ||
166 | for i in range(10): | ||
167 | name = "layer%d" % i | ||
168 | lvers[name] = Layer_Version.objects.create(layer=Layer.objects.create(name=name), | ||
169 | project=self.project) | ||
170 | if i: | ||
171 | LayerVersionDependency.objects.create(layer_version=lvers["layer%d" % (i - 1)], | ||
172 | depends_on=lvers[name]) | ||
173 | # Check dinamically added deps | ||
174 | self.assertEqual(lvers['layer0'].get_alldeps(self.project.id), | ||
175 | [lvers['layer%d' % n] for n in range(1, i+1)]) | ||
176 | |||
177 | # Check chain of deps created in previous loop | ||
178 | for i in range(10): | ||
179 | self.assertEqual(lvers['layer%d' % i].get_alldeps(self.project.id), | ||
180 | [lvers['layer%d' % n] for n in range(i+1, 10)]) | ||