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
|
import requests
import socket
import time
import json
import mmap
from test_helper import ApiTestCase
class TestBasics(ApiTestCase):
def test_unauth(self):
r = requests.get(self.url("/?command=stats"))
self.assertEqual(r.status_code, requests.codes.unauthorized)
def test_auth_stats(self):
r = self.session.get(self.url("/?command=stats"))
self.assertEqual(r.status_code, requests.codes.ok)
def test_ping(self):
for _ in range(10):
r = self.pingFunc()
j = r.json()
if j['status'] == 'ok':
break
time.sleep(1)
self.assertEqual(j['status'], 'ok')
def test_ping_acl_deny(self):
self.writeCmdToConsole('setACL({})')
r = self.pingFunc()
self.assertEqual(r.status_code, 401)
j = r.json()
self.assertEqual(j['status'], 'failure')
self.assertEqual(j['reason'], 'Source IP Address not in ACL')
self.writeCmdToConsole('setACL({"0.0.0.0/0"})')
def test_getBL(self):
r = self.getBLFunc()
j = r.json()
self.assertEqual(j['bl_entries'], [])
def test_customFunc(self):
r = self.customFunc("custom1")
j = r.json()
self.assertEqual(j['r_attrs']['login'], 'custom1')
def test_customGetFunc(self):
r = self.customGetFunc("testCustomGet")
t = r.text
self.assertEqual(t, '1.2.3.4/32\n')
def test_deviceParsing(self):
r = self.allowFuncDeviceProtocol('foobar', '127.0.0.1', "12432", '"name" "Mac OS X Mail" "version" "10.0 (3226)" "os" "Mac OS X" "os-version" "10.12 (16A323)" "vendor" "Apple Inc."', "imap")
j = r.json()
self.assertRegex(json.dumps(j), "Mac OS X")
r = self.allowFuncDeviceProtocol('foobar', '127.0.0.1', "12432", 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_3) AppleWebKit/537.75.14 (KHTML, like Gecko) Version/7.0.3 Safari/7046A194A', "http")
j = r.json()
self.assertRegex(json.dumps(j), "Mac OS X")
r = self.allowFuncDeviceProtocol('foobar', '127.0.0.1', "12432", 'OpenXchange.Android.Mail/1.0+1234 (OS: 7.0; device: Samsung/GT9700)', "mobileapi")
j = r.json()
self.assertRegex(json.dumps(j), "Android")
def test_getDBStats(self):
self.reportFunc('dbstats', '1.4.3.2', '1234', False);
r = self.getDBStatsLogin('dbstats')
j = r.json();
self.assertRegex(json.dumps(j), "countLogins")
r = self.getDBStatsIP('1.4.3.2')
j = r.json();
self.assertRegex(json.dumps(j), "countLogins")
self.assertRegex(json.dumps(j), "bl_reason")
self.assertRegex(json.dumps(j), '"blacklisted": false')
def chunkGen(self):
payload = dict()
payload['login'] = "chunky"
payload['remote'] = "127.0.0.1"
payload['pwhash'] = "1234"
payload['success'] = True
payload['attrs'] = {}
yield json.dumps(payload).encode()
def testChunked(self):
r = self.session.post(
self.url("/?command=report"),
data=self.chunkGen(),
headers={'Content-Type': 'application/json'})
j = r.json()
self.assertEqual(j['status'], 'ok')
|
