Initial creation of tests

Author: vitorfhc

cfaster/__main__.py

@@ -22,7 +22,7 @@ def set_log_levels(level=logging.WARNING):
 
 @group()
 @option('-v', '--verbose', is_flag=True, help='Output INFO level logs.')
-@version_option(__version__, message='v%(version)s')
+@version_option(__version__, message='%(version)s')
 def main_cmd(verbose):
     if verbose:
         set_log_levels(logging.INFO)

tests/__init__.py

@@ -0,0 +1,12 @@
+import re
+from click.testing import CliRunner
+from cfaster.__main__ import main_cmd
+
+runner = CliRunner()
+
+def test_version():
+    result = runner.invoke(main_cmd, ['--version'])
+    reg = r'\d+\.\d+\.\d+'
+    match = re.match(reg, result.output)
+    assert result.exit_code == 0
+    assert match is not None

tests/test_main_cmd.py

@@ -0,0 +1,50 @@
+from click.testing import CliRunner
+from cfaster.__main__ import scrape
+
+runner = CliRunner()
+
+invalid_urls = [
+        'http://domain.com/problem',
+        'https://domain.com/problem',
+        'domain.com',
+        'https://codeforces.com',
+        'https://codeforces.com/contests',
+        'sadaddsg.sda..sad',
+        'www.codeforces.com/sad/dasdas/problem/A',
+        'www.codeforces.com',
+]
+
+valid_urls = [
+    'https://codeforces.com/contest/1172/problem/A',
+    'https://codeforces.com/contest/1172/problem/C1',
+    'https://codeforces.com/contest/1172/problem/E',
+    'https://codeforces.com/contest/1174/problem/B',
+    'https://codeforces.com/contest/1174/problem/D',
+]
+
+def test_no_url():
+    result = runner.invoke(scrape)
+    assert result.exit_code == 2
+
+def test_invalid_urls():
+    for invalid_url in invalid_urls:
+        result = runner.invoke(scrape, invalid_url)
+        assert result.exit_code != 0
+
+def test_verbose_scrape():
+    pass
+
+def test_valid_url():
+    pass
+
+def test_count_inputs():
+    pass
+
+def test_count_outputs():
+    pass
+
+def test_inputs():
+    pass
+
+def test_outputs():
+    pass