Add tests for repo-alias expansion

2 files changed, 13 insertions, 0 deletions

diff --git a/morphlib/localrepocache.py b/morphlib/localrepocache.py
index a3e28d57..a5487cd9 100644
--- a/morphlib/localrepocache.py
+++ b/morphlib/localrepocache.py
@@ -185,6 +185,12 @@ class LocalRepoCache(object):
             path = self._cache_name(repourl)
             yield repourl, path
     
+    def pull_url(self, reponame):
+        '''Expand a possibly shortened repo name to a pull url.'''
+        
+    def push_url(self, reponame):
+        '''Expand a possibly shortened repo name to a push url.'''
+    
     def has_repo(self, reponame):
         '''Have we already got a cache of a given repo?'''
         for repourl, path in self._base_iterate(reponame):
diff --git a/morphlib/localrepocache_tests.py b/morphlib/localrepocache_tests.py
index dfa92518..5ee28a99 100644
--- a/morphlib/localrepocache_tests.py
+++ b/morphlib/localrepocache_tests.py
@@ -27,6 +27,7 @@ class LocalRepoCacheTests(unittest.TestCase):
         bundle_base_url = 'http://lorry.example.com/bundles/'
         self.reponame = 'upstream:reponame'
         self.repourl = 'git://example.com/reponame'
+        self.pushurl = 'example.com:reponame.git'
         escaped_url = 'git___example_com_reponame'
         self.bundle_url = '%s%s.bndl' % (bundle_base_url, escaped_url)
         self.cachedir = '/cache/dir'
@@ -77,6 +78,12 @@ class LocalRepoCacheTests(unittest.TestCase):
         self.cache.add(path)
         return True
 
+    def test_expands_shortened_url_correctly_for_pulling(self):
+        self.assertEqual(self.lrc.pull_url(self.reponame), self.repourl)
+
+    def test_expands_shortened_url_correctly_for_pushing(self):
+        self.assertEqual(self.lrc.push_url(self.reponame), self.repourl)
+
     def test_has_not_got_shortened_repo_initially(self):
         self.assertFalse(self.lrc.has_repo(self.reponame))