2022-03-22 14:53:34 -07:00
|
|
|
import os
|
|
|
|
|
import unittest
|
|
|
|
|
from jc.exceptions import ParseError
|
|
|
|
|
import jc.parsers.asciitable
|
|
|
|
|
|
|
|
|
|
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MyTests(unittest.TestCase):
|
|
|
|
|
|
|
|
|
|
def test_asciitable_nodata(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with no data
|
|
|
|
|
"""
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse('', quiet=True), [])
|
|
|
|
|
|
|
|
|
|
def test_asciitable_m_pure_ascii(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a pure ASCII table
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
| type | tota | used | fr ee | shar | buff | avai |
|
|
|
|
|
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
| Mem | 3861 | 2228 | 3364 | 1183 | 2743 | 3389 |
|
|
|
|
|
+--------+--------+--------+--------+--------+--------+--------+
|
|
|
|
|
| | | | | test 2 | | |
|
|
|
|
|
+--------+--------+--------+--------+--------+--------+--------+
|
|
|
|
|
| last | last | last | ab cde | | | final |
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
'''
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"tota": "3861",
|
|
|
|
|
"used": "2228",
|
|
|
|
|
"fr_ee": "3364",
|
|
|
|
|
"shar": "1183",
|
|
|
|
|
"buff": "2743",
|
|
|
|
|
"avai": "3389"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": None,
|
|
|
|
|
"tota": None,
|
|
|
|
|
"used": None,
|
|
|
|
|
"fr_ee": None,
|
|
|
|
|
"shar": "test 2",
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": None
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "last",
|
|
|
|
|
"tota": "last",
|
|
|
|
|
"used": "last",
|
|
|
|
|
"fr_ee": "ab cde",
|
|
|
|
|
"shar": None,
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": "final"
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
|
|
|
|
def test_asciitable_m_unicode(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a unicode table
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
╒════════╤════════╤════════╤════════╤════════╤════════╤════════╕
|
|
|
|
|
│ type │ total │ used │ fr ee │ shar │ buff │ avai │
|
|
|
|
|
╞════════╪════════╪════════╪════════╪════════╪════════╪════════╡
|
|
|
|
|
│ Mem │ 3861 │ 2228 │ 3364 │ 1183 │ 2743 │ 3389 │
|
|
|
|
|
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
|
|
|
|
|
│ Swap │ 2097 │ 0 │ 2097 │ │ │ │
|
|
|
|
|
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
|
|
|
|
|
│ last │ last │ last │ ab cde │ │ │ final │
|
|
|
|
|
╘════════╧════════╧════════╧════════╧════════╧════════╧════════╛
|
|
|
|
|
'''
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"total": "3861",
|
|
|
|
|
"used": "2228",
|
|
|
|
|
"fr_ee": "3364",
|
|
|
|
|
"shar": "1183",
|
|
|
|
|
"buff": "2743",
|
|
|
|
|
"avai": "3389"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "Swap",
|
|
|
|
|
"total": "2097",
|
|
|
|
|
"used": "0",
|
|
|
|
|
"fr_ee": "2097",
|
|
|
|
|
"shar": None,
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": None
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "last",
|
|
|
|
|
"total": "last",
|
|
|
|
|
"used": "last",
|
|
|
|
|
"fr_ee": "ab cde",
|
|
|
|
|
"shar": None,
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": "final"
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
|
|
|
|
def test_asciitable_pure_ascii_extra_spaces(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a pure ASCII table that has heading and
|
|
|
|
|
trailing spaces and newlines.
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
| type | total | used | fr ee | shar | buff | avai
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
| Mem | 3861 | 2228 | 3364 | 1183 | 2743 | 3389
|
|
|
|
|
+--------+--------+--------+--------+--------+--------+--------+
|
|
|
|
|
| | | | test 2 | |
|
|
|
|
|
+--------+--------+--------+--------+--------+--------+--------+
|
|
|
|
|
| last | last | last | ab cde | | | final |
|
|
|
|
|
+========+========+========+========+========+========+========+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
'''
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"total": "3861",
|
|
|
|
|
"used": "2228",
|
|
|
|
|
"fr_ee": "3364",
|
|
|
|
|
"shar": "1183",
|
|
|
|
|
"buff": "2743",
|
|
|
|
|
"avai": "3389"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": None,
|
|
|
|
|
"total": None,
|
|
|
|
|
"used": None,
|
|
|
|
|
"fr_ee": None,
|
|
|
|
|
"shar": "test 2",
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": None
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "last",
|
|
|
|
|
"total": "last",
|
|
|
|
|
"used": "last",
|
|
|
|
|
"fr_ee": "ab cde",
|
|
|
|
|
"shar": None,
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": "final"
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
|
|
|
|
def test_asciitable_unicode_extra_spaces(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a pure ASCII table that has heading and
|
|
|
|
|
trailing spaces and newlines.
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
╒════════╤════════╤════════╤════════╤════════╤════════╤════════╕
|
|
|
|
|
type │ total │ used │ free │ shar │ buff │ avai
|
|
|
|
|
╞════════╪════════╪════════╪════════╪════════╪════════╪════════╡
|
|
|
|
|
Mem │ 3861 │ 2228 │ 3364 │ 1183 │ 2743 │ 3389
|
|
|
|
|
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
|
|
|
|
|
Swap │ 2097 │ 0 │ 2097 │ │ │
|
|
|
|
|
╘════════╧════════╧════════╧════════╧════════╧════════╧════════╛
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
'''
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"total": "3861",
|
|
|
|
|
"used": "2228",
|
|
|
|
|
"free": "3364",
|
|
|
|
|
"shar": "1183",
|
|
|
|
|
"buff": "2743",
|
|
|
|
|
"avai": "3389"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "Swap",
|
|
|
|
|
"total": "2097",
|
|
|
|
|
"used": "0",
|
|
|
|
|
"free": "2097",
|
|
|
|
|
"shar": None,
|
|
|
|
|
"buff": None,
|
|
|
|
|
"avai": None
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
|
|
|
|
def test_asciitable_markdown(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a markdown table
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
| type | total | used | free | shared | buff cache | available |
|
|
|
|
|
|--------|---------|--------|---------|----------|--------------|-------------|
|
|
|
|
|
| Mem | 3861332 | 222820 | 3364176 | 11832 | 274336 | 3389588 |
|
|
|
|
|
| Swap | 2097148 | 0 | 2097148 | | | |
|
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"total": "3861332",
|
|
|
|
|
"used": "222820",
|
|
|
|
|
"free": "3364176",
|
|
|
|
|
"shared": "11832",
|
|
|
|
|
"buff_cache": "274336",
|
|
|
|
|
"available": "3389588"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "Swap",
|
|
|
|
|
"total": "2097148",
|
|
|
|
|
"used": "0",
|
|
|
|
|
"free": "2097148",
|
|
|
|
|
"shared": None,
|
|
|
|
|
"buff_cache": None,
|
|
|
|
|
"available": None
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
|
|
|
|
def test_asciitable_simple(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a simple table
|
|
|
|
|
"""
|
|
|
|
|
input = '''
|
|
|
|
|
type total used free shared buff cache available
|
|
|
|
|
------ ------- ------ ------- -------- ------------ -----------
|
|
|
|
|
Mem 3861332 222820 3364176 11832 274336 3389588
|
|
|
|
|
Swap 2097148 0 2097148
|
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"type": "Mem",
|
|
|
|
|
"total": "3861332",
|
|
|
|
|
"used": "222820",
|
|
|
|
|
"free": "3364176",
|
|
|
|
|
"shared": "11832",
|
|
|
|
|
"buff_cache": "274336",
|
|
|
|
|
"available": "3389588"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"type": "Swap",
|
|
|
|
|
"total": "2097148",
|
|
|
|
|
"used": "0",
|
|
|
|
|
"free": "2097148",
|
|
|
|
|
"shared": None,
|
|
|
|
|
"buff_cache": None,
|
|
|
|
|
"available": None
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
2022-03-22 15:55:59 -07:00
|
|
|
def test_asciitable_pretty_ansi(self):
|
|
|
|
|
"""
|
|
|
|
|
Test 'asciitable' with a pretty table with ANSI codes
|
|
|
|
|
"""
|
|
|
|
|
input = '''┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━┓ \n ┃\x1b[1m \x1b[0m\x1b[1mReleased \x1b[0m\x1b[1m \x1b[0m┃\x1b[1m \x1b[0m\x1b[1mTitle \x1b[0m\x1b[1m \x1b[0m┃\x1b[1m \x1b[0m\x1b[1m Box Office\x1b[0m\x1b[1m \x1b[0m┃ \n ┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━┩ \n │\x1b[36m \x1b[0m\x1b[36mDec 20, 2019\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mStar Wars: The Rise of Skywalker \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m $952,110,690\x1b[0m\x1b[32m \x1b[0m│ \n │\x1b[36m \x1b[0m\x1b[36mMay 25, 2018\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mSolo: A Star Wars Story \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m $393,151,347\x1b[0m\x1b[32m \x1b[0m│ \n │\x1b[36m \x1b[0m\x1b[36mDec 15, 2017\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mStar Wars Ep. V111: The Last Jedi\x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m$1,332,539,889\x1b[0m\x1b[32m \x1b[0m│ \n │\x1b[36m \x1b[0m\x1b[36mDec 16, 2016\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mRogue One: A Star Wars Story \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m$1,332,439,889\x1b[0m\x1b[32m \x1b[0m│ \n └──────────────┴───────────────────────────────────┴────────────────┘ \n'''
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
{
|
|
|
|
|
"released": "Dec 20, 2019",
|
|
|
|
|
"title": "Star Wars: The Rise of Skywalker",
|
|
|
|
|
"box_office": "$952,110,690"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"released": "May 25, 2018",
|
|
|
|
|
"title": "Solo: A Star Wars Story",
|
|
|
|
|
"box_office": "$393,151,347"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"released": "Dec 15, 2017",
|
|
|
|
|
"title": "Star Wars Ep. V111: The Last Jedi",
|
|
|
|
|
"box_office": "$1,332,539,889"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"released": "Dec 16, 2016",
|
|
|
|
|
"title": "Rogue One: A Star Wars Story",
|
|
|
|
|
"box_office": "$1,332,439,889"
|
|
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
self.assertEqual(jc.parsers.asciitable.parse(input, quiet=True), expected)
|
|
|
|
|
|
2022-03-22 14:53:34 -07:00
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
unittest.main()
|
