def test_bot_create_empty(bot, gmi):
from lowerpines.endpoints.group import Group
group = Group(gmi, name='serializer test group')
group.save()
fake_request = Request(HttpRequest())
fake_request.user = bot.owner
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert serializer.is_valid()
new_bot = serializer.save(owner=bot.owner)
gmi_bot = gmi.bots.get(name='test')
assert new_bot.name == 'test'
assert new_bot.slug == 'floop'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 0
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/floop/callback/'
assert gmi_bot.name == 'test'
def test_bot_view_delete(monkeypatch, gmi, client):
monkeypatch.setattr('saucerbot.groupme.models.get_gmi', lambda a: gmi)
u = gmi.user.get()
u.user_id = 'abcdef'
u.save()
# login first
client.get('/groupme/oauth/?access_token=faketoken')
from lowerpines.endpoints.group import Group
from lowerpines.exceptions import NoneFoundException
from saucerbot.groupme.models import Bot
group = Group(gmi, name='view test group')
group.save()
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': ['system_messages']
}
resp = client.post('/groupme/api/bots/',
content_type='application/json',
data=data)
assert resp.status_code == 201
new_bot = Bot.objects.get(slug='floop')
gmi_bot = gmi.bots.get(name='test')
assert new_bot.name == 'test'
assert new_bot.slug == 'floop'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 1
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/floop/callback/'
assert gmi_bot.name == 'test'
resp = client.delete('/groupme/api/bots/floop/')
assert resp.status_code == 204
# The db bot is gone
with pytest.raises(Bot.DoesNotExist):
Bot.objects.get(slug='floop')
# The groupme bot is gone too
with pytest.raises(NoneFoundException):
gmi.bots.get(name='test')
def test_bot_failed_update(bot, gmi):
from lowerpines.endpoints.group import Group
group = Group(gmi, name='serializer test group')
group.save()
group2 = Group(gmi, name='serializer test group2')
group2.save()
fake_request = Request(HttpRequest())
fake_request.user = bot.owner
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': ['handler_test1', 'handler_test2']
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert serializer.is_valid()
new_bot = serializer.save(owner=bot.owner)
gmi_bot = gmi.bots.get(name='test')
new_bot_handlers = set(h.handler_name for h in new_bot.handlers.all())
assert new_bot.name == 'test'
assert new_bot.slug == 'floop'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 2
assert new_bot_handlers == {'handler_test1', 'handler_test2'}
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/floop/callback/'
assert gmi_bot.name == 'test'
data = {
'name': 'new_test',
'slug': 'new_floop',
'group': group2.group_id,
'handlers': ['handler_test1', 'handler_test3', 'handler_test4']
}
serializer = BotSerializer(new_bot, data=data, partial=False, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'group'}
assert 'may not be changed' in serializer.errors['group'][0]
def test_bot_create_invalid(bot, gmi):
from lowerpines.endpoints.group import Group
group = Group(gmi, name='serializer test group')
group.save()
fake_request = Request(HttpRequest())
fake_request.user = bot.owner
serializer = BotSerializer(data={}, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'name', 'group'}
serializer = BotSerializer(data={'name': 'test'}, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'group'}
serializer = BotSerializer(data={'slug': 'floop'}, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'name', 'group'}
serializer = BotSerializer(data={'group': group.group_id}, context={'request': fake_request})
print(Group.get_all(gmi))
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'name'}
serializer = BotSerializer(data={'name': 'test', 'slug': 'floop'},
context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'group'}
serializer = BotSerializer(data={'slug': 'floop', 'group': group.group_id},
context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'name'}
def setup_bot(db, gmi, monkeypatch):
"""
Create a bot for saucerbot tests
"""
monkeypatch.setattr('saucerbot.groupme.models.get_gmi', lambda a: gmi)
from lowerpines.group import Group
from saucerbot.groupme.models import User, Bot
user = User.objects.create(access_token='123456', user_id='123456')
group = Group(user.gmi, name='test group')
group.save()
bot = Bot.objects.create(owner=user,
group=group,
name='saucerbot',
slug='saucerbot')
return bot
def test_bot_update(bot, gmi):
from lowerpines.endpoints.group import Group
from lowerpines.exceptions import NoneFoundException
group = Group(gmi, name='serializer test group')
group.save()
fake_request = Request(HttpRequest())
fake_request.user = bot.owner
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': ['handler_test1', 'handler_test2']
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert serializer.is_valid()
new_bot = serializer.save(owner=bot.owner)
gmi_bot = gmi.bots.get(name='test')
new_bot_handlers = set(h.handler_name for h in new_bot.handlers.all())
assert new_bot.name == 'test'
assert new_bot.slug == 'floop'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 2
assert new_bot_handlers == {'handler_test1', 'handler_test2'}
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/floop/callback/'
assert gmi_bot.name == 'test'
data = {
'name': 'new_test',
'slug': 'new_floop',
'group': group.group_id,
'handlers': ['handler_test1', 'handler_test3', 'handler_test4']
}
serializer = BotSerializer(new_bot, data=data, partial=False, context={'request': fake_request})
assert serializer.is_valid()
updated_bot = serializer.save()
updated_bot_handlers = set(h.handler_name for h in updated_bot.handlers.all())
assert updated_bot.name == 'new_test'
assert updated_bot.slug == 'new_floop'
assert updated_bot.bot_id == gmi_bot.bot_id
assert updated_bot.group_id == group.group_id
assert updated_bot.handlers.count() == 3
assert updated_bot_handlers == {'handler_test1', 'handler_test3', 'handler_test4'}
# The gmi bot got fixed too
gmi_bot = gmi.bots.get(name='new_test')
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/new_floop/callback/'
assert gmi_bot.name == 'new_test'
# The old bot is gone
with pytest.raises(NoneFoundException):
gmi.bots.get(name='test')
def test_bot_create(bot, gmi):
from lowerpines.endpoints.group import Group
group = Group(gmi, name='serializer test group')
group.save()
fake_request = Request(HttpRequest())
fake_request.user = bot.owner
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': None
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'handlers'}
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': ['fok']
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert not serializer.is_valid()
assert set(serializer.errors.keys()) == {'handlers'}
data = {
'name': 'test',
'slug': 'floop',
'group': group.group_id,
'handlers': ['handler_test1']
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert serializer.is_valid()
new_bot = serializer.save(owner=bot.owner)
gmi_bot = gmi.bots.get(name='test')
assert new_bot.name == 'test'
assert new_bot.slug == 'floop'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 1
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/floop/callback/'
assert gmi_bot.name == 'test'
data = {
'name': 'Test Bot 43',
'group': group.group_id,
'handlers': ['handler_test1']
}
serializer = BotSerializer(data=data, context={'request': fake_request})
assert serializer.is_valid()
new_bot = serializer.save(owner=bot.owner)
gmi_bot = gmi.bots.get(name='Test Bot 43')
assert new_bot.name == 'Test Bot 43'
assert new_bot.slug == 'test-bot-43'
assert new_bot.bot_id == gmi_bot.bot_id
assert new_bot.group_id == group.group_id
assert new_bot.handlers.count() == 1
assert gmi_bot.callback_url == 'https://localhost/groupme/api/bots/test-bot-43/callback/'
assert gmi_bot.name == 'Test Bot 43'
test_bot = None
test_group = None
if allow_non_deterministic:
TEST_GROUP_NAME = "TestGroup"
try:
test_group = gmi.groups.get(name=TEST_GROUP_NAME)
except NoneFoundException:
test_group = Group(
gmi,
name=TEST_GROUP_NAME,
description="Here is the description",
image_url=
"https://i.groupme.com/750x700.jpeg.bda7c13e72f00b58193bd6af2114cb24c3919d1a",
)
test_group.save()
TEST_BOT_NAME = "TestBot"
try:
test_bot = test_group.bots.get(name=TEST_BOT_NAME)
except NoneFoundException:
test_bot = Bot(
gmi,
group_id=test_group.group_id,
name=TEST_BOT_NAME,
callback_url="http://example.com",
avatar_url=
"https://i.groupme.com/750x700.jpeg.bda7c13e72f00b58193bd6af2114cb24c3919d1a",
dm_notification=False,
)
test_bot.save()