def test_get_context_data(self, get_mock): request = RequestFactory().get("/") request.session = {"next_url": "/"} request.META["HTTP_HOST"] = "localhost:3333" view = Login() view.request = request response_mock = Mock(status_code=200) response_mock.json.return_value = { "name": "oauth", "data": { "authorizeUrl": "http://something.com/?redirect=__redirect_url__" } } get_mock.return_value = response_mock data = view.get_context_data() self.assertDictEqual( data["scheme_info"], { "name": "oauth", "data": { "authorizeUrl": "http://something.com/?redirect=__redirect_url__" }, }) self.assertEqual( data["authorize_url"], "http://something.com/?redirect=http://localhost:3333/auth/callback/" )
def test_scheme_info(self, get_mock): view = Login() expected_url = "{}/auth/scheme".format(settings.TSURU_HOST) self.assertDictEqual(view.scheme_info(), {}) get_mock.assert_called_with(expected_url) response_mock = Mock(status_code=200) response_mock.json.return_value = {"name": "oauth"} get_mock.return_value = response_mock self.assertDictEqual(view.scheme_info(), {"name": "oauth"}) get_mock.assert_called_with(expected_url)
def test_get_context_data_with_data_is_none(self, get_mock): request = RequestFactory().get("/") request.META["HTTP_HOST"] = "localhost:3333" request.session = {"next_url": "/"} view = Login() view.request = request response_mock = Mock(status_code=200) response_mock.json.return_value = {"name": "native", "data": None} get_mock.return_value = response_mock data = view.get_context_data() self.assertDictEqual(data["scheme_info"], { "name": "native", "data": None, })