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_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'])