t

test_quickcat.py

@@ -198,6 +198,44 @@ def test_cli_load_file_bulk(self):
                 # TODO re-running same file adds images even though they have same URL
                 # bug in mongomock (ignoring unique)???
 
+    def test_cli_categories(self):
+        runner = CliRunner()
+
+        self.assertEqual(Category.objects.count(), 0)
+
+        result = runner.invoke(cli.categories, ['cat A'], input='y\n')
+        self.assertEqual(result.exit_code, 0)
+        self.assertEqual(result.output, '''\
+Existing categories will be removed and these will be added
+cat A
+Continue? [y/N]: y
+Categories recreated
+''')
+        self.assertEqual(Category.objects.count(), 1)
+        self.assertEqual(Category.objects.first().name, 'cat A')
+
+        result = runner.invoke(cli.categories, ['cat B', 'cat C'], input='n\n')
+        self.assertEqual(result.exit_code, 0)
+        self.assertEqual(result.output, '''\
+Existing categories will be removed and these will be added
+cat B, cat C
+Continue? [y/N]: n
+''')
+        self.assertEqual(Category.objects.count(), 1)
+        self.assertEqual(Category.objects.first().name, 'cat A')
+
+        result = runner.invoke(cli.categories, ['cat B', 'cat C'], input='y\n')
+        self.assertEqual(result.exit_code, 0)
+        self.assertEqual(result.output, '''\
+Existing categories will be removed and these will be added
+cat B, cat C
+Continue? [y/N]: y
+Categories recreated
+''')
+        self.assertEqual(Category.objects.count(), 2)
+        self.assertEqual(Category.objects.all()[0].name, 'cat B')
+        self.assertEqual(Category.objects.all()[1].name, 'cat C')
+
 
 if __name__ == '__main__':
     unittest.main()