def test_get_template_missing_user_from_rec_db(self):
user = _get_user('vansika')
recommendations_cf_recording._get_template(active_section='top_artist', user=user)
self.assertTemplateUsed('recommendations_cf_recording/top_artist.html')
self.assert_context('active_section', 'top_artist')
self.assert_context('user', user)
user = _get_user('vansika')
recommendations_cf_recording._get_template(active_section='similar_artist', user=user)
self.assertTemplateUsed('recommendations_cf_recording/similar_artist.html')
self.assert_context('active_section', 'similar_artist')
self.assert_context('user', user)
def test_get_template_empty_repsonce_similar_artist(
self, mock_get_recommendations, mock_get_rec):
user = _get_user('vansika_1')
mock_get_rec.return_value = UserRecommendationsData(
**{
'recording_mbid': {
'similar_artist': [{
'recording_mbid':
"9f5a56f4-1f83-4681-b319-70a734d0d047",
'score': 0.9
}]
},
'created': datetime.utcnow(),
'user_id': 1
})
mock_get_recommendations.return_value = []
recommendations_cf_recording._get_template(
active_section='similar_artist', user=user)
self.assertTemplateUsed(
'recommendations_cf_recording/similar_artist.html')
self.assert_context('active_section', 'similar_artist')
self.assert_context('user', user)
error_msg = "An error occurred while processing your request. Check back later!"
self.assert_context('error_msg', error_msg)
def similar_artist(user_name: str):
""" Show similar artist user recommendations """
user = _get_user(user_name)
template = _get_template(active_section='similar_artist', user=user)
return template
def top_artist(user_name: str):
""" Show top artist user recommendations """
user = _get_user(user_name)
template = _get_template(active_section='top_artist', user=user)
return template
def test_get_template_missing_user_from_rec_db(self):
user = _get_user('vansika')
recommendations_cf_recording._get_template(active_section='top_artist', user=user)
self.assertTemplateUsed('recommendations_cf_recording/top_artist.html')
self.assert_context('active_section', 'top_artist')
self.assert_context('user', user)
error_msg = "Recommended tracks for the user have not been calculated. Check back later."
self.assert_context('error_msg', error_msg)
user = _get_user('vansika')
recommendations_cf_recording._get_template(active_section='similar_artist', user=user)
self.assertTemplateUsed('recommendations_cf_recording/similar_artist.html')
self.assert_context('active_section', 'similar_artist')
self.assert_context('user', user)
error_msg = "Recommended tracks for the user have not been calculated. Check back later."
self.assert_context('error_msg', error_msg)
def info(user_name):
""" Show info about the recommended tracks
"""
user = _get_user(user_name)
return render_template("recommendations_cf_recording/info.html",
user=user,
active_section='info')
def test_get_template(self, mock_get_recommendations, mock_get_rec, mock_curr_user):
# active_section = 'top_artist'
user = _get_user('vansika_1')
created = datetime.utcnow()
mock_get_rec.return_value = UserRecommendationsData(**{
'recording_mbid': {
'top_artist': [{
'recording_mbid': "9f5a56f4-1f83-4681-b319-70a734d0d047",
'score': 0.9
}]
},
'created': datetime.utcnow(),
'user_id': 1
})
recommendations = [{
'listened_at': 0,
'track_metadata': {
'artist_name': "Ultravox",
'track_name': "Serenade (special remix)",
'release_name': "Quartet",
'additional_info': {
'recording_mbid': "af5a56f4-1f83-4681-b319-70a734d0d047",
'artist_mbids': ["6a70b322-9aa9-41b3-9dce-824733633a1c"]
}
}
}]
mock_get_recommendations.return_value = recommendations
mock_curr_user.id = 10
mock_curr_user.musicbrainz_id = 'vansika'
mock_curr_user.auth_token = 'yyyy'
recommendations_cf_recording._get_template(active_section='top_artist', user=user)
mock_get_rec.assert_called_with(user.id)
mock_get_recommendations.assert_called_once()
self.assertTemplateUsed('recommendations_cf_recording/top_artist.html')
self.assert_context('active_section', 'top_artist')
self.assert_context('user', user)
self.assert_context('last_updated', created.strftime('%d %b %Y'))
expected_props = {
"user": {
"id": 2,
"name": 'vansika_1',
},
"current_user": {
"id": 10,
"name": 'vansika',
"auth_token": 'yyyy',
},
"spotify": {},
"youtube": {},
"api_url": current_app.config["API_URL"],
"web_sockets_server_url": current_app.config['WEBSOCKETS_SERVER_URL'],
"recommendations": recommendations,
"sentry_dsn": current_app.config.get("LOG_SENTRY", {}).get("dsn")
}
received_props = ujson.loads(self.get_context_variable('props'))
self.assertEqual(expected_props, received_props)
# only assert fields that should change with 'active_section'
# here active_section = 'similar_artist'
mock_get_rec.return_value = UserRecommendationsData(**{
'recording_mbid': {
'similar_artist': [{
'recording_mbid': "9f5a56f4-1f83-4681-b319-70a734d0d047",
'score': 0.9
}]
},
'created': datetime.utcnow(),
'user_id': 1
})
recommendations_cf_recording._get_template(active_section='similar_artist', user=user)
self.assertTemplateUsed('recommendations_cf_recording/similar_artist.html')
self.assert_context('active_section', 'similar_artist')
received_props = ujson.loads(self.get_context_variable('props'))
self.assertEqual(expected_props, received_props)
