예제 #1
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Reminders(self.bot, db_file=":memory:")
     self.ctx = MockContext()
예제 #2
0
parser.add_argument('name',
                    metavar='NAME',
                    type=str,
                    nargs='?',
                    help='Name of the function to be run')
parser.add_argument('input',
                    metavar='FILE',
                    type=argparse.FileType('r'),
                    nargs='?',
                    default=sys.stdin,
                    help='File to get input from, or "-" for stdin')
args = parser.parse_args()

# Set up context
response = ''
context = MockContext.MockContext(args.name, '$LATEST')
logging.basicConfig(
    stream=sys.stderr,
    format=("[%(levelname)s]\t%(asctime)s\t{req}\t%(message)s").format(
        req=context.aws_request_id),
    datefmt="%Y-%m-%dT%H:%M:%S%Z",
    level=(logging.ERROR - args.verbose * 10))

try:
    # Run the function
    module = importlib.import_module('src.{name}.index'.format(name=args.name))
    event = common.get_payload(args.input,
                               context.identity.cognito_identity_id, '{}')
    event = json.loads(event)
    response = module.handler(event, context)
except Exception as exc:
예제 #3
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Commands(self.bot)
     self.ctx = MockContext()
예제 #4
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Attachments(self.bot, utils.full_path("sample_config"))
     self.ctx = MockContext()
예제 #5
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Notifications(self.bot, db_file=":memory:")
     self.ctx = MockContext()
예제 #6
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Greetings(self.bot, db_file=":memory:")
     self.ctx = MockContext()
예제 #7
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Reaction(self.bot)
     self.ctx = MockContext()
예제 #8
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = CustomCommands(self.bot, db_file=":memory:")
     self.ctx = MockContext()
예제 #9
0
 def setUp(self):
     self.bot = MockBot()
     self.cog = Logs(self.bot)
     self.ctx = MockContext()