forked from ZennoLab/capmonstercloud-client-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfc_image_test.py
More file actions
106 lines (84 loc) · 6.79 KB
/
Copy pathfc_image_test.py
File metadata and controls
106 lines (84 loc) · 6.79 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import unittest
from pydantic.error_wrappers import ValidationError
from capmonstercloudclient.requests import FunCaptchaComplexImageTaskRequest
from capmonstercloudclient.exceptions import NumbersImagesErrors, TaskNotDefinedError, ZeroImagesErrors, \
UserAgentNotDefinedError
class RecaptchaImageRequestTest(unittest.TestCase):
websiteUrlExample = 'https://lessons.zennolab.com/captchas/recaptcha/v2_simple.php?level=middle'
imageUrlsExamples = ["https://i.postimg.cc/s2ZDrHXy/fc1.jpg"]
metadataExample = {'Task': 'Pick the image that is the correct way up'}
def testImagesTypes(self):
with self.assertRaises(ValidationError):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesBase64='[]')
with self.assertRaises(ValidationError):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesUrls='[]')
def testImagesFilling(self):
with self.assertRaises(ZeroImagesErrors,
msg='Empty array imagesUrls must be cause ZeroImagesErrors'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesUrls=[],
userAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36.")
with self.assertRaises(ZeroImagesErrors,
msg='Empty array imagesBase64 must be cause ZeroImagesErrors'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesBase64=[])
with self.assertRaises(ZeroImagesErrors,
msg='Both empty arrays imagesBase64 and imagesUrls must be cause ZeroImagesErrors'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample)
request.getTaskDict()
def testNumbersImages(self):
with self.assertRaises(NumbersImagesErrors,
msg='Empty array must be cause ZeroImagesErrors'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesUrls=RecaptchaImageRequestTest.imageUrlsExamples*2,
userAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36.")
with self.assertRaises(NumbersImagesErrors,
msg='Empty array must be cause ZeroImagesErrors'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesBase64=['base64image']*2,
userAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36.")
def testAllRequiredFieldsFilling(self):
required_fields = ['class', 'type', 'websiteUrl', 'metadata', 'userAgent']
metadata_fields = ['Task']
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata=RecaptchaImageRequestTest.metadataExample,
imagesUrls=RecaptchaImageRequestTest.imageUrlsExamples,
userAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36.")
request_dict = request.getTaskDict()
for i in required_fields:
self.assertTrue(i in list(request_dict.keys()),
msg=f'Required field {i} not in {request_dict}')
metadata_dict = request_dict['metadata']
for i in metadata_fields:
self.assertTrue(i in list(metadata_dict.keys()),
msg=f'Required field {i} not in {request_dict}')
self.assertEqual(request_dict['class'], 'funcaptcha')
self.assertEqual(request_dict['type'], 'ComplexImageTask')
def testTaskDefined(self):
with self.assertRaises(TaskNotDefinedError,
msg='Expect that empty "Task" field will be cause TaskNotDefinedError'):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata={},
imagesUrls=RecaptchaImageRequestTest.imageUrlsExamples)
with self.assertRaises(TaskNotDefinedError):
request = FunCaptchaComplexImageTaskRequest(websiteUrl=RecaptchaImageRequestTest.websiteUrlExample,
metadata={'dsfsdf': 'sdfsdf'},
imagesUrls=RecaptchaImageRequestTest.imageUrlsExamples,
userAgent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36.")
@unittest.skip('Doesnt work right now')
def testUserAgentWithUrl(self):
with self.assertRaises(UserAgentNotDefinedError):
request = FunCaptchaComplexImageTaskRequest(metadata=RecaptchaImageRequestTest.metadataExample,
imagesUrls=RecaptchaImageRequestTest.imageUrlsExamples)
request.getTaskDict()
if __name__ == '__main__':
unittest.main()