60 lines
2.9 KiB
Python
60 lines
2.9 KiB
Python
import unittest
|
|
from random import randint
|
|
from funkload.FunkLoadTestCase import FunkLoadTestCase
|
|
|
|
class bench(FunkLoadTestCase):
|
|
|
|
def setUp(self):
|
|
self.server_url = self.conf_get('main', 'url')
|
|
|
|
def test_movies(self):
|
|
server_url = self.server_url
|
|
id_user_start = self.conf_getInt('test_movies', 'id_user_start')
|
|
id_user_end = self.conf_getInt('test_movies', 'id_user_end')
|
|
for i in range(id_user_start,id_user_end+1):
|
|
self.get(server_url+"/recommendations?user_id="+str(i), description='Get Movies')
|
|
|
|
def test_ratings(self):
|
|
server_url = self.server_url
|
|
id_user_start = self.conf_getInt('test_ratings', 'id_user_start')
|
|
id_user_end = self.conf_getInt('test_ratings', 'id_user_end')
|
|
for i in range(id_user_start,id_user_end+1):
|
|
self.get(server_url+"/movieratings?user_id="+str(i), description='Get Rating')
|
|
|
|
def test_ratings_update(self):
|
|
server_url = self.server_url
|
|
id_user = self.conf_getInt('test_ratings_update', 'id_user')
|
|
id_movies_rating_start = self.conf_getInt('test_ratings_update', 'id_movies_rating_start')
|
|
id_movies_rating_stop = self.conf_getInt('test_ratings_update', 'id_movies_rating_stop')
|
|
for i in range(id_movies_rating_start,id_movies_rating_stop+1):
|
|
self.post(server_url+"/movieratings?user_id="+str(id_user),
|
|
params=[
|
|
['score', str(randint(0,5))],
|
|
],
|
|
description='Update Rating')
|
|
|
|
def test_recommendations_v1(self):
|
|
server_url = self.server_url
|
|
id_user_start = self.conf_getInt('test_recommendations_v1', 'id_user_start')
|
|
id_user_end = self.conf_getInt('test_recommendations_v1', 'id_user_end')
|
|
for i in range(id_user_start,id_user_end+1):
|
|
self.get(server_url+"/recommendations?user_id="+str(i)+"&?processing_mode=1", description='Get Recommendations v1')
|
|
|
|
def test_recommendations_v2(self):
|
|
server_url = self.server_url
|
|
id_user_start = self.conf_getInt('test_recommendations_v2', 'id_user_start')
|
|
id_user_end = self.conf_getInt('test_recommendations_v2', 'id_user_end')
|
|
for i in range(id_user_start,id_user_end+1):
|
|
self.get(server_url+"/recommendations?user_id="+str(i)+"&?processing_mode=2", description='Get Recommendations v2')
|
|
|
|
def test_recommendations_v3(self):
|
|
server_url = self.server_url
|
|
id_user_start = self.conf_getInt('test_recommendations_v3', 'id_user_start')
|
|
id_user_end = self.conf_getInt('test_recommendations_v3', 'id_user_end')
|
|
for i in range(id_user_start,id_user_end+1):
|
|
self.get(server_url+"/recommendations?user_id="+str(i)+"&?processing_mode=3", description='Get Recommendations v3')
|
|
|
|
|
|
if __name__ in ('main', '__main__'):
|
|
unittest.main()
|