added initial test for creating courses

oldclesleycode · oldclesleycode · commit 063606342ac4 · 2018-12-27T19:13:49.000-05:00
diff --git a/classroom/snippets/base_test.py b/classroom/snippets/base_test.py
@@ -0,0 +1,53 @@
+# Copyright 2018 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import print_function
+import sys
+import unittest
+import httplib2
+from oauth2client.client import GoogleCredentials
+from googleapiclient import errors
+from googleapiclient.discovery import build
+from oauth2client import file, client, tools
+
+SCOPES = 'https://www.googleapis.com/auth/classroom.courses'
+
+
+class BaseTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.credentials = cls.create_credentials()
+        http = cls.credentials.authorize(httplib2.Http())
+        cls.credentials.refresh(http)
+        cls.service = build('classroom', 'v1', http=http)
+        cls.stdout = sys.stdout
+        sys.stdout = None
+
+    @classmethod
+    def tearDownClass(cls):
+        # Restore STDOUT.
+        sys.stdout = cls.stdout
+
+    @classmethod
+    def create_credentials(cls):
+        store = file.Storage('token.json')
+        credentials = None
+        if not credentials or credentials.invalid:
+            flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
+            credentials = tools.run_flow(flow, store)
+        return credentials
+
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/classroom/snippets/classroom_snippets.py b/classroom/snippets/classroom_snippets.py
@@ -40,6 +40,7 @@ def create_course(self):
         course = service.courses().create(body=course).execute()
         print('Course created:', course.get('name'), course.get('id'))
         # [END classroom_create_course]
+        return course
 
     def get_course(self, course_id):
         """ Retrieves a classroom course by its id. """
@@ -134,7 +135,7 @@ def add_alias_existing(self, course_id):
         }
         try:
             courseAlias = service.courses().aliases().create(
-                courseId=courseId,
+                courseId=course_id,
                 body=courseAlias).execute()
         except errors.HttpError:
             print('Alias Creation Failed')
diff --git a/classroom/snippets/test_snippets.py b/classroom/snippets/test_snippets.py
@@ -0,0 +1,36 @@
+
+# Copyright 2018 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+from pprint import pformat
+from base_test import BaseTest
+from classroom_snippets import ClassroomSnippets
+
+
+class SnippetsTest(BaseTest):
+
+    @classmethod
+    def setUpClass(cls):
+        super(SnippetsTest, cls).setUpClass()
+        cls.snippets = ClassroomSnippets(cls.service)
+
+    def test_create_course(self):
+        course = self.snippets.create_course()
+        self.assertIsNotNone(course)
+        # delete course as part of cleanup
+
+
+if __name__ == '__main__':
+    unittest.main()
