Unit tests for student_view, studio_view for documents and calendar; test for checking document url

google_drive/templates/html/google_docs_edit.html

@@ -33,7 +33,7 @@
       </ul>
     </div>
   </div>
-  <div class="xblock-actions" id="google-docs-actions">
+  <div class="xblock-actions">
     <span class="xblock-editor-error-message"></span>
     <ul>
       <li class="action-item">

google_drive/tests/unit/test_docs.py

@@ -9,7 +9,7 @@ from xblock.runtime import KvsFieldData, DictKeyValueStore
 from google_drive import GoogleDocumentBlock, GoogleCalendarBlock
 
 from nose.tools import (
-    assert_equals
+    assert_equals, assert_in
 )
 
 def make_request(body, method='POST'):
@@ -31,6 +31,33 @@ def make_calendar_block():
     db_model = KvsFieldData(key_store)
     return GoogleCalendarBlock(runtime, db_model, Mock())
 
+def test_document_templates_contents():
+    block = make_document_block()
+
+    student_fragment = block.render('student_view', Mock())
+    assert_in('<div class="google-docs-xblock-wrapper"', student_fragment.content)
+    assert_in('&#39;self.display_name&#39;', student_fragment.content)
+    assert_in('&#39;self.embed_code&#39;', student_fragment.content)
+
+    studio_fragment = block.render('studio_view', Mock())
+    assert_in('<div class="wrapper-comp-settings is-active editor-with-buttons google-edit-wrapper" id="settings-tab">', studio_fragment.content)
+    assert_in('<div class="user-inputs-and-validation">', studio_fragment.content)
+    assert_in('<div class="xblock-inputs editor_content_wrapper">', studio_fragment.content)
+    assert_in('<div class="xblock-actions">', studio_fragment.content)
+
+def test_calendar_templates_contents():
+    block = make_calendar_block()
+
+    student_fragment = block.render('student_view', Mock())
+    assert_in('<div class="google-calendar-xblock-wrapper">', student_fragment.content)
+    assert_in('&#39;iframe&#39;', student_fragment.content)
+
+    studio_fragment = block.render('studio_view', Mock())
+    assert_in('<div class="wrapper-comp-settings is-active editor-with-buttons google-edit-wrapper" id="settings-tab">', studio_fragment.content)
+    assert_in('<div class="user-inputs-and-validation">', studio_fragment.content)
+    assert_in('<div class="xblock-inputs editor_content_wrapper">', studio_fragment.content)
+    assert_in('<div class="xblock-actions">', studio_fragment.content)
+
 def test_studio_document_submit():
     block = make_document_block()
 
@@ -60,3 +87,29 @@ def test_calendar_document_submit():
     assert_equals(block.display_name, "Google Calendar")
     assert_equals(block.calendar_id, "google1234")
     assert_equals(block.default_view, 1)
+
+def test_check_document_url():
+    block = make_document_block()
+
+    data = json.dumps({
+        'url': "https://docs.google.com/presentation/d/1x2ZuzqHsMoh1epK8VsGAlanSo7r9z55ualwQlj-ofBQ/embed?start=true&loop=true&delayms=10000"
+    })
+    res = block.handle('check_url', make_request(data))
+
+    assert_equals(json.loads(res.body), {'status_code': 200})
+
+    data = json.dumps({
+        'url': 'undefined'
+    })
+    res = block.handle('check_url', make_request(data))
+
+    assert_equals(json.loads(res.body), {'status_code': 400})
+
+    data = json.dumps({
+        'url': "https://docs.google.com/presentation/d/1x2ZuzqHsMoh1epK8VsdsadfGAlanSo7r9z55ualwQlj-ofBQ/embed?start=true&loop=true&delayms=10000"
+    })
+    res = block.handle('check_url', make_request(data))
+
+    assert_equals(json.loads(res.body), {'status_code': 404})
+
+