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)]) | ||
