def test1_get_osinfo_from_lsb_release(self): with mock.patch( '__builtin__.open', mock.mock_open(), create=True) as m: m.side_effect = IOError() result = get_osinfo_from_lsb_release() self.assertFalse(result)
def test_get_osinfo_from_lsb_release(self): data = ['DISTRIB_ID=Ubuntu', 'DISTRIB_RELEASE=15.10', 'DISTRIB_CODENAME=wily' 'DISTRIB_DESCRIPTION="Ubuntu 15.10"'] 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_lsb_release() self.assertEqual(result['os'], 'ubuntu') self.assertEqual(result['version'], '15.10')