Ejemplo n.º 1
0
    def test_request_raises_error(self, urlopen, request):
        exception_data = MagicMock()
        exception_data.read.return_value.decode.return_value = '{"error": 1}'
        urlopen.side_effect = HTTPError("", "", "", "", exception_data)

        ygg = YggAuth()
        self.assertFalse(urlopen.called)
        self.assertFalse(request.called)
        res = ygg._ygg_req("/test", {"a": "b"})

        # Read the response
        self.assertTrue(exception_data.read.called)
        self.assertTrue(exception_data.read.return_value.decode.called)

        self.assertEqual(res, {"error": 1})
Ejemplo n.º 2
0
    def test_request_is_done(self, urlopen, request):
        decode = urlopen.return_value.read.return_value.decode
        decode.return_value = '{"test": 1}'
        ygg = YggAuth()
        self.assertFalse(urlopen.called)
        self.assertFalse(request.called)
        res = ygg._ygg_req("/test", [{"a": "b"}, "c", "d", "e"])

        # First create the request
        request.assert_called_once_with(
            url="https://authserver.mojang.com/test",
            data=b'[{"a": "b"}, "c", "d", "e"]',
            headers={"Content-Type": "application/json"},
        )

        # Then send it
        urlopen.assert_called_once_with(request.return_value)

        # Read the response
        self.assertTrue(urlopen.return_value.read.called)
        self.assertTrue(decode.called)

        self.assertEqual(res, {"test": 1})
Ejemplo n.º 3
0
 def setUp(self):
     self.ygg = YggAuth()
     self.assertFalse(self.ygg.username)
     self.assertFalse(self.ygg.password)
     self.assertFalse(self.ygg.client_token)
     self.assertFalse(self.ygg.access_token)
Ejemplo n.º 4
0
class YggAuthTest(unittest.TestCase):
    def setUp(self):
        self.ygg = YggAuth()
        self.assertFalse(self.ygg.username)
        self.assertFalse(self.ygg.password)
        self.assertFalse(self.ygg.client_token)
        self.assertFalse(self.ygg.access_token)

    def test_authenticate_success(self, ygg_req):
        ygg_req.return_value = {"accessToken": "myaccess", "clientToken": "mytoken"}

        res = self.ygg.authenticate("user", "pass", "clienttoken")

        ygg_req.assert_called_once_with(
            "/authenticate",
            {
                "agent": {"name": "Minecraft", "version": 1},
                "username": "******",
                "password": "******",
                "clientToken": "clienttoken",
            },
        )

        self.assertEqual(self.ygg.username, "user")
        self.assertEqual(self.ygg.password, "pass")
        self.assertEqual(self.ygg.client_token, "mytoken")
        self.assertEqual(self.ygg.access_token, "myaccess")

        self.assertEqual(res, ygg_req.return_value)

    def test_authenticate_failure(self, ygg_req):
        ygg_req.return_value = {"error": 1}

        res = self.ygg.authenticate("user", "pass", "clienttoken")

        ygg_req.assert_called_once_with(
            "/authenticate",
            {
                "agent": {"name": "Minecraft", "version": 1},
                "username": "******",
                "password": "******",
                "clientToken": "clienttoken",
            },
        )

        self.assertEqual(self.ygg.username, "user")
        self.assertEqual(self.ygg.password, "pass")
        self.assertEqual(self.ygg.client_token, "clienttoken")
        self.assertEqual(self.ygg.access_token, None)

        self.assertEqual(res, ygg_req.return_value)

    def test_refresh_success(self, ygg_req):
        ygg_req.return_value = {"accessToken": "myaccess", "clientToken": "mytoken"}

        res = self.ygg.refresh("clienttoken", "accesstoken")

        ygg_req.assert_called_once_with("/refresh", {"accessToken": "accesstoken", "clientToken": "clienttoken"})

        self.assertEqual(self.ygg.username, None)
        self.assertEqual(self.ygg.password, None)
        self.assertEqual(self.ygg.client_token, "mytoken")
        self.assertEqual(self.ygg.access_token, "myaccess")

        self.assertEqual(res, ygg_req.return_value)

    def test_refresh_failure(self, ygg_req):
        ygg_req.return_value = {"error": 1}

        res = self.ygg.refresh("clienttoken", "accesstoken")

        ygg_req.assert_called_once_with("/refresh", {"accessToken": "accesstoken", "clientToken": "clienttoken"})

        self.assertEqual(self.ygg.username, None)
        self.assertEqual(self.ygg.password, None)
        self.assertEqual(self.ygg.client_token, "clienttoken")
        self.assertEqual(self.ygg.access_token, "accesstoken")

        self.assertEqual(res, ygg_req.return_value)

    def test_signout(self, ygg_req):
        ygg_req.return_value = {"whatever": "dict"}

        res = self.ygg.signout("user", "pass")

        ygg_req.assert_called_once_with("/signout", {"username": "******", "password": "******"})

        self.assertEqual(self.ygg.username, "user")
        self.assertEqual(self.ygg.password, "pass")
        self.assertEqual(self.ygg.client_token, None)
        self.assertEqual(self.ygg.access_token, None)

        self.assertEqual(res, ygg_req.return_value)

    def test_invalidate(self, ygg_req):
        ygg_req.return_value = {"whatever": "dict"}

        res = self.ygg.invalidate("clienttoken", "accesstoken")

        ygg_req.assert_called_once_with("/invalidate", {"clientToken": "clienttoken", "accessToken": "accesstoken"})

        self.assertEqual(self.ygg.username, None)
        self.assertEqual(self.ygg.password, None)
        self.assertEqual(self.ygg.client_token, "clienttoken")
        self.assertEqual(self.ygg.access_token, "accesstoken")

        self.assertEqual(res, ygg_req.return_value)

    def test_validate(self, ygg_req):
        ygg_req.return_value = {"whatever": "dict"}

        res = self.ygg.validate("accesstoken")

        ygg_req.assert_called_once_with("/validate", {"accessToken": "accesstoken"})

        self.assertEqual(self.ygg.username, None)
        self.assertEqual(self.ygg.password, None)
        self.assertEqual(self.ygg.client_token, None)
        self.assertEqual(self.ygg.access_token, "accesstoken")

        self.assertEqual(res, ygg_req.return_value)