class IntsCTestCase(unittest.TestCase):
def setUp(self):
self.pass_ = IntsPass('c')
def test_c(self):
with tempfile.NamedTemporaryFile(mode='w', delete=False) as tmp_file:
tmp_file.write('Compute 123L + 0x456 + 0789!\n')
state = self.pass_.new(tmp_file.name)
(_, state) = self.pass_.transform(tmp_file.name, state, None)
with open(tmp_file.name) as variant_file:
variant = variant_file.read()
os.unlink(tmp_file.name)
self.assertEqual(variant, 'Compute 123 + 0x456 + 0789!\n')
def setUp(self):
self.pass_ = IntsPass('c')
class IntsATestCase(unittest.TestCase):
def setUp(self):
self.pass_ = IntsPass('a')
def test_a(self):
with tempfile.NamedTemporaryFile(mode='w', delete=False) as tmp_file:
tmp_file.write('Compute 123L + 0x456 + 0789!\n')
state = self.pass_.new(tmp_file.name)
(_, state) = self.pass_.transform(tmp_file.name, state, None)
with open(tmp_file.name) as variant_file:
variant = variant_file.read()
os.unlink(tmp_file.name)
self.assertEqual(variant, 'Compute 123L + 0x56 + 0789!\n')
def test_success_a(self):
with tempfile.NamedTemporaryFile(mode='w', delete=False) as tmp_file:
tmp_file.write('Compute 123L + 0x456 + 0789!\n')
state = self.pass_.new(tmp_file.name)
(result, state) = self.pass_.transform(tmp_file.name, state, None)
iteration = 1
while result == PassResult.OK and iteration < 10:
state = self.pass_.advance_on_success(tmp_file.name, state)
if state is None:
break
(result, state) = self.pass_.transform(tmp_file.name, state, None)
iteration += 1
with open(tmp_file.name) as variant_file:
variant = variant_file.read()
os.unlink(tmp_file.name)
self.assertEqual(iteration, 4)
self.assertEqual(variant, 'Compute 3L + 0x6 + 0789!\n')
def test_no_success_a(self):
with tempfile.NamedTemporaryFile(mode='w', delete=False) as tmp_file:
tmp_file.write('Compute 123L + 0x456 + 0789!\n')
state = self.pass_.new(tmp_file.name)
(result, state) = self.pass_.transform(tmp_file.name, state, None)
iteration = 1
while result == PassResult.OK and iteration < 10:
with tempfile.NamedTemporaryFile(mode='w',
delete=False) as tmp_file:
tmp_file.write('Compute 123L + 0x456 + 0789!\n')
state = self.pass_.advance(tmp_file.name, state)
if state is None:
break
(result, state) = self.pass_.transform(tmp_file.name, state, None)
iteration += 1
os.unlink(tmp_file.name)
self.assertEqual(iteration, 2)