def test_delete(self):
# Requires login:
res = self.test_client.post("/goals/delete/", data={'id': 0})
setup.assertRequiresLogin(self, res)
self.login()
## Validates form:
res = self.test_client.post("/goals/delete/", data={'foo': 'bar'})
setup.assertInvalid(self, res, "id")
## works at all
new_goal = self.create_test_goal()
res = self.test_client.post("/goals/delete/", data={'id': new_goal['id']})
setup.assertOk(self, res, 204)
## 404s
res = self.test_client.post("/goals/delete/", data={'id': new_goal['id']})
setup.assert404(self, res)
## Can't someone *else*'s goal.
new_goal = self.create_test_goal()
self.logout()
self.login_other_user()
res = self.test_client.post("/goals/delete/", data={'id': new_goal['id']})
setup.assertInvalidCredentials(self, res)
def test_update(self):
tc = self.test_client
res = tc.post('/register/', data=self.new_user)
setup.assertOk(self, res, 201)
user = json.loads(res.data)
post_data = {"password": self.new_user['password'], "first_name": "new-first-name"}
res = tc.post('/users/update/', data=post_data)
setup.assertInvalid(self, res, 'email')
post_data['email'] = "not-real-email"
res = tc.post('/users/update/', data=post_data)
setup.assertInvalidCredentials(self, res)
post_data['email'] = self.new_user['email']
post_data['password'] = '******'
res = tc.post('/users/update/', data=post_data)
setup.assertInvalidCredentials(self, res)
post_data['password'] = self.new_user['password']
res = tc.post('/users/update/', data=post_data)
setup.assertOk(self, res)
res = tc.post('/login/', data={'email': self.new_user['email'], 'password': self.new_user['password']})
setup.assertOk(self, res, 201)
res = tc.get('/users/me/')
setup.assertOk(self, res)
updated_user = json.loads(res.data)
for key, value in user.iteritems():
if (key == 'first_name'): self.assertEqual(updated_user[key], 'new-first-name')
else: self.assertEqual(updated_user[key], user[key])
def test_delete(self):
# Requires login:
res = self.test_client.post("/goals/delete/", data={'id': 0})
setup.assertRequiresLogin(self, res)
self.login()
## Validates form:
res = self.test_client.post("/goals/delete/", data={'foo': 'bar'})
setup.assertInvalid(self, res, "id")
## works at all
new_goal = self.create_test_goal()
res = self.test_client.post("/goals/delete/",
data={'id': new_goal['id']})
setup.assertOk(self, res, 204)
## 404s
res = self.test_client.post("/goals/delete/",
data={'id': new_goal['id']})
setup.assert404(self, res)
## Can't someone *else*'s goal.
new_goal = self.create_test_goal()
self.logout()
self.login_other_user()
res = self.test_client.post("/goals/delete/",
data={'id': new_goal['id']})
setup.assertInvalidCredentials(self, res)
def test_get(self):
tc = self.test_client
## Login required
setup.assertRequiresLogin(self, tc.get("/goals/0/check-ins/0/"))
## Goal exists
self.login()
setup.assert404(self, tc.get("/goals/0/check-ins/0/"))
## user is goal owner
goal = self.create_test_numeric_goal()
goal_id = str(goal.get_id())
self.logout()
self.login_other_user()
setup.assertInvalidCredentials(self, tc.get("/goals/" + goal_id + "/check-ins/0/"))
## Check in must exist
self.logout()
self.login()
setup.assert404(self, tc.get("/goals/" + goal_id + "/check-ins/0/"))
## Returns correctly
ci = CheckIn(goal, Timeframe.get_current_timeframe(goal.check_in_frequency_name), 1)
ci.persist()
tfid = str(ci.timeframe)
res = tc.get("/goals/" + goal_id + "/check-ins/" + tfid + "/")
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), ci.to_dict())
def test_get(self):
tc = self.test_client
## Login required
setup.assertRequiresLogin(self, tc.get("/goals/0/check-ins/0/"))
## Goal exists
self.login()
setup.assert404(self, tc.get("/goals/0/check-ins/0/"))
## user is goal owner
goal = self.create_test_numeric_goal()
goal_id = str(goal.get_id())
self.logout()
self.login_other_user()
setup.assertInvalidCredentials(
self, tc.get("/goals/" + goal_id + "/check-ins/0/"))
## Check in must exist
self.logout()
self.login()
setup.assert404(self, tc.get("/goals/" + goal_id + "/check-ins/0/"))
## Returns correctly
ci = CheckIn(
goal,
Timeframe.get_current_timeframe(goal.check_in_frequency_name), 1)
ci.persist()
tfid = str(ci.timeframe)
res = tc.get("/goals/" + goal_id + "/check-ins/" + tfid + "/")
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), ci.to_dict())
def test_get_by_time(self):
tc = self.test_client
## Login required
setup.assertRequiresLogin(self, tc.get("/goals/0/check-ins/"))
## Goal must exist
self.login()
setup.assert404(self, tc.get("/goals/0/check-ins/"))
## User is the goal's owner
goal = self.create_test_numeric_goal()
goal_id = str(goal.get_id())
self.logout()
self.login_other_user()
setup.assertInvalidCredentials(self, tc.get("/goals/" + goal_id + "/check-ins/"))
## Start and end are present
self.logout()
self.login()
setup.assertBadData(self, tc.get("/goals/" + goal_id + "/check-ins/"), "start and end")
setup.assertBadData(self, tc.get("/goals/" + goal_id + "/check-ins/?start=banana"), "start and end")
setup.assertBadData(self, tc.get("/goals/" + goal_id + "/check-ins/?end=banana"), "start and end")
## Start and end in format YYYY-MM-DD HH:mm:ss
res = tc.get("/goals/" + goal_id + "/check-ins/?start=2016-01-01 10:00:00&end=banana")
setup.assertBadData(self, res, "'banana' does not match format")
res = tc.get("/goals/" + goal_id + "/check-ins/?start=strawberry&end=2016-01-01 10:00:00")
setup.assertBadData(self, res, "'strawberry' does not match format")
## Returns empty for no check-ins
res = tc.get("/goals/" + goal_id + "/check-ins/?start=2016-01-01 00:00:00&end=2016-01-08 00:00:00")
setup.assertOk(self, res)
data = json.loads(res.data)
self.assertIn("check-ins", data)
self.assertEqual(len(data['check-ins']), 0)
## Correctly returns multiple check-ins
ci = CheckIn(goal, Timeframe(goal.check_in_frequency_name, datetime.datetime(2015, 12, 31)), 0)
ci.persist()
for x in range(0, 8): #one preceding and one postceding so we can make sure this limits the results
tf = ci.timeframe_obj
new_tf = Timeframe(tf.frequency_name, tf.start + datetime.timedelta(1))
ci = CheckIn(ci.goal_obj, new_tf, x)
ci.persist()
res = tc.get("/goals/" + goal_id + "/check-ins/?start=2016-01-01 00:00:00&end=2016-01-08 00:00:00")
setup.assertOk(self, res)
data = json.loads(res.data)
self.assertIn("check-ins", data)
self.assertEqual(len(data['check-ins']), 7)
for index, check_in in enumerate(data['check-ins']):
self.assertEqual(index, check_in['value'])
def test_get(self):
## Requires login
res = self.test_client.get("/goals/0/")
setup.assertRequiresLogin(self, res)
## 404s
self.login()
res = self.test_client.get("/goals/0/")
setup.assert404(self, res)
## Can access own private goals
goal = self.create_test_goal()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), goal)
## Cannot access other's private goals
self.logout()
self.login_other_user()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertInvalidCredentials(self, res)
## Can access own public goals
self.logout()
self.login()
res = self.test_client.post("/goals/update/",
data={
'id': goal['id'],
'public': True
})
setup.assertOk(self, res)
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
goal['public'] = True
self.assertEqual(json.loads(res.data), goal)
## Can access others' public goals
self.logout()
self.login_other_user()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), goal)
def test_update_password(self):
tc = self.test_client
res = tc.post("/register/", data=self.new_user)
setup.assertOk(self, res, 201)
## Shoudl fail if missing a key
res = tc.post('/users/update-password/', data={"email": self.new_user['email'], "new_password": "******"})
setup.assertInvalid(self, res, 'old_password')
## should fail for non-existent emails
res = tc.post('/users/update-password/', data={"email": "notreal", "old_password": "******", "new_password": "******"})
setup.assertInvalidCredentials(self, res)
res = tc.post('/users/update-password/', data={"email": self.new_user['email'], "old_password": "******", "new_password": "******"})
setup.assertInvalidCredentials(self, res)
res = tc.post('/users/update-password/', data={"email": self.new_user['email'], "old_password": self.new_user['password'], "new_password": "******"})
setup.assertOk(self, res)
res = tc.post("/login/", data={"email": self.new_user['email'], "password": "******"})
setup.assertOk(self, res, 201)
def test_get(self):
## Requires login
res = self.test_client.get("/goals/0/")
setup.assertRequiresLogin(self, res)
## 404s
self.login()
res = self.test_client.get("/goals/0/")
setup.assert404(self, res)
## Can access own private goals
goal = self.create_test_goal()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), goal)
## Cannot access other's private goals
self.logout()
self.login_other_user()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertInvalidCredentials(self, res)
## Can access own public goals
self.logout()
self.login()
res = self.test_client.post("/goals/update/", data={'id': goal['id'], 'public': True})
setup.assertOk(self, res)
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
goal['public'] = True
self.assertEqual(json.loads(res.data), goal)
## Can access others' public goals
self.logout()
self.login_other_user()
res = self.test_client.get('/goals/' + str(goal['id']) + '/')
setup.assertOk(self, res, 200)
self.assertEqual(json.loads(res.data), goal)
def test_destroy(self):
user = User(self.new_user['email'], self.new_user['password'], self.new_user['first_name'], self.new_user['last_name'])
user.persist()
self.assertTrue(user.exists()) ## I don't understand why but this is necessary or else line 45 fails
tc = self.test_client
res = tc.post("/users/delete/", data={"email": "fake email"})
setup.assertInvalid(self, res, 'password')
res = tc.post("/users/delete/", data={"email": "fake email", "password": "******"})
data = json.loads(res.data)
setup.assertInvalidCredentials(self, res)
self.assertTrue(user.exists())
tc = self.test_client
res = tc.post("/users/delete/", data={"email": user.email, "password": "******"})
data = json.loads(res.data)
setup.assertInvalidCredentials(self, res)
self.assertTrue(user.exists())
res = tc.post("/users/delete/", data={"email": user.email, "password": "******"})
setup.assertOk(self, res, 204)
self.assertFalse(user.exists())
def test_update_email(self):
tc = self.test_client
post_data = {"old_email": self.new_user['email'], "new_email": "*****@*****.**", "password": self.new_user['password']}
res = tc.post('/users/update-email/', data=post_data)
setup.assertInvalidCredentials(self, res)
res = tc.post('/register/', data=self.new_user)
setup.assertOk(self, res, 201)
post_data['password'] = '******'
res = tc.post('/users/update-email/', data=post_data)
setup.assertInvalidCredentials(self, res)
post_data['password'] = None
res = tc.post('/users/update-email/', data=post_data)
setup.assertInvalid(self, res, 'password')
post_data['password'] = self.new_user['password']
res = tc.post('/users/update-email/', data=post_data)
setup.assertOk(self, res)
res = tc.post('/login/', data={"email": post_data['new_email'], "password": post_data['password']})
setup.assertOk(self, res, 201)
def test_update(self):
## Requires login
res = self.test_client.post("/goals/update/", data={'id': 0})
setup.assertRequiresLogin(self, res)
## Validates form
self.login()
res = self.test_client.post("/goals/update/", data={'favorite_muppet': 'kermit'})
setup.assertInvalid(self, res, 'id')
## 404's
res = self.test_client.post("/goals/update/", data={'id': 0, 'name': 'test goal 2.0'})
setup.assert404(self, res)
## Actually works
goal = self.create_test_goal()
data = {'id': goal['id']}
data['name'] = "Test Goal 2.0"
data['prompt'] = "Guess how many eggs I had for breakfast today!"
data['frequency'] = "monthly"
data['target'] = 100
data['input_type'] = "binary"
data['active'] = False
data['public'] = True
data['check_in_frequency'] = "daily"
res = self.test_client.post("/goals/update/", data=data)
setup.assertOk(self, res, 200)
updated_goal = Goal.pull_by_id(goal['id'])
for key, val in data.iteritems():
if (key == 'frequency' or key == 'check_in_frequency'):
self.assertEqual(Frequency.get_name_by_id(getattr(updated_goal, key)), val)
else:
self.assertEqual(getattr(updated_goal, key), val)
## If any part is invalid, no part should go through
bad_data = {'id': goal['id']}
bad_data['name'] = "Test Goal 3.0"
bad_data['prompt'] = "How many jars of sand did you collect?"
bad_data['frequency'] = "daily"
bad_data['target'] = 1000
bad_data['input_type'] = "numeric"
bad_data['active'] = True
bad_data['public'] = "banana"
bad_data['check_in_frequency'] = "daily"
res = self.test_client.post("/goals/update/", data=bad_data)
setup.assertBadData(self, res, "Public must be a boolean")
updated_goal = Goal.pull_by_id(goal['id'])
for key, val in data.iteritems():
if (key == 'frequency' or key == 'check_in_frequency'):
self.assertEqual(Frequency.get_name_by_id(getattr(updated_goal, key)), val)
else:
self.assertEqual(getattr(updated_goal, key), val)
## Can't update someone else's goal
self.logout()
self.login_other_user()
res = self.test_client.post("/goals/update/", data=bad_data) ## using bad_data here ensures we don't update
setup.assertInvalidCredentials(self, res)
def test_update(self):
## Requires login
res = self.test_client.post("/goals/update/", data={'id': 0})
setup.assertRequiresLogin(self, res)
## Validates form
self.login()
res = self.test_client.post("/goals/update/",
data={'favorite_muppet': 'kermit'})
setup.assertInvalid(self, res, 'id')
## 404's
res = self.test_client.post("/goals/update/",
data={
'id': 0,
'name': 'test goal 2.0'
})
setup.assert404(self, res)
## Actually works
goal = self.create_test_goal()
data = {'id': goal['id']}
data['name'] = "Test Goal 2.0"
data['prompt'] = "Guess how many eggs I had for breakfast today!"
data['frequency'] = "monthly"
data['target'] = 100
data['input_type'] = "binary"
data['active'] = False
data['public'] = True
data['check_in_frequency'] = "daily"
res = self.test_client.post("/goals/update/", data=data)
setup.assertOk(self, res, 200)
updated_goal = Goal.pull_by_id(goal['id'])
for key, val in data.iteritems():
if (key == 'frequency' or key == 'check_in_frequency'):
self.assertEqual(
Frequency.get_name_by_id(getattr(updated_goal, key)), val)
else:
self.assertEqual(getattr(updated_goal, key), val)
## If any part is invalid, no part should go through
bad_data = {'id': goal['id']}
bad_data['name'] = "Test Goal 3.0"
bad_data['prompt'] = "How many jars of sand did you collect?"
bad_data['frequency'] = "daily"
bad_data['target'] = 1000
bad_data['input_type'] = "numeric"
bad_data['active'] = True
bad_data['public'] = "banana"
bad_data['check_in_frequency'] = "daily"
res = self.test_client.post("/goals/update/", data=bad_data)
setup.assertBadData(self, res, "Public must be a boolean")
updated_goal = Goal.pull_by_id(goal['id'])
for key, val in data.iteritems():
if (key == 'frequency' or key == 'check_in_frequency'):
self.assertEqual(
Frequency.get_name_by_id(getattr(updated_goal, key)), val)
else:
self.assertEqual(getattr(updated_goal, key), val)
## Can't update someone else's goal
self.logout()
self.login_other_user()
res = self.test_client.post(
"/goals/update/",
data=bad_data) ## using bad_data here ensures we don't update
setup.assertInvalidCredentials(self, res)
def test_check_in(self):
tc = self.test_client
# Requires Login:
res = tc.post("/goals/1/check-ins/", data={"value": 1})
setup.assertRequiresLogin(self, res)
## Requires "value"
self.login()
res = tc.post("/goals/1/check-ins/", data={"not-value": 1})
setup.assertInvalid(self, res, "value")
## Must be a goal that exists
res = tc.post("/goals/0/check-ins/", data={"value": 1})
setup.assert404(self, res)
## User must own the goal in question
numeric_goal = self.create_test_numeric_goal()
numeric_goal_id = str(numeric_goal.get_id()) ## You have to do this before you log out, for some reason?
binary_goal = self.create_test_binary_goal()
binary_goal_id = str(binary_goal.get_id())
self.logout()
self.login_other_user()
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": 1})
setup.assertInvalidCredentials(self, res)
## Check-in must conform (true / false vs. numeric)
self.logout()
self.login()
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": "true"})
setup.assertBadData(self, res, "Value must be numeric")
res = tc.post("/goals/" + binary_goal_id + "/check-ins/", data={"value": 10})
setup.assertBadData(self, res, "Value must be a boolean")
## Check-in is returned with 201 if no timeframe is given, returning current timeframe (both numeric & binary)
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": 1})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'], Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
res = tc.post("/goals/" + binary_goal_id + "/check-ins/", data={"value": True})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), binary_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'], Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## An updated check-in is returned with 200 if no timeframe is given, returning current timeframe
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": 3})
setup.assertOk(self, res, 200)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 3)
self.assertEqual(data['timeframe'], Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## Check-in is returned with 201 if timeframe ID is given, returning correct timeframe
tf = Timeframe.get_timeframe("daily", datetime.datetime(2016, 1, 1))
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": 1, "timeframe": tf.get_id()})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'], tf.to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## Updated check-in is returned with 200 if Timeframe ID is given, returning correct timeframe
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/", data={"value": 3, "timeframe": tf.get_id()})
setup.assertOk(self, res, 200)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 3)
self.assertEqual(data['timeframe'], tf.to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
def test_check_in(self):
tc = self.test_client
# Requires Login:
res = tc.post("/goals/1/check-ins/", data={"value": 1})
setup.assertRequiresLogin(self, res)
## Requires "value"
self.login()
res = tc.post("/goals/1/check-ins/", data={"not-value": 1})
setup.assertInvalid(self, res, "value")
## Must be a goal that exists
res = tc.post("/goals/0/check-ins/", data={"value": 1})
setup.assert404(self, res)
## User must own the goal in question
numeric_goal = self.create_test_numeric_goal()
numeric_goal_id = str(numeric_goal.get_id(
)) ## You have to do this before you log out, for some reason?
binary_goal = self.create_test_binary_goal()
binary_goal_id = str(binary_goal.get_id())
self.logout()
self.login_other_user()
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={"value": 1})
setup.assertInvalidCredentials(self, res)
## Check-in must conform (true / false vs. numeric)
self.logout()
self.login()
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={"value": "true"})
setup.assertBadData(self, res, "Value must be numeric")
res = tc.post("/goals/" + binary_goal_id + "/check-ins/",
data={"value": 10})
setup.assertBadData(self, res, "Value must be a boolean")
## Check-in is returned with 201 if no timeframe is given, returning current timeframe (both numeric & binary)
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={"value": 1})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'],
Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
res = tc.post("/goals/" + binary_goal_id + "/check-ins/",
data={"value": True})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), binary_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'],
Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## An updated check-in is returned with 200 if no timeframe is given, returning current timeframe
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={"value": 3})
setup.assertOk(self, res, 200)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 3)
self.assertEqual(data['timeframe'],
Timeframe.get_current_timeframe('daily').to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## Check-in is returned with 201 if timeframe ID is given, returning correct timeframe
tf = Timeframe.get_timeframe("daily", datetime.datetime(2016, 1, 1))
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={
"value": 1,
"timeframe": tf.get_id()
})
setup.assertOk(self, res, 201)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 1)
self.assertEqual(data['timeframe'], tf.to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
## Updated check-in is returned with 200 if Timeframe ID is given, returning correct timeframe
res = tc.post("/goals/" + numeric_goal_id + "/check-ins/",
data={
"value": 3,
"timeframe": tf.get_id()
})
setup.assertOk(self, res, 200)
data = json.loads(res.data)
self.assertIn("id", data)
self.assertEqual(str(data['goal']), numeric_goal_id)
self.assertEqual(data['value'], 3)
self.assertEqual(data['timeframe'], tf.to_dict())
self.assertEqual(CheckIn.pull_by_id(data['id']).to_dict(), data)
def test_get_by_time(self):
tc = self.test_client
## Login required
setup.assertRequiresLogin(self, tc.get("/goals/0/check-ins/"))
## Goal must exist
self.login()
setup.assert404(self, tc.get("/goals/0/check-ins/"))
## User is the goal's owner
goal = self.create_test_numeric_goal()
goal_id = str(goal.get_id())
self.logout()
self.login_other_user()
setup.assertInvalidCredentials(
self, tc.get("/goals/" + goal_id + "/check-ins/"))
## Start and end are present
self.logout()
self.login()
setup.assertBadData(self, tc.get("/goals/" + goal_id + "/check-ins/"),
"start and end")
setup.assertBadData(
self, tc.get("/goals/" + goal_id + "/check-ins/?start=banana"),
"start and end")
setup.assertBadData(
self, tc.get("/goals/" + goal_id + "/check-ins/?end=banana"),
"start and end")
## Start and end in format YYYY-MM-DD HH:mm:ss
res = tc.get("/goals/" + goal_id +
"/check-ins/?start=2016-01-01 10:00:00&end=banana")
setup.assertBadData(self, res, "'banana' does not match format")
res = tc.get("/goals/" + goal_id +
"/check-ins/?start=strawberry&end=2016-01-01 10:00:00")
setup.assertBadData(self, res, "'strawberry' does not match format")
## Returns empty for no check-ins
res = tc.get(
"/goals/" + goal_id +
"/check-ins/?start=2016-01-01 00:00:00&end=2016-01-08 00:00:00")
setup.assertOk(self, res)
data = json.loads(res.data)
self.assertIn("check-ins", data)
self.assertEqual(len(data['check-ins']), 0)
## Correctly returns multiple check-ins
ci = CheckIn(
goal,
Timeframe(goal.check_in_frequency_name,
datetime.datetime(2015, 12, 31)), 0)
ci.persist()
for x in range(
0, 8
): #one preceding and one postceding so we can make sure this limits the results
tf = ci.timeframe_obj
new_tf = Timeframe(tf.frequency_name,
tf.start + datetime.timedelta(1))
ci = CheckIn(ci.goal_obj, new_tf, x)
ci.persist()
res = tc.get(
"/goals/" + goal_id +
"/check-ins/?start=2016-01-01 00:00:00&end=2016-01-08 00:00:00")
setup.assertOk(self, res)
data = json.loads(res.data)
self.assertIn("check-ins", data)
self.assertEqual(len(data['check-ins']), 7)
for index, check_in in enumerate(data['check-ins']):
self.assertEqual(index, check_in['value'])
