def config(self, mock_sessionmaker, mock_create_engine): self.mock_sessionmaker = mock_sessionmaker self.mock_create_engine = mock_create_engine self.mock_create_engine.return_value = "engine" self.mock_session = mock.Mock() self.mock_session.return_value = "session" self.mock_sessionmaker.return_value = self.mock_session self.session = Session()
def main(): parser = argparse.ArgumentParser(description='Delete ong in the system.') parser.add_argument('name', help='Name of ong') args = parser.parse_args() name = args.name session = Session() o = Ong(session=session.get_session(), name=name) o.delete()
def main(): parser = argparse.ArgumentParser(description='Add new ong in the system.') parser.add_argument('name', help='Name of new ong') parser.add_argument('-H', '--homepage', metavar='URL', nargs='?', help='URL for ong homepage', default='#') parser.add_argument('-N', '--completename', metavar='CNAME', nargs='?', help='Complete name of new ong') args = parser.parse_args() name = args.name session = Session() o = Ong(session=session.get_session(), name=name) return o.save()