Python Captcha.get_captcha Examples

Example #1

File: test_captcha.py Project: fightwing/play_with_12306

class TestCaptcha(TestCase):
    def setUp(self):
        self.captcha = Captcha()
        result = Utils.get_json_data_from_file('./', 'captcha.json')
        api.get = MagicMock()
        api.get().json = MagicMock(return_value=result)

    def test_get_captcha(self):
        image_data = self.captcha.get_captcha()
        self.assertIsNotNone(image_data)
        self.assertTrue(Utils.is_base64(image_data.encode()))

Example #2

class TestCaptcha(TestCase):
    def setUp(self):
        self.captcha = Captcha()
        result = Utils.get_json_data_from_file('./', 'captcha.json')
        api.get = MagicMock()
        api.get().json = MagicMock(return_value=result)

    def test_get_captcha(self):
        image_data = self.captcha.get_captcha()
        self.assertIsNotNone(image_data)
        self.assertTrue(Utils.is_base64(image_data.encode()))

    def test_check_captcha_auto(self):
        image_data = Utils.get_json_data_from_file('./', 'captcha.json')
        image_indexes = Captcha.verify_captcha_auto(image_data["image"])
        self.assertEqual(image_indexes, [2, 4])