Merge pull request #80 from GrappigPanda/unit_tests

Resolves #79, #72, #67, #52

Author: Ian C

.gitignore

@@ -1,4 +1,5 @@
 dist*
 build*
 .idea*
-*.pyc
+*.pyc
+*.swp

.travis.yml

@@ -0,0 +1,11 @@
+language: python
+python:
+  - "2.6"
+  - "2.7"
+
+before_script:
+  - virtualenv ../venv
+  - source ../venv/bin/activate
+  - pip install -r requirements.txt
+script:
+  - nose2

pyg/Pygemony.py

@@ -74,8 +74,11 @@ def file_handler(self):
             file_type = detect_mimetype(file_)
             # We're looking for startswith('text/'). Mimetype returns
             # None if it can't determine file type. Remove if either is True
-            if file_type[0].startswith("application") or file_type[0] is None:
-                files_found.remove(file_)
+            try:
+                if file_type[0].startswith("application") or file_type[0] is None:
+                    files_found.remove(file_)
+            except (AttributeError, IndexError) as e:
+                print "Failed to open file {} with error of {}".format(file_, e)
 
         for file_ in files_found:
             try:
@@ -94,6 +97,7 @@ def lookup_language(self):
         lang_map = {'cpp': LanguageCPP,
                     'python': LanguagePython,
                     'javascript': LanguageJavascript,
-                    'c': LanguageC}
+                    'c': LanguageC,
+                    'go': LanguageGo}
         langs = [i for i in self.github.get_languages()]
         return [lang_map[str(langs[0][0]).lower()]()]

pyg/languages.py

@@ -29,3 +29,11 @@ def __init__(self):
         self.multi_comment = ['/*', '*/']
         self.file_exts = ['*.js', '.node']
         self.ignore = ['node_modules']
+
+class LanguageGo:
+    def __init__(self):
+        self.single_comment = '//'
+        self.multi_comment = ['/*', '*/']
+        self.file_exts = ['*.go']
+        self.ignore = []
+

requirements.txt

@@ -1,4 +1,6 @@
 github3.py==0.9.5
 requests==2.9.1
+mock==1.3.0
 uritemplate.py==0.3.0
 wheel==0.24.0
+nose2==0.5.0

setup.py

@@ -7,6 +7,7 @@ def read(fname):
 
 setup(
     name='pygemony',
+    install_requires=read('requirements.txt'),
     version='0.0.1a',
     description=('Parse TODO from github directory'),
     license="MIT",

unit_tests/__init__.py

@@ -0,0 +1 @@
+

unit_tests/test_pygemony.py

@@ -0,0 +1,11 @@
+from mock import patch
+
+import unittest
+
+from pyg.Pygemony import Pygemony
+
+class PygemonyTestCase(unittest.TestCase):
+
+    def test_hashing_pass(self):
+        md5 = Pygemony.hash_todo("# TODO(ian): Testing 123", 5, 'test.py')
+        self.assertEqual(md5, "3ce18a4c1ef5500be307fd49bae6a37e")