From 59bd004538b9d3def65f672e10a2d4f657b3c143 Mon Sep 17 00:00:00 2001
From: mutantmonkey <mutantmonkey@mutantmonkey.in>
Date: Thu, 31 May 2012 22:44:03 -0700
Subject: [PATCH] add hokie stalker tests

---
 modules/test/test_hs.py | 44 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 modules/test/test_hs.py

diff --git a/modules/test/test_hs.py b/modules/test/test_hs.py
new file mode 100644
index 0000000..980b954
--- /dev/null
+++ b/modules/test/test_hs.py
@@ -0,0 +1,44 @@
+"""
+test_hs.py - tests for the hokie stalker module
+author: mutantmonkey <mutantmonkey@mutantmonkey.in>
+"""
+
+# add current working directory to path
+import sys
+sys.path.append('.')
+
+import re
+import unittest
+from mock import MagicMock, Mock
+from modules.hs import search, hs
+
+class TestHs(unittest.TestCase):
+    def setUp(self):
+        self.phenny = MagicMock()
+
+    def test_search(self):
+        data = search('john')
+
+        assert len(data) >= 1
+        assert 'uid' in data[0]
+        assert 'cn' in data[0]
+
+    def test_single(self):
+        input = Mock(group=lambda x: 'marchany')
+        hs(self.phenny, input)
+
+        out = self.phenny.reply.call_args[0][0]
+        m = re.match(
+            '^.* - http://search\.vt\.edu/search/person\.html\?person=\d+$',
+            out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_multi(self):
+        input = Mock(group=lambda x: 'john')
+        hs(self.phenny, input)
+
+        out = self.phenny.reply.call_args[0][0]
+        m = re.match(
+            '^Multiple results found; try http://search\.vt\.edu/search/people\.html\?q=.*$',
+            out, flags=re.UNICODE)
+        self.assertTrue(m)