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)
