Rev 487: Fix test. in file:///home/jelmer/bzr-svn/0.4/

[Jelmer Vernooij]

------------------------------------------------------------
revno: 487
committer: Jelmer Vernooij <jelmer at samba.org>
branch nick: 0.4
timestamp: Sun 2007-06-17 22:56:08 +0200
message:
  Fix test.
modified:
  revids.py
  tests/test_revids.py
=== modified file 'revids.py'
--- a/revids.py	2007-06-17 20:49:21 +0000
+++ b/revids.py	2007-06-17 20:56:08 +0000
@@ -121,7 +121,7 @@
     def lookup_dist_to_origin(self, revid):
         revno = self.cachedb.execute(
                 "select dist_to_origin from revmap where revid='%s'" % revid).fetchone()
-        if revno is not None:
+        if revno is not None and revno[0] is not None:
             return int(revno[0])
         return None
 

=== modified file 'tests/test_revids.py'
--- a/tests/test_revids.py	2007-06-17 20:49:21 +0000
+++ b/tests/test_revids.py	2007-06-17 20:56:08 +0000
@@ -49,6 +49,15 @@
                                     50)
         self.assertEquals(50,
                 revidmap.lookup_dist_to_origin("bla"))
+
+    def test_lookup_dist_notset(self):
+        revidmap = RevidMap()
+        revidmap.insert_revid("bloe", "someotherpath", 42, 42, "brainslug") 
+        self.assertIs(None,
+                revidmap.lookup_dist_to_origin("bloe"))
+
+    def test_lookup_dist_notfound(self):
+        revidmap = RevidMap()
         self.assertIs(None,
                 revidmap.lookup_dist_to_origin("blabla"))