def test_get_an_owner(self):
#Test the api can get a given owners name.
self.owner = Owner(name="rakaz")
self.owner.save()
owner = Owner.objects.get(name= "rakaz")
response = self.client.get("/owners/{}/".format(owner.id))
self.assertEqual(200, response.status_code)
def test_update_owner(self):
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.patch("/owners/{}/".format(self.owner.id), {"name": "rakaz hadash"})
response_data = json.loads(response.content)
self.assertEqual(response_data,{'id': 1, 'name': 'rakaz hadash'})
self.assertEqual(response.status_code, 200)
def test_creat_new_calendar_for_specific_owner(self):
self.owner = Owner(name="rakaz")
self.owner.save()
self.calendar= Calendar(month="2020-07-23", owner=self.owner )
self.calendar.save()
print(self.calendar.id, self.owner.id)
response = self.client.post("/owners/{}/calendar/{}/".format(self.owner.id, self.calendar.id))
print(response.status_code)
def test_get_list_of_owners(self):
self.owner1 = Owner(name="rakaz.a")
self.owner1.save()
self.owner2 = Owner(name="rakaz.b")
self.owner2.save()
response = self.client.get("/owners/")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()["count"], 2)
# verify owner list includes expected owners:
self.assertEqual({"id": 2 ,"name": "rakaz.b"} in response.json()['results'], True)
def addowner():
form = AddForm()
if form.validate_on_submit():
name = form.name.data
pup_id = form.pup_id.data
puppy_to_assign = Puppy.query.get(pup_id)
# This if-statement checks if the Puppy exists.
if puppy_to_assign is not None:
# This checks if puppy already has an owner.
if puppy_to_assign.ownerid is None:
add_owner = Owner(name, pup_id)
db.session.add(add_owner)
db.session.commit()
flash("Successfully added new owner.", 'success')
return redirect(url_for('owners.listowner'))
else:
flash("Error: Cannot add owner. Puppy already has one.",
'danger')
return redirect(url_for('owners.addowner'))
else:
flash(
"Error: Cannot add new owner. Puppy ID provided does not exist.",
'danger')
return redirect(url_for('owners.addowner'))
return render_template('addowner.html', form=form)
def add_owner():
form = OwnerForm(request.form)
if form.validate():
new_owner = Owner(request.form["first_name"],
request.form["last_name"])
db.session.add(new_owner)
db.session.commit()
return redirect(url_for("owners.index"))
return render_template("new.html", form=form)
class OwnerTestCase(TestCase):
def setUp(self):
self.owner_name = "rakaz noar"
self.owner = Owner(name=self.owner_name)
super(OwnerTestCase, self).setUp()
# def test_owner_created(self):
# old_count = Owner.objects.count()
# self.owner.save()
# new_count = Owner.objects.count()
# self.assertNotEqual(old_count, new_count)
def test_api_can_get_an_owner(self):
#Test the api can get a given owners name.
self.owner.save()
owner = Owner.objects.get(name=self.owner_name)
response = self.client.get("/owners/{}".format(owner.id))
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertContains(response, owner)
class CalendarTestCase(APITestCase):
def setUp(self):
pass
def test_creat_new_calendar_for_specific_owner(self):
self.owner = Owner(name="rakaz")
self.owner.save()
self.calendar= Calendar(month="2020-07-23", owner=self.owner )
self.calendar.save()
print(self.calendar.id, self.owner.id)
response = self.client.post("/owners/{}/calendar/{}/".format(self.owner.id, self.calendar.id))
print(response.status_code)
#self.assertEqual(201, response.status_code)
def test_creating_new_calendar_for_no_owner_fails(self):
pass
def test_get_calendar(self):
pass
def add():
form = Addform()
if form.validate_on_submit():
name = form.name.data
puppy_id = form.puppy_id.data
new = Owner(name, puppy_id)
db.session.add(new)
db.session.commit()
return redirect(url_for('puppy.list'))
return render_template('add_owner.html', form=form)
def add():
form = AddForm()
if form.validate_on_submit():
name = form.name.data
task_id = form.task_id.data
new_owner = Owner(name, task_id)
db.session.add(new_owner)
db.session.commit()
return redirect(url_for('index'))
return render_template('add_owner.html', form=form)
def add():
form = AddForm()
if form.validate_on_submit():
id_puppy = form.id_puppy.data
owner = form.owner.data
if owner != "" and id_puppy != "":
puppy = Puppy.query.get(id_puppy)
if puppy != None and puppy.owner == None:
owner = Owner(owner, id_puppy)
db.session.add(owner)
db.session.commit()
return redirect(url_for('puppies.list'))
else:
return render_template(
"error.html", error="Puppy id you provided is not valid")
else:
return render_template(
"error.html", error="Please enter owner's name and puppy id.")
return render_template("addOwner.html", form=form)
def setUp(self):
self.owner_name = "rakaz noar"
self.owner = Owner(name=self.owner_name)
super(OwnerTestCase, self).setUp()
def test_owner_object_delete(self):
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.delete("/owners/{}/".format(self.owner.id))
self.assertEqual(204, response.status_code)
def test_create_owner(self):
#self.owner_name = "rakaz noar"
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.post("/owners/", {"name": "rakaz"})
self.assertEqual(201, response.status_code)
class OwnerTestCase(APITestCase):
def setUp(self):
pass
#self.username = "******"
#self.email = "*****@*****.**"
#self.password = "******"
#self.user = User.objects.create_user(self.username, self.email, self.password)
#self.token = Token.objects.create(user=self.user)
#self.api_authentication()
#def api_authentication(self):
#self.client.credentials(HTTP_AUTHORIZATION='Token ' + self.token.key)
def test_create_owner(self):
#self.owner_name = "rakaz noar"
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.post("/owners/", {"name": "rakaz"})
self.assertEqual(201, response.status_code)
def test_get_an_owner(self):
#Test the api can get a given owners name.
self.owner = Owner(name="rakaz")
self.owner.save()
owner = Owner.objects.get(name= "rakaz")
response = self.client.get("/owners/{}/".format(owner.id))
self.assertEqual(200, response.status_code)
def test_get_list_of_owners(self):
self.owner1 = Owner(name="rakaz.a")
self.owner1.save()
self.owner2 = Owner(name="rakaz.b")
self.owner2.save()
response = self.client.get("/owners/")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()["count"], 2)
# verify owner list includes expected owners:
self.assertEqual({"id": 2 ,"name": "rakaz.b"} in response.json()['results'], True)
def test_update_owner(self):
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.patch("/owners/{}/".format(self.owner.id), {"name": "rakaz hadash"})
response_data = json.loads(response.content)
self.assertEqual(response_data,{'id': 1, 'name': 'rakaz hadash'})
self.assertEqual(response.status_code, 200)
def test_owner_object_delete(self):
self.owner = Owner(name="rakaz")
self.owner.save()
response = self.client.delete("/owners/{}/".format(self.owner.id))
self.assertEqual(204, response.status_code)