def _create_sample_file(self, *lines): """General abstraction to be used with the -f commands. Keep in mind that you need to delete the created file, just to be cleaner. """ if not lines: lines = (self.sample_mac, ) return _test_utils.create_sample_file(*lines)
def test__split_file(self): separators = ['|', '%', '!', '*', ':', ',', '\n'] for separator in separators: lines = [] for letter in string.ascii_letters: line = letter * 10 lines.append(line) fpath = _test_utils.create_sample_file(separator.join(lines)) try: self.assertEqual(utils._split_file(fpath, separator), lines) finally: os.remove(fpath)
def test_fetch_macs_from_file(self): slines = ['11.11.11.11.11 # one', '22.11.11.11.11 # two', 'Invalid mac # thats ok', '# line with comment'] expmacs = ['11.11.11.11.11', '22.11.11.11.11', 'Invalid mac'] separators = [':', '*', '!', '\n'] for sep in separators: sfile = _test_utils.create_sample_file(sep.join(slines)) try: for mac in utils.fetch_macs_from_file(sfile, '\n'): self.assertIn(mac, expmacs) finally: os.remove(sfile)