def test_raise_exception(self):
with self.assertRaises(ValueError):
parse_int(' ', safe=False)
with self.assertRaises(ValueError):
parse_int('a', safe=False)
with self.assertRaises(ValueError):
parse_int('1.2', safe=False)
def get_int_from_element(element: XmlElement, name: str) -> Optional[int]:
return parse_int(get_text_from_element(element, name))
def resolve_current(root, _info):
_current = get_int_from_element(root, 'count')
if _current is None:
return parse_int(root.text)
return _current
def resolve_counts(result_count: XmlElement, _info):
current = get_text(result_count)
return parse_int(current)
def test_valid_int(self):
self.assertEqual(parse_int('1'), 1)
self.assertEqual(parse_int('666'), 666)
self.assertEqual(parse_int('01'), 1)
self.assertEqual(parse_int(' 1'), 1)
self.assertEqual(parse_int(' 666 '), 666)
def test_none(self):
self.assertIsNone(parse_int(None))
self.assertIsNone(parse_int(''))
self.assertIsNone(parse_int('a'))
self.assertIsNone(parse_int('1.2'))
def get_offset_from_cursor(cursor: str) -> str:
plain_cursor = b64decode(cursor.encode('utf-8')).decode('utf-8')
_, offset = plain_cursor.split(':', 1)
return parse_int(offset, safe=False)
def resolve_max_concurrent_hosts(root: Dict[str, str], _info) -> int:
return parse_int(root.get("max_hosts"))
def resolve_create_assets_min_qod(root: Dict[str, str], _info) -> int:
return parse_int(root.get("assets_min_qod"))
def resolve_auto_delete_reports(root: Dict[str, str], _info) -> int:
auto_delete = root.get("auto_delete")
if auto_delete == "keep":
return parse_int(root.get("auto_delete_data"))
return None
