def test_main_function(self, mocked_square, mocked_cube):
# underling function are mocks so calling main(5) will return mock
mocked_square.return_value = 1
mocked_cube.return_value = 0
self.assertEquals(main(5), 1)
mocked_square.assert_called_once_with(5)
mocked_cube.assert_called_once_with(5)
def test_main_function(self, mocked_square, mocked_cube):
# underling function are mocks so calling main(5) will return mock
mocked_square.return_value = 1
mocked_cube.return_value = 0
self.assertEquals(main(5), 1)
mocked_square.assert_called_once_with(5)
mocked_cube.assert_called_once_with(5)
def event_handle(event):
print(event)
try:
userId = event['source']['userId']
except:
print('error cannot get userId')
return ''
try:
rtoken = event['replyToken']
except:
print('error cannot get rtoken')
return ''
try:
msgId = event["message"]["id"]
msgType = event["message"]["type"]
except:
print('error cannot get msgID, and msgType')
sk_id = np.random.randint(1, 17)
replyObj = StickerSendMessage(package_id=str(1), sticker_id=str(sk_id))
line_bot_api.reply_message(rtoken, replyObj)
return ''
try:
if msgType == "text":
msg = str(event["message"]["text"])
function.main(msg)
replyObj = TextSendMessage(text='เรียบร้อยครับ')
line_bot_api.reply_message(rtoken, replyObj)
else:
sk_id = np.random.randint(1, 17)
replyObj = StickerSendMessage(package_id=str(1),
sticker_id=str(sk_id))
line_bot_api.reply_message(rtoken, replyObj)
except:
print('error cannot reply msg')
replyObj_sk = StickerSendMessage(package_id=str(11537),
sticker_id=str(52002755))
line_bot_api.reply_message(rtoken, replyObj_sk)
replyObj_txt = TextSendMessage(text='ระบบขัดข้อง')
line_bot_api.reply_message(rtoken, replyObj_txt)
return ''
def test_output_contacts():
"""Test function."""
# Construct a mock HTTP request.
req = func.HttpRequest(method="GET",
body=None,
url="/api/HttpTrigger",
params={"name": "Test"})
# Call the function.
resp = function.main(req)
# Check the output.
assert resp.get_body().decode() == "Hello Test"
def test_main_function(monkeypatch):
monkeypatch.setattr('function.square', lambda x: 1)
monkeypatch.setattr('function.cube', lambda x: 0)
assert main(5) == 1
# -*- coding: utf-8 -*- from Config import config from function import main num = 6 flag = config.True_Small # config.True_Small # True: 只测试小数据 main(num, flag=flag)
def test_main_function(monkeypatch):
monkeypatch.setattr('function.square', lambda x: 1)
monkeypatch.setattr('function.cube', lambda x: 0)
assert main(5) == 1