Fix setup.py to run yarn with all repo formats by default

1 files changed, 11 insertions, 6 deletions

diff --git a/setup.py b/setup.py
index 351a6e0e..c1336f70 100755
--- a/setup.py
+++ b/setup.py
@@ -166,12 +166,17 @@ class Check(Command):
             os.remove('.coverage')
 
         if self.yarns and got_yarn:
-            runcmd(
-                ['yarn', '-s', 'yarns/obnam.sh'] +
-                ['--env',
-                 'REPOSITORY_FORMAT=' +
-                 os.environ.get('REPOSITORY_FORMAT', '6')] +
-                glob.glob('yarns/*.yarn'))
+            if 'REPOSITORY_FORMAT' in os.environ:
+                formats = [os.environ['REPOSITORY_FORMAT']]
+            else:
+                formats = ['6', 'simple']
+
+            for format in formats:
+                print 'run yarn for repository format %s' % format
+                runcmd(
+                    ['yarn', '-s', 'yarns/obnam.sh'] +
+                    ['--env', 'REPOSITORY_FORMAT=' + format] +
+                    glob.glob('yarns/*.yarn'))
 
         num_clients = '2'
         num_generations = '16'