def test_no(self): self.assertEqual(parser.get_response('no'), parser.Response.No) self.assertEqual(parser.get_response('NO'), parser.Response.No) self.assertEqual(parser.get_response('n'), parser.Response.No) self.assertEqual(parser.get_response('false'), parser.Response.No) self.assertEqual(parser.get_response('f'), parser.Response.No) self.assertEqual(parser.get_response('disable'), parser.Response.No) self.assertEqual(parser.get_response('d'), parser.Response.No) self.assertEqual(parser.get_response('off'), parser.Response.No) self.assertEqual(parser.get_response('0'), parser.Response.No)
def test_yes(self): self.assertEqual(parser.get_response('yes'), parser.Response.Yes) self.assertEqual(parser.get_response('YES'), parser.Response.Yes) self.assertEqual(parser.get_response('y'), parser.Response.Yes) self.assertEqual(parser.get_response('true'), parser.Response.Yes) self.assertEqual(parser.get_response('t'), parser.Response.Yes) self.assertEqual(parser.get_response('enable'), parser.Response.Yes) self.assertEqual(parser.get_response('e'), parser.Response.Yes) self.assertEqual(parser.get_response('on'), parser.Response.Yes) self.assertEqual(parser.get_response('1'), parser.Response.Yes)
async def commandSetAutoPurge(args: ChatCommandArgs) -> bool: twitchUser: str = args.message.lower[1] nick: str = args.message[1] db: DatabaseMain cursor: aioodbc.cursor.Cursor async with DatabaseMain.acquire() as db, await db.cursor() as cursor: query: str = '''\ SELECT 1 FROM auto_purge WHERE broadcaster=? AND twitchUser=? ''' await cursor.execute(query, (args.chat.channel, twitchUser)) row: Optional[Tuple[int]] = await cursor.fetchone() if row is None: value: bool = False if len(args.message) >= 3: response: parser.Response value = bool( parser.get_response(args.message.lower[2], default=parser.Yes)) query = ''' INSERT INTO auto_purge (broadcaster, twitchUser, stopcommands) VALUES (?, ?, ?) ''' await cursor.execute(query, (args.chat.channel, twitchUser, value)) args.chat.send(f'Enabled Auto-Purge on {nick}') else: query = ''' DELETE FROM auto_purge WHERE broadcaster=? AND twitchUser=? ''' await cursor.execute(query, (args.chat.channel, twitchUser)) args.chat.send(f'Disabled Auto-Purge on {nick}') await db.commit() await library.reset_auto_purges(args.chat.channel, args.data) return True
async def feature(data: CacheStore, channel: str, message: Message, send: Send) -> bool: action: str = message.lower[2] if len(message) >= 3 else '' feature_: str = message.lower[1] features: Mapping[str, Optional[str]] = lib.items.feature.features() if feature_ not in features or features[feature_] is None: send('Unrecognized feature: ' + feature_) return True response: parser.Response = parser.get_response(action, default=parser.Yes) if response == parser.Yes: return await feature_add(data, channel, feature_, send) if response == parser.No: return await feature_remove(data, channel, feature_, send) send('Unrecognized second parameter: ' + action) return True
def test_unknown(self): self.assertEqual(parser.get_response('Kappa'), parser.Response.Unknown)
def test_default(self): self.assertEqual(parser.get_response('', default=parser.Response.Yes), parser.Response.Yes) self.assertEqual(parser.get_response('', default=parser.Response.No), parser.Response.No)
def test(self): self.assertEqual(parser.get_response(''), parser.Response.Unknown)