Signed-off-by: Fabiano Fidêncio <fidencio@xxxxxxxxxx>
---
tests/osinfo.py | 44 ++++++++++++
tests/test_isoinfo.py | 151 ++++++++++++++++++++++++++++++++++++++++++
2 files changed, 195 insertions(+)
create mode 100644 tests/test_isoinfo.py
diff --git a/tests/osinfo.py b/tests/osinfo.py
index 4391cd0..0f4205d 100644
--- a/tests/osinfo.py
+++ b/tests/osinfo.py
@@ -37,6 +37,11 @@ class Os():
return shortid.text
shortid = property(_get_shortid)
+ def _get_distro(self):
+ distro = self._root.find('distro')
+ return distro.text
+ distro = property(_get_distro)
+
class Image():
def __init__(self, root):
@@ -61,6 +66,13 @@ class Media():
return None
url = property(_get_url)
+ def _get_iso(self):
+ iso = self._root.find('iso')
+ if iso is not None:
+ return ISO(iso)
+ return None
+ iso = property(_get_iso)
+
class Tree():
def __init__(self, root):
@@ -84,3 +96,35 @@ class URL():
logging.info("response: %s; code: %d",
responses[response.status_code], response.status_code)
return response.ok
+
+
+class ISO():
+ def __init__(self, root):
+ self._root = root
+
+ def _get_value(self, name, return_type=str, default=''):
+ entry = self._root.find(name)
+ return return_type(entry.text) if entry is not None else default
+
+ def _get_volumeid(self):
+ return self._get_value('volume-id')
+ volumeid = property(_get_volumeid)
+
+ def _get_publisherid(self):
+ return self._get_value('publisher-id')
+ publisherid = property(_get_publisherid)
+
+ def _get_applicationid(self):
+ return self._get_value('application-id')
+ applicationid = property(_get_applicationid)
+
+ def _get_systemid(self):
+ return self._get_value('system-id')
+ systemid = property(_get_systemid)
+
+ def _get_volumesize(self):
+ return self._get_value('volume-size', int, 0)
+ volumesize = property(_get_volumesize)
+
+
+
diff --git a/tests/test_isoinfo.py b/tests/test_isoinfo.py
new file mode 100644
index 0000000..504892c
--- /dev/null
+++ b/tests/test_isoinfo.py
@@ -0,0 +1,151 @@
+#!/usr/bin/env python3
+
+
+import logging
+import os
+import re
+import pytest
+
+from . import util
+
+
+OSES = util.oses()
+
+def _os_id(_os):
+ return _os.shortid
+
+@pytest.mark.parametrize('_os', [*OSES], ids=_os_id)
+def test_iso_detection(_os):
+ for isodatamedia in _get_isodatamedias(_os):
+ detected = []
+ for __os in OSES:
+ for media in __os.medias:
+ if isodatamedia.match(media.iso):
+ if _os.shortid != __os.shortid:
+ logging.warning(
+ 'ISO \'%s\' was matched by OS \'%s\' while it '
+ 'should only be matched by OS \'%s\'',
+ isodatamedia.filename, __os.shortid, _os.shortid)
+ else:
+ logging.info('ISO \'%s\' matched by OS \'%s\'',
+ isodatamedia.filename, __os.shortid)
+ # For several distros we do not have the volume-size
+ # set as part of our DB, thus multiple detections may
+ # occur. Although this case is not the optimal, as long
+ # as we detect the very same distro it's okay-ish.
+ if not __os.shortid in detected:
+ detected.append(__os.shortid)
+
+ if len(detected) != 1:
+ logging.warning('Some ISOs have been matched several times by '
+ 'the different OSes, which indicates an issue '
+ 'in the DB entries.')
+ assert len(detected) == 1
+
+
+class _ISODataMedia():
+ def __init__(self, filename, shortid, volumeid, publisherid, systemid,
+ applicationid, volumesize):
+
+ self.filename = filename
+ self.shortid = shortid
+ self.volumeid = volumeid if volumeid is not None else ''
+ self.publisherid = publisherid if publisherid is not None else ''
+ self.systemid = systemid if systemid is not None else ''
+ self.applicationid = applicationid \
+ if applicationid is not None else ''
+ self.volumesize = volumesize if volumesize is not None else 0
+
+ def match(self, media):
+ if media.volumeid == '' and \
+ media.systemid == '' and \
+ media.publisherid == '' and \
+ media.applicationid == '' and \
+ media.volumesize == 0:
+ return False
+
+ volumesize = media.volumesize
+ if volumesize == 0:
+ volumesize = self.volumesize
+
+ logging.warning("media.volumeid: %s | self.volumeid: %s", media.volumeid, self.volumeid)
+ if bool(re.match(media.volumeid, self.volumeid)) and \
+ bool(re.match(media.publisherid, self.publisherid)) and \
+ bool(re.match(media.applicationid, self.applicationid)) and \
+ bool(re.match(media.systemid, self.systemid)) and \
+ volumesize == self.volumesize:
+ return True
+
+ return False
+
+def _get_value(string, prefix, return_type=str):
+ if string.startswith(prefix):
+ return return_type(string.split(': ')[-1].strip())
+ return None
+
+def _get_volumeid(string):
+ return _get_value(string, 'Volume id: ')
+
+def _get_publisherid(string):
+ return _get_value(string, 'Publisher id: ')
+
+def _get_systemid(string):
+ return _get_value(string, 'System id: ')
+
+def _get_applicationid(string):
+ return _get_value(string, 'Application id: ')
+
+def _get_logicalblock(string):
+ return _get_value(string, 'Logical block size is: ', int)
+
+def _get_volumesize(string):
+ return _get_value(string, 'Volume size is: ', int)
+
+def _get_isodatamedia(filepath, shortid):
+ volumeid = None
+ publisherid = None
+ systemid = None
+ applicationid = None
+ logicalblock = None
+ volumesize = None
+
+ with open(filepath, 'r') as out:
+ for line in out.readlines():
+ if volumeid is None:
+ volumeid = _get_volumeid(line)
+ if publisherid is None:
+ publisherid = _get_publisherid(line)
+ if systemid is None:
+ systemid = _get_systemid(line)
+ if applicationid is None:
+ applicationid = _get_applicationid(line)
+ if logicalblock is None:
+ logicalblock = _get_logicalblock(line)
+ if volumesize is None:
+ volumesize = _get_volumesize(line)
+
+ if logicalblock is not None and volumesize is not None:
+ volumesize *= logicalblock
+ else:
+ volumesize = None
+
+ return _ISODataMedia(filepath, shortid, volumeid, publisherid, systemid,
+ applicationid, volumesize)
+
+def _get_isodatamedias(_os):
+ isodata_path = os.path.join(
+ os.path.dirname(os.path.realpath(__file__)),
+ 'isodata')
+ shortid_path = os.path.join(isodata_path, _os.distro, _os.shortid)
+
+ medias = []
+ if not os.path.exists(shortid_path):
+ return []
+
+ for _file in os.listdir(shortid_path):
+ path = os.path.join(shortid_path, _file)
+ if not os.path.exists(path):
+ continue
+
+ medias.append(_get_isodatamedia(path, _os.shortid))
+ return medias
--
2.20.1
_______________________________________________
Libosinfo mailing list
Libosinfo@xxxxxxxxxx
https://www.redhat.com/mailman/listinfo/libosinfo
