def test_get_survey_form_with_request_form(self):
app.SIGN_KEY = "test_key"
email = "*****@*****.**"
d = {"work": "dba"}
with app.app.test_request_context("/"):
form = app.get_survey_form(email, werkzeug.ImmutableMultiDict(d))
self.assertEqual("dba", form.work.data)
app.SIGN_KEY = None
def test_save_user_should_pass_form_to_template(self, mock):
reload(app)
app.SIGN_KEY = "test_key"
with patch("app.get_survey_form"):
with app.app.test_request_context("/"):
app.before_request()
app.save_user("First", "Last", "*****@*****.**")
app.teardown_request(None)
mock.assert_called_with("confirmation.html", form=app.get_survey_form("*****@*****.**"))
app.SIGN_KEY = None
def test_confirmation_with_email_and_sign(self, render, mock_get, mock):
self._mock_requests(
mock, mock_get, {"email": "*****@*****.**", "name": "Foo Bar"}, {"access_token": "testtoken"}
)
render.return_value = ""
reload(app)
app.SIGN_KEY = "key"
with patch("app.get_survey_form"):
resp = self.api.get("/register/github?code=code21")
self.assertEqual(200, resp.status_code)
render.assert_called_with("confirmation.html", form=app.get_survey_form("*****@*****.**"))
def test_confirmation_template_with_email_and_signature(self, render, get):
data = {"first_name": "First", "last_name": "Last", "email": "*****@*****.**"}
self._mock_requests(get, data)
render.return_value = ""
reload(app)
app.SIGN_KEY = "key"
self.addCleanup(lambda: self.db.users.remove())
with patch("app.get_survey_form"):
resp = self.api.get("/register/facebook?access_token=123awesometoken456")
self.assertEqual(200, resp.status_code)
render.assert_called_with("confirmation.html", form=app.get_survey_form(data["email"]))
def test_save_user(self, render):
render.return_value = "template rendered"
reload(app)
app.SIGN_KEY = "123456"
with patch("app.get_survey_form"):
with app.app.test_request_context("/"):
app.before_request()
content, status = app.save_user("Francisco", "Souza", "*****@*****.**", identity="123")
app.teardown_request(None)
self.assertEqual("template rendered", content)
self.assertEqual(200, status)
render.assert_called_with("confirmation.html", form=app.get_survey_form("*****@*****.**"))
u = self.db.users.find_one(
{"email": "*****@*****.**", "first_name": "Francisco", "last_name": "Souza", "identity": "123"}
)
self.assertIsNotNone(u)
def test_confirmation_template_with_email_in_context(self, render, get):
app.GOOGLE_USER_IP = "127.0.0.1"
app.GOOGLE_API_KEY = "key"
self.addCleanup(self.clean_api_client)
m = Mock()
m.json.return_value = {
"id": "1234",
"email": "*****@*****.**",
"verified_email": True,
"given_name": "Francisco",
"family_name": "Souza",
"name": "Francisco Souza",
"gender": "male",
}
get.return_value = m
render.return_value = ""
reload(app)
app.SIGN_KEY = "key"
with patch("app.get_survey_form"):
resp = self.api.get("/register/gplus?token=mytoken&token_type=Bearer")
self.assertEqual(200, resp.status_code)
render.assert_called_with("confirmation.html", form=app.get_survey_form("*****@*****.**"))
def test_get_survey_form_should_initialize_signature_field(self, mock):
app.SIGN_KEY = "test_key"
email = "*****@*****.**"
form = app.get_survey_form(email)
self.assertEqual(app.sign(email), form.signature.data)
app.SIGN_KEY = None
def test_get_survey_form_should_initializa_email_field(self, mock):
app.SIGN_KEY = "test_key"
email = "*****@*****.**"
form = app.get_survey_form(email)
self.assertEqual(email, form.email.data)
app.SIGN_KEY = None
def test_get_survey_form_should_return_form(self, mock):
app.SIGN_KEY = "test_key"
app.get_survey_form("*****@*****.**")
mock.assert_called_once()
app.SIGN_KEY = None
