PsychoticNinja/JLU5-SupyPlugins
1
0
Fork 0
mirror of https://github.com/jlu5/SupyPlugins.git synced 2025-05-06 11:33:41 -05:00
Code Issues Projects Releases Wiki Activity

Grapnel: add tests (DB and HTTP POST)

Browse Source
This commit is contained in:
James Lu 2022-07-25 23:11:03 -07:00
parent 393e7fea01
commit acf23329fc
1 changed files with 152 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

154
Grapnel/test.py
View File

@ -28,11 +28,161 @@
### ###
from supybot.test import * import io
import json
import re
import urllib.parse
from supybot import conf
from supybot.test import ChannelHTTPPluginTestCase, FakeHTTPConnection, TestRequestHandler
class GrapnelTestCase(PluginTestCase): TEST_BASEURL = 'http://grapnel.test'
class GrapnelTestCase(ChannelHTTPPluginTestCase):
plugins = ('Grapnel',) plugins = ('Grapnel',)
config = {
'plugins.grapnel.baseurl': TEST_BASEURL
}
timeout = 10
# Like HTTPPluginTestCase.request, but with JSON Content-Type header and data
def jsonPost(self, url, json_data):
assert url.startswith('/')
wfile = io.BytesIO()
rfile = io.BytesIO()
# wfile and rfile are reversed in Limnoria's HTTPPluginTestCase.request too?
connection = FakeHTTPConnection(wfile, rfile)
headers = {'Content-Type': 'application/json'}
connection.request('POST', url, json_data, headers)
rfile.seek(0)
handler = TestRequestHandler(rfile, wfile)
wfile.seek(0)
return (handler._response, wfile.read()) # pylint: disable=protected-access
# def setUp(self):
# super().setUp()
# self.myVerbose = verbosity.MESSAGES
def _getLinkFragment(self, m):
url_re = re.search(fr'{TEST_BASEURL}(\/grapnel\/\d+\?.*)', m.args[1])
url_fragment = url_re.group(1)
assert url_fragment
return url_fragment
def _addHook(self, channel='#test'):
m = self.assertRegexp(f"grapnel add {channel}", fr"Webhook #\d+ created.*{TEST_BASEURL}/grapnel")
return self._getLinkFragment(m)
def testPOSTSuccess(self):
url_fragment = self._addHook()
(respCode, body) = self.jsonPost(url_fragment, json.dumps({"text": "123456"}))
self.assertEqual(respCode, 200, body.decode())
self.assertSnarfRegexp(' ', r'\[.*?\] 123456')
def testPOSTCustomFormat(self):
url_fragment = self._addHook()
sp = urllib.parse.urlsplit(url_fragment)
sp_query = urllib.parse.parse_qs(sp.query)
#print("OLD", url_fragment, sp_query)
url_fragment = sp.path + '?' + urllib.parse.urlencode({
'token': sp_query['token'][0], 'sender': 'foobar'})
#print("NEW", url_fragment)
with conf.supybot.plugins.grapnel.format.context("Limnoria says: [$name] $text"):
(respCode, body) = self.jsonPost(url_fragment, json.dumps({"text": "123456"}))
self.assertEqual(respCode, 200, body.decode())
self.assertSnarfResponse(' ', 'Limnoria says: [foobar] 123456')
def testPOSTInvalidJSON(self):
url_fragment = self._addHook()
(respCode, body) = self.jsonPost(url_fragment, "<html></html>")
self.assertEqual(respCode, 400, body.decode())
self.assertSnarfNoResponse(' ')
def testPOSTMissingToken(self):
url_fragment = self._addHook()
sp = urllib.parse.urlsplit(url_fragment)
url_fragment = sp.path
(respCode, body) = self.jsonPost(url_fragment, json.dumps({"text": "hello world"}))
self.assertEqual(respCode, 401, body.decode())
self.assertSnarfNoResponse(' ')
def testPOSTWrongToken(self):
url_fragment = self._addHook()
# parse the URL and change the token
sp = urllib.parse.urlsplit(url_fragment)
url_fragment = sp.path + '?' + urllib.parse.urlencode({'token': 'obvious-incorrect'})
(respCode, body) = self.jsonPost(url_fragment, json.dumps({"text": "hello world"}))
self.assertEqual(respCode, 403, body.decode())
self.assertSnarfNoResponse(' ')
def testPOSTUnknownHookID(self):
url_fragment = self._addHook()
# parse the URL and change the path
sp = urllib.parse.urlsplit(url_fragment)
url_fragment = '/grapnel/12345?' + sp.query
(respCode, body) = self.jsonPost(url_fragment, json.dumps({"text": "hello world"}))
self.assertEqual(respCode, 404, body.decode())
self.assertSnarfNoResponse(' ')
def testPOSTBadContentType(self):
url_fragment = self._addHook()
(respCode, body) = self.request(url_fragment, method='POST')
self.assertEqual(respCode, 400, body.decode())
self.assertSnarfNoResponse(' ')
def testGet(self):
url_fragment_1 = self._addHook()
url_fragment_2 = self._addHook(channel='#limnoria')
self.assertRegexp('grapnel get 1', fr'Webhook #1 for #test@test.*{re.escape(url_fragment_1)}')
self.assertRegexp('grapnel get 2', fr'Webhook #2 for #limnoria@test.*{re.escape(url_fragment_2)}')
def testRemove(self):
url_fragment_1 = self._addHook(channel='#bots')
url_fragment_2 = self._addHook(channel='#bots')
url_fragment_3 = self._addHook(channel='#bots')
self.assertNotError('grapnel remove 2')
self.assertRegexp('grapnel listhooks #bots', '#1 #3$')
# check that the removed hook errors
(respCode, body) = self.jsonPost(url_fragment_2, json.dumps({"text": "this goes nowhere"}))
self.assertEqual(respCode, 404, body.decode())
self.assertSnarfNoResponse(' ')
# can't remove it twice
self.assertError('grapnel remove 2')
def testListAndCaseNormalization(self):
url_fragment_1 = self._addHook(channel='#dev')
url_fragment_2 = self._addHook(channel='#Dev')
url_fragment_3 = self._addHook(channel='#limnoria')
self.assertResponse('grapnel listhooks #dev', 'Webhooks for #dev@test: #1 #2')
self.assertResponse('grapnel listhooks #DEV', 'Webhooks for #dev@test: #1 #2')
self.assertResponse('grapnel listhooks #Limnoria', 'Webhooks for #limnoria@test: #3')
self.assertRegexp('grapnel listhooks #', r'\#\@test.*\(none\)')
def testResetTokenWorks(self):
url_fragment_old = self._addHook()
m = self.assertRegexp("grapnel resettoken 1", fr"{TEST_BASEURL}/grapnel")
url_fragment_new = self._getLinkFragment(m)
# check that they're the same hook ID
self.assertNotEqual(url_fragment_old, url_fragment_new, "Webhook URLs should be different")
self.assertIn("grapnel/1?", url_fragment_old)
self.assertIn("grapnel/1?", url_fragment_new)
# old token gives 403
(respCode, body) = self.jsonPost(url_fragment_old, json.dumps({"text": "hello world"}))
self.assertEqual(respCode, 403, body.decode())
self.assertSnarfNoResponse(' ')
# new token is OK
(respCode, body) = self.jsonPost(url_fragment_new, json.dumps({"text": "the game"}))
self.assertEqual(respCode, 200, body.decode())
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: # vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: