Add 0.0.1 logic to new cros_mark.

cros_mark_as_stable_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.
 
 """Unit tests for cros_mark_as_stable.py."""
 
 
 import mox
@@ skipping 107 matching lines @@
     # Test with ebuild with revision number.
     no_rev, no_version, revision = cros_mark_as_stable._EBuild._ParseEBuildPath(
         '/path/test_package-0.0.1-r1.ebuild')
     self.assertEquals(no_rev, '/path/test_package-0.0.1')
     self.assertEquals(no_version, '/path/test_package')
     self.assertEquals(revision, 1)
 
   def testParseEBuildPathNoRevisionNumber(self):
     # Test with ebuild without revision number.
     no_rev, no_version, revision = cros_mark_as_stable._EBuild._ParseEBuildPath(
-        '/path/test_package-0.0.1.ebuild')
+        '/path/test_package-9999.ebuild')
     self.assertEquals(no_rev, '/path/test_package-0.0.1')
     self.assertEquals(no_version, '/path/test_package')
     self.assertEquals(revision, 0)
 
 
 class EBuildStableMarkerTest(mox.MoxTestBase):
 
   def setUp(self):
     mox.MoxTestBase.setUp(self)
     self.mox.StubOutWithMock(cros_mark_as_stable, '_SimpleRunCommand')
     self.mox.StubOutWithMock(cros_mark_as_stable, 'RunCommand')
     self.mox.StubOutWithMock(os, 'unlink')
     self.m_ebuild = self.mox.CreateMock(cros_mark_as_stable._EBuild)
+    self.m_ebuild.is_stable = True
     self.m_ebuild.package = 'test_package'
     self.m_ebuild.current_revision = 1
     self.m_ebuild.ebuild_path_no_revision = '/path/test_package-0.0.1'
     self.m_ebuild.ebuild_path_no_version = '/path/test_package'
     self.m_ebuild.ebuild_path = '/path/test_package-0.0.1-r1.ebuild'
     self.revved_ebuild_path = '/path/test_package-0.0.1-r2.ebuild'
 
   def testRevEBuild(self):
     self.mox.StubOutWithMock(cros_mark_as_stable.fileinput, 'input')
     self.mox.StubOutWithMock(cros_mark_as_stable.os.path, 'exists')
@@ skipping 122 matching lines @@
 
   def setUp(self):
     mox.MoxTestBase.setUp(self)
     self.mox.StubOutWithMock(cros_mark_as_stable.os, 'walk')
     self.mox.StubOutWithMock(cros_mark_as_stable, 'RunCommand')
     self.package = 'chromeos-base/test_package'
     self.root = '/overlay/chromeos-base/test_package'
     self.package_path = self.root + '/test_package-0.0.1.ebuild'
     paths = [[self.root, [], []]]
     cros_mark_as_stable.os.walk("/overlay").AndReturn(paths)
-    self.mox.StubOutWithMock(cros_mark_as_stable, '_FindStableEBuilds')
+    self.mox.StubOutWithMock(cros_mark_as_stable, '_FindUprevCandidates')
 
 
   def testWantedPackage(self):
     overlays = {"/overlay": []}
     package = _Package(self.package)
-    cros_mark_as_stable._FindStableEBuilds([]).AndReturn(package)
+    cros_mark_as_stable._FindUprevCandidates([]).AndReturn(package)
     self.mox.ReplayAll()
     cros_mark_as_stable._BuildEBuildDictionary(overlays, False, [self.package])
     self.mox.VerifyAll()
     self.assertEquals(len(overlays), 1)
     self.assertEquals(overlays["/overlay"], [package])
 
   def testUnwantedPackage(self):
     overlays = {"/overlay": []}
     package = _Package(self.package)
-    cros_mark_as_stable._FindStableEBuilds([]).AndReturn(package)
+    cros_mark_as_stable._FindUprevCandidates([]).AndReturn(package)
     self.mox.ReplayAll()
     cros_mark_as_stable._BuildEBuildDictionary(overlays, False, [])
     self.assertEquals(len(overlays), 1)
     self.assertEquals(overlays["/overlay"], [])
     self.mox.VerifyAll()
 
 
 if __name__ == '__main__':
   unittest.main()