def index():
if request.args.get('isbn') is not None:
books = Book.objects(isbn=request.args.get('isbn'))
elif request.args.get('title') is not None:
books = Book.objects(title=request.args.get('title'))
else:
books = Book.objects.all()
return books.to_json()
def test_invalid_data_entry(app):
invalid_book = Book(price="123",
published=datetime.datetime.utcnow(),
publisher="Test company")
with pytest.raises(ValidationError):
invalid_book.save()
invalid_customer = Customer(phone="123",
customer_since=datetime.datetime.utcnow(),
orders=[])
with pytest.raises(ValidationError):
invalid_customer.save()
def create_order(user_id, copies, book_id):
book = Book.objects(id=book_id).first()
customer = Customer.objects(id=user_id).first()
books = []
total = 0.0
while copies > 0:
books.append(book)
total = round(total + Book.objects.with_id(book_id).price, 2)
copies -= 1
"""checking to see if there's a order id already stored in session"""
if session.get("order_id"):
order = Order.objects(id=session.get("order_id")).first()
order.total_price = round(order.total_price + total, 2)
for book in books:
order.books.append(book)
order.save()
else:
order = Order(customer_name="{} {}".format(customer.first_name,
customer.last_name),
books=books,
shipping_address=customer.address,
total_price=total,
order_status="processing",
order_date=datetime.datetime.now).save()
session["order_id"] = str(order.id)
return order
def mongo_db_seed(db_name):
"""
This is a sample mongodb test for populating sample data into mongodb
:return: the db connection
"""
mongoengine.connect(db_name, host='localhost', port=27017)
fake = Faker()
for x in range(10):
Book(title=fake.sentence(nb_words=10,
variable_nb_words=True,
ext_word_list=None),
isbn=fake.isbn13(separator="-"),
author=fake.name(),
price=round(random.uniform(0, 100), 2),
published=datetime.datetime.utcnow(),
publisher=fake.company()).save()
for book in Book.objects:
Inventory(
book=book.id,
stock=random.randint(1, 100),
).save()
for x in range(10):
Customer(first_name=fake.first_name(),
last_name=fake.last_name(),
address=fake.address(),
email=fake.email(),
password='******',
phone=fake.phone_number(),
customer_since=datetime.datetime.utcnow(),
orders=[]).save()
for customer in Customer.objects:
for x in range(random.randint(1, 5)):
# the Book.objects.aggregate returns a dict. which is weird...was expecting an obj
books = [
book['_id'] for book in Book.objects.aggregate(*[{
"$sample": {
"size": random.randint(1, 3)
}
}])
]
total = 0.0
for book in books:
total = round(total + Book.objects.with_id(book).price)
order = Order(customer_name="{} {}".format(customer.first_name,
customer.last_name),
books=books,
shipping_address=customer.address,
total_price=total,
order_status="processing",
order_date=datetime.datetime.utcnow()).save()
customer.orders.append(order.id)
customer.save()
def test_valid_data_entry(app):
book_input = Book(title="Test book title",
isbn="Test12345",
author="Test author",
price="123",
published=datetime.datetime.utcnow(),
publisher="Test company")
book_input.save()
# verify book is successfully saved and able to be retrieved and their values match
book_saved = Book.objects(title=book_input.title).first()
assert book_saved.title == book_input.title
assert book_saved.isbn == book_input.isbn
assert book_saved.author == book_input.author
assert book_saved.price == book_input.price
assert book_saved.published.strftime(
"%m/%d/%Y, %H:%M:%S") == book_input.published.strftime(
"%m/%d/%Y, %H:%M:%S")
assert book_saved.publisher == book_input.publisher
# verify book can be retrieved by isbn
book_saved = Book.objects(isbn=book_input.isbn).first()
assert book_saved.title == book_input.title
assert book_saved.isbn == book_input.isbn
assert book_saved.author == book_input.author
assert book_saved.price == book_input.price
assert book_saved.published.strftime(
"%m/%d/%Y, %H:%M:%S") == book_input.published.strftime(
"%m/%d/%Y, %H:%M:%S")
assert book_saved.publisher == book_input.publisher
# verify the book can be ordered by a customer
sample_customer = Customer.objects.first()
order = Order(customer_name="{} {}".format(sample_customer.first_name,
sample_customer.last_name),
books=[book_saved.id],
shipping_address=sample_customer.address,
total_price=book_saved.price,
order_status="processing",
order_date=datetime.datetime.utcnow()).save()
sample_customer.orders.append(order.id)
sample_customer.save()
# Look for customer by order. Customer ID should match
order_customer = Customer.objects(orders=order.id).first()
assert order_customer.id == sample_customer.id
def create_order(copies, isbn):
book = Book.objects(isbn=isbn).get()
customer = Customer.objects(first_name="Kennth").get()