Update prebuilt.py to filter Packages database.

prebuilt_unittest.py

 #!/usr/bin/python
 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import mox
 import os
 import prebuilt
 import shutil
 import tempfile
@@ skipping 173 matching lines @@
                     'overlay-variant-arm-tegra2-vogue', 'make.conf'),}
     for target in targets:
       self.assertEqual(prebuilt.DetermineMakeConfFile(target), targets[target])
 
   def testDetermineMakeConfGarbage(self):
     """Ensure an exception is raised on bad input."""
     self.assertRaises(prebuilt.UnknownBoardFormat, prebuilt.DetermineMakeConfFile,
                       'asdfasdf')
 
 
+class TestPackagesFileFiltering(unittest.TestCase):
+
+  def setUp(self):
+    self.mox = mox.Mox()
+
+  def tearDown(self):
+    self.mox.UnsetStubs()
+    self.mox.VerifyAll()
+
+  def testFilterAllPackages(self):
+    self.mox.StubOutWithMock(prebuilt, 'ShouldFilterPackage')
+    prebuilt.ShouldFilterPackage("public1").AndReturn(False)
+    prebuilt.ShouldFilterPackage("private").AndReturn(True)
+    prebuilt.ShouldFilterPackage("public2").AndReturn(False)
+    full_packages_file = [
+      "foo: bar\n", "\n",
+      "CPV: public1\n", "foo: bar1\n", "\n",
+      "CPV: private\n", "foo: bar2\n", "\n",
+      "CPV: public2\n", "foo: bar3\n", "\n",
+    ]
+    private_packages_file = [
+      "foo: bar\n", "\n",
+      "CPV: public1\n", "foo: bar1\n", "\n",
+      "CPV: public2\n", "foo: bar3\n", "\n",
+    ]
+    self.mox.ReplayAll()
+    temp_packages_file = tempfile.NamedTemporaryFile()
+    temp_packages_file.write("".join(full_packages_file))
+    temp_packages_file.flush()
+    new_packages_file = prebuilt.FilterPackagesFile(temp_packages_file.name)
+    new_contents = open(new_packages_file.name).read()
+    self.assertEqual("".join(private_packages_file), new_contents)
+    self.assertEqual("".join(private_packages_file), new_packages_file.read())
+    new_packages_file.close()
+
+
 if __name__ == '__main__':
   unittest.main()