def test_update_latest_import(self): user = db_user.get_or_create(3, 'updatelatestimportuser') self.assertEqual(int(user['latest_import'].strftime('%s')), 0) val = int(time.time()) db_user.update_latest_import(user['musicbrainz_id'], val) user = db_user.get_by_mb_id(user['musicbrainz_id']) self.assertEqual(int(user['latest_import'].strftime('%s')), val)
def test_reset_import_timestamp_post(self): self.temporary_login(self.user['id']) val = int(time.time()) db_user.update_latest_import(self.user['musicbrainz_id'], val) response = self.client.post( url_for('profile.reset_latest_import_timestamp'), data={ 'reset': 'yes', 'token': self.user['auth_token'] }) self.assertStatus(response, 302) # should have redirected to the import page self.assertRedirects(response, url_for('profile.info')) ts = db_user.get(self.user['id'])['latest_import'].strftime('%s') self.assertEqual(int(ts), 0)