Test for embed handler

Author: arikfr

tests/handlers/test_embed.py

@@ -0,0 +1,12 @@
+from tests import BaseTestCase
+
+
+class TestEmbedVisualization(BaseTestCase):
+    def test_sucesss(self):
+        vis = self.factory.create_visualization()
+        vis.query.latest_query_data = self.factory.create_query_result()
+        vis.query.save()
+
+        res = self.make_request("get", "/embed/query/{}/visualization/{}".format(vis.query.id, vis.id), is_json=False)
+        self.assertEqual(res.status_code, 200)
+

tests/test_authentication.py

@@ -121,10 +121,8 @@ def test_creates_vaild_new_user(self):
         name = 'Test User'
 
         with patch('redash.authentication.google_oauth.login_user') as login_user_mock:
-
             create_and_login_user(self.factory.org, name, email)
 
             self.assertTrue(login_user_mock.called)
             user = models.User.get(models.User.email == email)
 
-