diff --git a/tests/pylorax/test_discinfo.py b/tests/pylorax/test_discinfo.py new file mode 100644 index 00000000..3a326e9b --- /dev/null +++ b/tests/pylorax/test_discinfo.py @@ -0,0 +1,30 @@ +# +# Copyright (C) 2018 Red Hat, Inc. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . +# +import unittest +import tempfile + +from pylorax.discinfo import DiscInfo + +class DiscInfoTest(unittest.TestCase): + def discinfo_test(self): + with tempfile.NamedTemporaryFile(mode="w+t") as f: + di = DiscInfo("1.0", "x86_64") + di.write(f.name) + f.seek(0) + self.assertTrue(f.readline().strip() not in ["", None]) + self.assertEqual(f.readline().strip(), "1.0") + self.assertEqual(f.readline().strip(), "x86_64")