def test_run_via_main(self): """Tests running scriptable via 'main' method""" test_str = 'one\ntwo\n' self.stdin.write(test_str) self.stdin.seek(0) scriptable.main(['-test-', 'test_support.echo']) self.stdout.seek(0) self.assertEquals(self.stdout.read(), test_str)
def test_usage(self): """Test the 'main' method output usage information""" argv = ['-test'] ret = scriptable.main(argv) self.assertEquals(ret, 1) self.stdout.seek(0) self.assertIn('Usage', self.stdout.read())