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
|
import os
import unittest
import json
from typing import Dict
import jc.parsers.proc_net_netstat
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
class MyTests(unittest.TestCase):
f_in: Dict = {}
f_json: Dict = {}
@classmethod
def setUpClass(cls):
fixtures = {
'proc_net_netstat': (
'fixtures/linux-proc/net_netstat',
'fixtures/linux-proc/net_netstat.json')
}
for file, filepaths in fixtures.items():
with open(os.path.join(THIS_DIR, filepaths[0]), 'r', encoding='utf-8') as a, \
open(os.path.join(THIS_DIR, filepaths[1]), 'r', encoding='utf-8') as b:
cls.f_in[file] = a.read()
cls.f_json[file] = json.loads(b.read())
def test_proc_net_netstat_nodata(self):
"""
Test 'proc_net_netstat' with no data
"""
self.assertEqual(jc.parsers.proc_net_netstat.parse('', quiet=True), [])
def test_proc_net_netstat(self):
"""
Test '/proc/net/netstat'
"""
self.assertEqual(jc.parsers.proc_net_netstat.parse(self.f_in['proc_net_netstat'], quiet=True),
self.f_json['proc_net_netstat'])
if __name__ == '__main__':
unittest.main()
|