def test1_get_osinfo_from_os_release(self): with mock.patch( '__builtin__.open', mock.mock_open(), create=True) as m: m.side_effect = IOError() result = get_osinfo_from_os_release() self.assertFalse(result)
def test_get_osinfo_from_os_release(self): data = ['NAME="Ubuntu"', 'VERSION="14.04.4 LTS, Trusty Tahr"', 'ID=ubuntu', 'ID_LIKE=debian', 'PRETTY_NAME="Ubuntu 14.04.4 LTS"', 'VERSION_ID="14.04"', 'HOME_URL="http://www.ubuntu.com/"', 'SUPPORT_URL="http://help.ubuntu.com/"', 'BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"' ] with mock.patch( '__builtin__.open', mock.mock_open(read_data="\n".join(data)), create=True) as m: m.return_value.__iter__.return_value = data result = get_osinfo_from_os_release() self.assertEqual(result['os'], 'ubuntu') self.assertEqual(result['version'], '14.04')