diff --git a/bitbake/lib/bb/tests/data.py b/bitbake/lib/bb/tests/data.py index cbc7c1ecd4..a895f6a58e 100644 --- a/bitbake/lib/bb/tests/data.py +++ b/bitbake/lib/bb/tests/data.py @@ -450,17 +450,64 @@ class TestFlags(unittest.TestCase): self.d = bb.data.init() self.d.setVar("foo", "value of foo") self.d.setVarFlag("foo", "flag1", "value of flag1") + self.d.setVarFlag("foo", "_defaultval_flag_flag1", "default of flag1") self.d.setVarFlag("foo", "flag2", "value of flag2") + self.d.setVarFlag("foo", "_defaultval_flag_flag2", "default of flag2") + self.d.setVarFlag("foo", "flag3", "value of flag3") + self.d.setVarFlag("foo", "_defaultval_flag_flagnovalue", "default of flagnovalue") def test_setflag(self): self.assertEqual(self.d.getVarFlag("foo", "flag1", False), "value of flag1") self.assertEqual(self.d.getVarFlag("foo", "flag2", False), "value of flag2") + self.assertDictEqual( + self.d.getVarFlags("foo"), + { + "flag1": "value of flag1", + "flag2": "value of flag2", + "flag3": "value of flag3", + "flagnovalue": "default of flagnovalue", + } + ) + self.assertDictEqual( + self.d.getVarFlags("foo", internalflags=True), + { + "_content": "value of foo", + "flag1": "value of flag1", + "flag2": "value of flag2", + "flag3": "value of flag3", + "_defaultval_flag_flag1": "default of flag1", + "_defaultval_flag_flag2": "default of flag2", + "_defaultval_flag_flagnovalue": "default of flagnovalue", + } + ) def test_delflag(self): self.d.delVarFlag("foo", "flag2") + self.d.delVarFlag("foo", "flag3") self.assertEqual(self.d.getVarFlag("foo", "flag1", False), "value of flag1") self.assertEqual(self.d.getVarFlag("foo", "flag2", False), None) + self.assertDictEqual( + self.d.getVarFlags("foo"), + { + "flag1": "value of flag1", + "flagnovalue": "default of flagnovalue", + } + ) + self.assertDictEqual( + self.d.getVarFlags("foo", internalflags=True), + { + "_content": "value of foo", + "flag1": "value of flag1", + "_defaultval_flag_flag1": "default of flag1", + "_defaultval_flag_flagnovalue": "default of flagnovalue", + } + ) + def test_delvar(self): + self.d.delVar("foo") + self.assertEqual(self.d.getVarFlag("foo", "flag1", False), None) + self.assertEqual(self.d.getVarFlag("foo", "flag2", False), None) + self.assertEqual(self.d.getVarFlags("foo", internalflags=True), None) class Contains(unittest.TestCase): def setUp(self):