extend tests to astro api, add test to makefile

Author: natronics

Diff for: Makefile

@@ -1,3 +1,6 @@
+test:
+	python ./run_tests.py
+
 clean:
 	find . -name "*.pyc" -type f -delete
 

Diff for: run_tests.py

@@ -28,7 +28,6 @@ def test_data(self):
         except:
             self.fail("ISS API not a valid JSON responce")
 
-
         # Success message
         self.assertEqual(data['message'], "success", "ISS API Did not return 'sucess' message")
 
@@ -39,3 +38,35 @@ def test_data(self):
         self.assertTrue('iss_position' in data)
         self.assertTrue('latitude' in data['iss_position'])
         self.assertTrue('longitude' in data['iss_position'])
+
+
+class AstrosTest(TestCase):
+    """Test the number of astros API"""
+
+    def setUp(self):
+        self.app = app
+        self.w = TestApp(self.app)
+
+    def test_load_astros(self):
+        r = self.w.get('/astros.json')
+        self.assertFalse(r.flashes)
+        r = self.w.get('/astros/')
+        self.assertFalse(r.flashes)
+        r = self.w.get('/astros/v1')
+        self.assertFalse(r.flashes)
+
+    def test_data(self):
+        r = self.w.get('/astros.json')
+        r.charset = 'utf8'
+        try:
+            data = json.loads(r.text)
+        except:
+            self.fail("ISS API not a valid JSON responce")
+
+        # Success message
+        self.assertEqual(data['message'], "success", "ISS API Did not return 'sucess' message")
+
+        # data exists
+        self.assertTrue('number' in data)
+        self.assertEqual(type(data['number']), int)
+        self.assertTrue('people' in data)