def add_item(request):
if (request.method == "POST"):
title = request.POST["title"]
priority = request.POST["priority"]
item = Item(title=title, priority=priority)
item.save()
return redirect("/")
def forge(count: int):
"""Generate fake message."""
from faker import Faker
import random
fake = Faker()
db.drop_all()
db.create_all()
click.echo("Generating a User...")
user = User(username="******")
user.set_password("123")
for i in range(count):
click.echo("Creating the item...")
item = Item(
user=user,
title=fake.sentence(),
status=random.randint(0, 1),
create_time=fake.date_time_this_year(),
)
db.session.add(item)
db.session.commit()
click.echo(f"Created {count} fake item.")
def test_save_and_retrieve_item(db):
first_item = Item()
first_item.text = "Hello Der how are you ?"
first_item.save()
second_item = Item()
second_item.text = "What the f**k is happening ?"
second_item.save()
items = Item.objects.all()
assert items.count() == 2
first_saved_item = items[0]
second_saved_item = items[1]
assert first_saved_item.text == 'Hello Der how are you ?'
assert second_saved_item.text == 'What the f**k is happening ?'
def new_item():
data = request.get_json()
if data is None or data['body'].strip() == '':
jsonify(message=_('Invalid item body.')), 400
item = Item(body=data['body'], author=current_user._get_current_object())
db.session.add(item)
db.session.commit()
return jsonify(html=render_template('_item.html', item=item), message='+1')
def post(self):
""" Create new item. """
item = Item(body=get_item_body(), author=g.current_user)
db.session.add(item)
db.session.commit()
response = jsonify(item_schema(item))
response.status_code = 201
response.headers['Location'] = url_for('.item',
item_id=item.id,
_external=True)
return response
def register():
username = fake.user_name()
while User.query.filter_by(username=username).first() is not None:
username = fake.user_name()
password = fake.word()
user = User(username=username)
user.set_password(password)
db.session.add(user)
db.session.commit()
item = Item(body=_('Witness something truly majestic'), author=user)
item2 = Item(body=_('Help a complete stranger'), author=user)
item3 = Item(body=_('Drive a motorcycle on the Great Wall of China'),
author=user)
item4 = Item(body=_('Sit on the Great Egyptian Pyramids'),
done=True,
author=user)
db.session.add_all([item, item2, item3, item4])
db.session.commit()
return jsonify(username=username,
password=password,
message=_('Generate success.'))
def client():
app.config.update(
TESTING=True,
WTF_CSRF_ENABLED=False,
SQLALCHEMY_DATABASE_URI="sqlite:///:memory:",
)
user = User(username="******")
user.set_password("123")
item = Item(title="hello flask", status=1)
with app.test_client() as client:
with app.test_request_context():
db.create_all()
db.session.add_all([user, item])
db.session.commit()
yield client
db.drop_all(app=app)
def test_default_text(self):
item = Item()
self.assertEqual(item.text, '')
def test_string_representation(self):
item = Item(text='some text')
self.assertEqual(str(item), 'some text')
def test_CAN_save_same_item_to_different_lists(self):
list1 = List.objects.create()
list2 = List.objects.create()
Item.objects.create(list=list1, text='bla')
item = Item(list=list2, text='bla')
item.full_clean()
def test_duplicate_items_are_invalid(self):
list_ = List.objects.create()
Item.objects.create(list=list_, text='bla')
with self.assertRaises(ValidationError):
item = Item(list=list_, text='bla')
item.full_clean()
def test_cannot_save_empty_list_items(self):
list_ = List.objects.create()
item = Item(list=list_, text='')
with self.assertRaises(ValidationError):
item.save()
item.full_clean()
def test_item_is_related_to_list(self):
list_ = List.objects.create()
item = Item()
item.list = list_
item.save()
self.assertIn(item, list_.item_set.all())
def test_item_is_related_to_list(self):
list_ = List.objects.create()
item = Item()
item.list = list_
item.save()
self.assertIn(item, list_.item_set.all())
def add(request,content):
newItem = Item(content=content,done=0)
newItem.save()
print(newItem.id)
return HttpResponse(json.dumps({"code": 1,"id":newItem.id}), content_type='application/json')
def test_cannot_save_empty_list_items(self):
list_ = List.objects.create()
item = Item(list=list_, text="")
with self.assertRaises(ValidationError):
item.save()
item.full_clean()
def test_CAN_save_same_item_to_different_lists(self):
list1 = List.objects.create()
list2 = List.objects.create()
Item.objects.create(list=list1, text="bla")
item = Item(list=list2, text="bla")
item.full_clean()
def test_duplicate_items_are_invalid(self):
list_ = List.objects.create()
Item.objects.create(list=list_, text="bla")
with self.assertRaises(ValidationError):
item = Item(list=list_, text="bla")
item.full_clean()
