import sys
import time
import uuid
+import distro
import platform
-
try:
import unittest2 as unittest
from unittest2 import SkipTest
answer = [None]*2 + [i[1] for i in expecting]
start = time.time()
- child = spawn(cmd, list(args), **spawn_opts)
+ child = spawn(cmd, list(args), encoding='utf-8', **spawn_opts)
if output:
child.logfile_read = output
"""Write `code` into `name`"""
path = os.path.join(self.meta, name)
data = code.encode('utf8') if isinstance(code, str) else code
- with open(path, 'w') as f:
+ with open(path, 'wb') as f:
f.write(data)
return path
def _psh(self, script, more_expecting=()):
- if (platform.linux_distribution()[0] == 'Fedora') and \
- (platform.linux_distribution()[1] == '20') and \
+ if (distro.linux_distribution()[0] == 'Fedora') and \
+ (distro.linux_distribution()[1] == '20') and \
(platform.architecture()[0] == '32bit'):
pat = SUDO_PASS_PROMPT_PATTERN_FEDORA_20_i586
else:
content = reader.read()
content = content.replace('\r', '\n').replace('\x00', '')
content = SHELL_COLOR_PATTERN.sub('', content)
- return content.decode('utf8', 'ignore')
+ return content
if hasattr(test, 'meta'):
content = get_log()
xml = ET.tostring(ts)
with open(self.xunit_file, 'w') as fp:
- fp.write(xml)
+ fp.write(xml.decode())
import os
import datetime
import platform
+import distro
import subprocess
from contextlib import contextmanager
"Fedora17-x86_64", "Ubuntu", "Ubuntu12.04", "Ubuntun12.10-i586".
'''
dist_name, dist_ver = \
- [i.strip() for i in platform.linux_distribution()[:2]]
+ [i.strip() for i in distro.linux_distribution()[:2]]
arch = platform.machine().strip()
return (dist_name,
arch,