Update test for adding and removing a package from a CustomImageRecipe
so that it uses the ProjectPackage and correct fields for the packages
included. Change the test for error condition to use an invalid package
id as ManyToMany remove() on package that isn't in the relationship does
not throw an error.

Signed-off-by: Michael Wood <[email protected]>
---
 bitbake/lib/toaster/toastergui/tests.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/bitbake/lib/toaster/toastergui/tests.py 
b/bitbake/lib/toaster/toastergui/tests.py
index 3683b51..d0e4676 100644
--- a/bitbake/lib/toaster/toastergui/tests.py
+++ b/bitbake/lib/toaster/toastergui/tests.py
@@ -30,7 +30,7 @@ from orm.models import Project, Release, BitbakeVersion, 
Package, LogMessage
 from orm.models import ReleaseLayerSourcePriority, LayerSource, Layer, Build
 from orm.models import Layer_Version, Recipe, Machine, ProjectLayer, Target
 from orm.models import CustomImageRecipe, ProjectVariable
-from orm.models import Branch
+from orm.models import Branch, ProjectPackage
 
 import toastermain
 import inspect
@@ -141,6 +141,11 @@ class ViewTests(TestCase):
 
         Package.objects.create(name='zpkg1', recipe=self.recipe1, build=build)
 
+        self.project_package = ProjectPackage.objects.create(
+            name="ppkg1",
+            recipe=self.recipe1,
+            project=self.project)
+
         # recipe with project for testing AvailableRecipe table
         self.recipe2 = Recipe.objects.create(layer_source=layersrc,
                                              name="fancy-recipe",
@@ -377,24 +382,27 @@ class ViewTests(TestCase):
         # add self.package to recipe
         response = self.client.put(reverse('xhr_customrecipe_packages',
                                            args=(self.customr.id,
-                                                 self.package.id)))
+                                                 self.project_package.id)))
 
         self.assertEqual(response.status_code, 200)
         self.assertEqual(json.loads(response.content),
-                         {"error": "ok",
-                          "dependencies_needed": []})
-        self.assertEqual(self.customr.package_set.first().name,
-                         self.package.name)
+                         {"error": "ok"})
+        self.assertEqual(self.customr.appends_set.first().name,
+                         self.project_package.name)
         # delete it
         del_url = reverse('xhr_customrecipe_packages',
                           args=(self.customr.id,
-                                self.customr.package_set.first().id))
+                                self.customr.appends_set.first().id))
 
         response = self.client.delete(del_url)
         self.assertEqual(response.status_code, 200)
         self.assertEqual(json.loads(response.content), {"error": "ok"})
-        self.assertFalse(self.customr.package_set.all())
-        # delete it again to test error condition
+        self.assertFalse(self.customr.includes_set.all())
+        # delete invalid package to test error condition
+        del_url = reverse('xhr_customrecipe_packages',
+                          args=(self.customr.id,
+                                99999))
+
         response = self.client.delete(del_url)
         self.assertEqual(response.status_code, 200)
         self.assertNotEqual(json.loads(response.content)["error"], "ok")
-- 
2.1.4

-- 
_______________________________________________
toaster mailing list
[email protected]
https://lists.yoctoproject.org/listinfo/toaster

Reply via email to