1 files changed, 45 insertions, 0 deletions
diff --git a/patchwork/tests/test_management.py b/patchwork/tests/test_management.py
index 66c6bad..d24eea9 100644
--- a/patchwork/tests/test_management.py
+++ b/patchwork/tests/test_management.py
@@ -5,6 +5,7 @@
 
 import os
 import sys
+import tempfile
 from io import StringIO
 
 from django.core.management import call_command
@@ -124,3 +125,47 @@ class ParsearchiveTest(TestCase):
 
         self.assertIn('Processed 1 messages -->', out.getvalue())
         self.assertIn('  1 dropped', out.getvalue())
+
+
+class ReplacerelationsTest(TestCase):
+
+    def test_invalid_path(self):
+        out = StringIO()
+        with self.assertRaises(SystemExit) as exc:
+            call_command('replacerelations', 'xyz123random', '-v 0',
+                         stdout=out)
+        self.assertEqual(exc.exception.code, 1)
+
+    def test_valid_relations(self):
+        test_submitter = utils.create_person()
+        utils.create_patches(8, submitter=test_submitter)
+        patch_ids = (models.Patch.objects
+                     .filter(submitter=test_submitter)
+                     .values_list('id', flat=True))
+
+        with tempfile.NamedTemporaryFile(delete=False,
+                                         mode='w+') as f1:
+            for i in range(0, len(patch_ids), 3):
+                # we write out the patch IDs this way so that we can
+                # have a mix of 3-patch and 2-patch lines without special
+                # casing the format string.
+                f1.write('%s\n' % ' '.join(map(str, patch_ids[i:(i + 3)])))
+
+        out = StringIO()
+        call_command('replacerelations', f1.name, stdout=out)
+        self.assertEqual(models.PatchRelation.objects.count(), 3)
+        os.unlink(f1.name)
+
+        patch_ids_with_missing = (
+            list(patch_ids) +
+            [i for i in range(max(patch_ids), max(patch_ids) + 3)]
+        )
+        with tempfile.NamedTemporaryFile(delete=False,
+                                         mode='w+') as f2:
+            for i in range(0, len(patch_ids_with_missing), 3):
+                f2.write('%s\n' % ' '.join(
+                    map(str, patch_ids_with_missing[i:(i + 3)])))
+
+        call_command('replacerelations', f2.name, stdout=out)
+        self.assertEqual(models.PatchRelation.objects.count(), 3)
+        os.unlink(f2.name)