Merge pull request CatalystCode#38 from CatalystCode/stub-cli

Adding CLI operation stubs and tests

Author: olgaliak

cli/src/cli.py

@@ -0,0 +1,41 @@
+import argparse
+
+from operations import (
+    init,
+    download,
+    upload,
+    abandon,
+    LOWER_LIMIT,
+    UPPER_LIMIT
+)
+
+if __name__ == "__main__":
+
+    # how i want to use the tool:
+    # python3 cli.py init --config /path/to/config.ini
+    # python3 cli.py download --num-images 40
+    # python3 cli.py upload
+    # python3 cli.py abandon
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        'operation',
+        choices=['init', 'download', 'upload', 'abandon']
+    )
+
+    parser.add_argument('-n', '--num-images', type=int)
+
+    parser.add_argument('-c', '--config')
+
+    args = parser.parse_args()
+
+    operation = args.operation
+
+    if operation == 'init':
+        init(args.config)
+    elif operation == 'download':
+        download(args.num_images)
+    elif operation == 'upload':
+        upload()
+    else:
+        abandon()

cli/src/operations.py

@@ -0,0 +1,38 @@
+DEFAULT_NUM_IMAGES = 40
+LOWER_LIMIT = 0
+UPPER_LIMIT = 100
+
+
+class MissingConfigException(Exception):
+    pass
+
+
+class ImageLimitException(Exception):
+    pass
+
+
+def init(config):
+    if (config is None):
+        raise MissingConfigException()
+
+    raise NotImplementedError
+
+
+def download(num_images):
+    images_to_download = num_images
+
+    if num_images is None:
+        images_to_download = DEFAULT_NUM_IMAGES
+
+    if images_to_download <= LOWER_LIMIT or images_to_download > UPPER_LIMIT:
+        raise ImageLimitException()
+
+    return images_to_download
+
+
+def upload():
+    raise NotImplementedError()
+
+
+def abandon():
+    raise NotImplementedError()

cli/src/test_operations.py

@@ -0,0 +1,49 @@
+import unittest
+from operations import (
+    init,
+    download,
+    upload,
+    abandon,
+    MissingConfigException,
+    ImageLimitException,
+    DEFAULT_NUM_IMAGES,
+    LOWER_LIMIT,
+    UPPER_LIMIT
+)
+
+
+class TestCLIOperations(unittest.TestCase):
+    def test_init(self):
+        with self.assertRaises(NotImplementedError):
+            init("fakeconfig")
+
+    def test_init_missing_config(self):
+        with self.assertRaises(MissingConfigException):
+            init(None)
+
+    def test_download_under_limit(self):
+        with self.assertRaises(ImageLimitException):
+            download(LOWER_LIMIT)
+
+    def test_download_over_limit(self):
+        with self.assertRaises(ImageLimitException):
+            download(UPPER_LIMIT + 1)
+
+    def test_download_missing_image_count(self):
+        downloaded_image_count = download(None)
+        self.assertEqual(DEFAULT_NUM_IMAGES, downloaded_image_count)
+
+    def test_download_with_image_count(self):
+        downloaded_image_count = download(10)
+        self.assertEqual(10, downloaded_image_count)
+
+    def test_upload(self):
+        with self.assertRaises(NotImplementedError):
+            upload()
+
+    def test_abandon(self):
+        with self.assertRaises(NotImplementedError):
+            abandon()
+
+if __name__ == '__main__':
+    unittest.main()