def test_utc_offset_kwarg_authority(self, mocked_urlopen):
'''UnifiVideoAPI init with utc_offset_sec should ignore offsets
reported by UniFi Video server.
'''
camera = DatetimeTimezoneTests.camera_w_offset(5)
utc_offset_sec = 12345
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(camera).encode('utf8')},
{'data': json.dumps(DatetimeTimezoneTests.bootstrap['3.9.12'])\
.encode('utf8')},
])
self.assertEqual(
UnifiVideoAPI(
api_key='****', utc_offset_sec=utc_offset_sec).utc_offset,
12345)
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(camera).encode('utf8')},
{'data': json.dumps(DatetimeTimezoneTests.bootstrap['3.10.13'])\
.encode('utf8')},
])
self.assertEqual(
UnifiVideoAPI(
api_key='****', utc_offset_sec=utc_offset_sec).utc_offset,
12345)
def test_init_against_bootstrap(self, mocked_urlopen):
"""Test API init against fresh bootstrap.json
Run through bootstrap JSONs from various UniFi Video versions.
"""
for bootstrap in files.test_files['bootstrap.json']:
with open(
os.path.join(
os.path.dirname(__file__),
'files/{}'.format(bootstrap['filename'])),
'rb') as f:
bootstrap_json = f.read()
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
# For GET /recording?...
{'data': json.dumps(empty_response).encode('utf8')},
# For GET /camera
{'data': json.dumps(empty_response).encode('utf8')},
# For GET /bootstrap
{'data': bootstrap_json }
],
set_cookies=True)
uva = UnifiVideoAPI(api_key='xxx')
self.assertEqual(len(uva.cameras), 0)
self.assertEqual(len(uva.recordings), 0)
self.assertEqual(uva.version, bootstrap['unifi_video_version'])
def test_aa_api_init(self, mocked_urlopen):
"""Test API init
UnifiVideoAPI should:
- throw ValueError when called without usermame-password pair
or API key
- throw UnifiVideoVersionError when check_ufv_version == True
and GET bootstrap response has an unknown version string
- throw UnifiVideoVersionError when check_ufv_version == True
and GET bootstrap response has an out of range version
"""
self.assertRaises(ValueError, UnifiVideoAPI)
# Invalid version
res_w_unk_ufv_ver = json.loads(responses['bootstrap'])
res_w_unk_ufv_ver['data'][0]['systemInfo']['version'] = 'qwerty'
mocked_urlopen.side_effect = mocked_response(json.dumps(
res_w_unk_ufv_ver).encode('utf8'), set_cookies=True)
self.assertRaises(UnifiVideoVersionError, UnifiVideoAPI,
api_key='xxxxxx')
# Version Lower than range
res_w_low_ufv_ver = json.loads(responses['bootstrap'])
res_w_low_ufv_ver['data'][0]['systemInfo']['version'] = '0.0.1'
mocked_urlopen.side_effect = mocked_response(json.dumps(
res_w_low_ufv_ver).encode('utf8'), set_cookies=True)
self.assertRaises(UnifiVideoVersionError, UnifiVideoAPI,
api_key='xxxxxx')
# Version Higher than range
res_w_high_ufv_ver = json.loads(responses['bootstrap'])
res_w_high_ufv_ver['data'][0]['systemInfo']['version'] = '99999.0.0'
mocked_urlopen.side_effect = mocked_response(json.dumps(
res_w_high_ufv_ver).encode('utf8'), set_cookies=True)
self.assertRaises(UnifiVideoVersionError, UnifiVideoAPI,
api_key='xxxxxx')
# Version in ok range
res_w_ok_ufv_ver = json.loads(responses['bootstrap'])
res_w_ok_ufv_ver['data'][0]['systemInfo']['version'] = '3.10.6'
mocked_urlopen.side_effect = mocked_response(arg_pile=[
{'data': responses['recordings']},
{'data': responses['camera']},
{'data': json.dumps(res_w_ok_ufv_ver).encode('utf8')},
])
self.assertIsInstance(UnifiVideoAPI(api_key='xxxxx'), UnifiVideoAPI)
def test_nooffset_init(self, mocked_urlopen):
'''Naive UnifiVideoAPI init against zero cam UniFi Video
version < v3.10.2 should result in utc_offset == None
'''
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(DatetimeTimezoneTests.bootstrap['3.9.12'])\
.encode('utf8')},
])
self.assertIsNone(UnifiVideoAPI(api_key='****').utc_offset)
def test_offset_from_cam(self, mocked_urlopen):
'''Naive UnifiVideoAPI init against UniFi Video version < v3.10.2
should result in utc_offset matching that of attached cameras
'''
camera = DatetimeTimezoneTests.camera_w_offset(5)
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(camera).encode('utf8')},
{'data': json.dumps(DatetimeTimezoneTests.bootstrap['3.9.12'])\
.encode('utf8')},
])
self.assertEqual(UnifiVideoAPI(api_key='****').utc_offset, 5 * 3600)
def test_camera_offset_consensus(self, mocked_urlopen):
'''Naive UnifiVideoAPI init against UniFi Video version < v3.10.2
with cameras of differing offsets should ignore any offset reported
by the cameras.
'''
cameras = {
'data': [
c['data'][0] for c in [
DatetimeTimezoneTests.cam_oid(
DatetimeTimezoneTests.camera_w_offset(i * 3600),
'{:024d}'.format(i))
for i in (1, 2, 3)]]
}
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(cameras).encode('utf8')},
{'data': json.dumps(DatetimeTimezoneTests.bootstrap['3.9.12'])\
.encode('utf8')},
])
self.assertIsNone(UnifiVideoAPI(api_key='****').utc_offset)
auth=mqtt_auth,
tls=None)
except Exception as ex:
logging.error(
'Exception while publishing to MQTT broker: {}'.format(ex))
while True:
start = datetime.now()
try:
# Use HTTPS and skip cert verification
uva = UnifiVideoAPI(api_key=nvr_api_key,
addr=nvr_host,
port=7443,
schema='https',
verify_cert=False)
except Exception as ex:
logging.error('Exception while communicating with : {}'.format(ex))
mqtt_msgs = []
for camera in uva.cameras:
name = camera.name
lastRecord = camera._data.get('lastRecordingStartTime', None)
if lastRecord != None and lastRecordMemory.get(name,
None) != lastRecord:
lastRecordMemory[name] = lastRecord
lastRecordReadable = datetime.utcfromtimestamp(
def get_ufva_w_mocked_urlopen(mocked_urlopen, *args, **kwargs):
mocked_urlopen.side_effect = mocked_response(*args, **kwargs)
return UnifiVideoAPI(username='******',
password='******',
addr='0.0.0.0')
def test_collections(self, mocked_urlopen):
'''Camera collections should be clear of any stales after refresh
'''
def oid(i):
return '{:024d}'.format(i)
bootstrap_version = '3.10.13'
sample_camera = json.loads(responses['camera'].decode('utf8'))
stages = [[
{'managed': True, 'state': 'CONNECTED', '_id': 0},
{'managed': True, 'state': 'CONNECTED', '_id': 1},
{'managed': True, 'state': 'CONNECTED', '_id': 2},
{'managed': True, 'state': 'DISCONNECTED', '_id': 3},
{'managed': True, 'state': 'DISCONNECTED', '_id': 4},
{'managed': False, 'state': 'CONNECTED', '_id': 5},
{
'cameras': {'ids': (0, 1, 2, 3, 4, 5)},
'managed_cameras': {'ids': (0, 1, 2, 3, 4)},
'active_cameras': {'ids': (0, 1, 2)},
}
], [
{'managed': True, 'state': 'CONNECTED', '_id': 0},
{'managed': True, 'state': 'CONNECTED', '_id': 1},
{'managed': True, 'state': 'DISCONNECTED', '_id': 2},
{'managed': True, 'state': 'DISCONNECTED', '_id': 3},
{'managed': True, 'state': 'DISCONNECTED', '_id': 4},
{'managed': False, 'state': 'CONNECTED', '_id': 5},
{
'cameras': {'ids': (0, 1, 2, 3, 4, 5)},
'managed_cameras': {'ids': (0, 1, 2, 3, 4)},
'active_cameras': {'ids': (0, 1)},
}
], [
{'managed': False, 'state': 'CONNECTED', '_id': 5},
{
'cameras': {'ids': (5,)},
'managed_cameras': {'ids': ()},
'active_cameras': {'ids': ()},
}
], [
{'managed': True, 'state': 'DISCONNECTED', '_id': 0},
{'managed': True, 'state': 'DISCONNECTED', '_id': 1},
{'managed': True, 'state': 'DISCONNECTED', '_id': 2},
{'managed': True, 'state': 'DISCONNECTED', '_id': 3},
{'managed': True, 'state': 'DISCONNECTED', '_id': 4},
{'managed': False, 'state': 'CONNECTED', '_id': 5},
{'managed': True, 'state': 'CONNECTED', '_id': 6},
{
'cameras': {'ids': (0, 1, 2, 3, 4, 5, 6)},
'managed_cameras': {'ids': (0, 1, 2, 3, 4, 6)},
'active_cameras': {'ids': (6,)},
}
],
]
def def_camera_data(fields):
fields['_id'] = oid(fields['_id'])
camera = deepcopy(sample_camera['data'][0])
camera.update(fields)
return camera
for bootstrap in files.test_files['bootstrap.json']:
if bootstrap['unifi_video_version'] == bootstrap_version:
bootstrap_fn = os.path.join(
os.path.dirname(__file__), 'files', bootstrap['filename'])
with open(bootstrap_fn, 'rb') as f:
bootstrap = json.loads(f.read().decode('utf8'))
#
# Init UnifiVideoAPI with empty camera and recording collections
#
mocked_urlopen.side_effect = mocked_response(
arg_pile=[
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(empty_response).encode('utf8')},
{'data': json.dumps(bootstrap).encode('utf8')},
],
set_cookies=True)
uva = UnifiVideoAPI(api_key='****')
for coll in (uva.cameras, uva.active_cameras, uva.managed_cameras):
self.assertEqual(len(coll), 0)
#
# Load the stages, test count and membership for all three camera
# collections
#
for stage in stages:
tests = stage.pop()
mocked_urlopen.side_effect = mocked_response('{}'.encode('utf8'))
mocked_urlopen.side_effect = mocked_response(
json.dumps(
{'data': list(map(def_camera_data, stage))}).encode('utf8'))
uva.refresh_cameras()
for coll_name, expected in tests.items():
self.assertEqual(
len(getattr(uva, coll_name)),
len(expected['ids']))
self.assertEqual(
set([oid(i) for i in expected['ids']]),
set([cam._id for cam in getattr(uva, coll_name)]))