Add test cases for remove all and permanent modes

Author: bridadan

test/detect/platform_database.py

@@ -123,6 +123,60 @@ def test_remove(self):
         self.assertEqual(self.pdb.remove('4753', permanent=False), 'Test_Platform')
         self.assertEqual(self.pdb.get('4753', None), None)
 
+    def test_remove_all(self):
+        """Test that multiple entries can be removed at once
+        """
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.assertEqual(self.pdb.get('4754', None), None)
+        self.pdb.add('4753', 'Test_Platform1', permanent=False)
+        self.pdb.add('4754', 'Test_Platform2', permanent=False)
+        self.assertEqual(self.pdb.get('4753', None), 'Test_Platform1')
+        self.assertEqual(self.pdb.get('4754', None), 'Test_Platform2')
+        self.pdb.remove('*', permanent=False)
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.assertEqual(self.pdb.get('4754', None), None)
+
+    def test_remove_permanent(self):
+        """Test that once something is removed permanently it no longer shows up
+        when queried
+        """
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.pdb.add('4753', 'Test_Platform', permanent=True)
+        self.assertEqual(self.pdb.get('4753', None), 'Test_Platform')
+
+        # Recreate platform database to simulate rerunning mbedls
+        self.pdb = PlatformDatabase([self.base_db_path])
+        self.assertEqual(self.pdb.get('4753', None), 'Test_Platform')
+        self.assertEqual(self.pdb.remove('4753', permanent=True), 'Test_Platform')
+        self.assertEqual(self.pdb.get('4753', None), None)
+
+        # Recreate platform database to simulate rerunning mbedls
+        self.pdb = PlatformDatabase([self.base_db_path])
+        self.assertEqual(self.pdb.get('4753', None), None)
+
+    def test_remove_all_permanent(self):
+        """Test that multiple entries can be removed permanently at once
+        """
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.assertEqual(self.pdb.get('4754', None), None)
+        self.pdb.add('4753', 'Test_Platform1', permanent=True)
+        self.pdb.add('4754', 'Test_Platform2', permanent=True)
+        self.assertEqual(self.pdb.get('4753', None), 'Test_Platform1')
+        self.assertEqual(self.pdb.get('4754', None), 'Test_Platform2')
+
+        # Recreate platform database to simulate rerunning mbedls
+        self.pdb = PlatformDatabase([self.base_db_path])
+        self.assertEqual(self.pdb.get('4753', None), 'Test_Platform1')
+        self.assertEqual(self.pdb.get('4754', None), 'Test_Platform2')
+        self.pdb.remove('*', permanent=True)
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.assertEqual(self.pdb.get('4754', None), None)
+
+        # Recreate platform database to simulate rerunning mbedls
+        self.pdb = PlatformDatabase([self.base_db_path])
+        self.assertEqual(self.pdb.get('4753', None), None)
+        self.assertEqual(self.pdb.get('4754', None), None)
+
     def test_bogus_add(self):
         """Test that add requires properly formatted platform ids
         """