def test_db_item_features(self):
itemid_list = ['novl123456', 'novl32435', 'novlerewqerq', 'novl432342234']
resource_type = 'novl'
prefer_res = {}
for itemid in itemid_list:
self.assertEqual( True, db_models_update.set_db_offline_item_features(itemid, resource_type, itemid+'_'+resource_type))
prefer_res[itemid] = itemid+'_'+resource_type
res = db_models.get_db_offline_item_features(itemid_list)
self.assertEqual(True, res[0])
self.assertEqual(prefer_res, res[1])
def test_cache_update_all_together(self):
uid = 'db_uid'
resource_type = 'movi'
item_id = 'db_item_id'
feature_name = 'db_feature_name'
self.assertEqual(True, db_models_update.set_db_offline_user_resource_visitinfo(uid, resource_type, 'db_visit_info_data'))
self.assertEqual(True, db_models_update.set_db_offline_user_feature_list(uid, resource_type, 'db_feature_list_data'))
self.assertEqual(True, db_models_update.set_db_offline_user_recommend_item_list(uid, resource_type, 'db_user_recommend_item_list'))
self.assertEqual(True, db_models_update.set_db_offline_item_features(item_id, resource_type, 'db_item_features_data'))
self.assertEqual(True, db_models_update.set_db_offline_item_recommend_item_list(item_id, resource_type, 'db_item_id_list_data'))
self.assertEqual(True, db_models_update.set_db_offline_feature_hot_item_list(feature_name, resource_type, 'db_hot_item_list_data'))
self.assertEqual(True, db_models_update.set_db_offline_global_hot_feature_list(resource_type, 'db_hot_feature_list_data'))
self.assertEqual(True, db_models_update.set_db_offline_item_id_new_to_old(resource_type, item_id, 'db_old_item_id_data'))
cache_uid = 'cache_uid'
cache_item_id = 'cache_item_id'
cache_feature_name = 'cache_feature_name'
self.assertEqual(True, cache_models.set_cache_online_user_resource_visitinfo(cache_uid, resource_type, 'on_cache_visit_info_data'))
self.assertEqual(True, cache_models.set_cache_offline_user_resource_visitinfo(cache_uid, resource_type, 'cache_visit_info_data'))
self.assertEqual(True, cache_models.set_cache_online_user_feature_list(cache_uid, resource_type, 'on_cache_feature_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_user_feature_list(cache_uid, resource_type, 'cache_feature_list_data'))
self.assertEqual(True, cache_models.set_cache_online_user_feature_reason(cache_uid, 'cache_feature_list_data'))
self.assertEqual(True, cache_models.set_cache_online_user_favor_item_list(cache_uid, resource_type, 'cache_item_id_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_user_recommend_item_list(cache_uid, resource_type, 'cache_feature_name_item_id_list_data'))
self.assertEqual(True, cache_models.set_cache_online_user_push_item_list(cache_uid, 'cache_item_id_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_item_recommend_item_list(cache_item_id, 'cache_item_id_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_item_features(cache_item_id, 'cache_item_feature_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_feature_hot_item_list(cache_feature_name, 'cache_item_id_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_global_hot_feature_list(resource_type, 'cache_item_feature_list_data'))
self.assertEqual(True, cache_models.set_cache_offline_item_id_new_to_old(cache_item_id, 'cache_old_item_id_data'))
import cache_update_process
all_resource_type_list = ['movi', 'news']
common_resource_type_list = ['movi']
self.assertEqual(True, cache_update_process.cache_update_all_together(all_resource_type_list, common_resource_type_list))
self.assertEqual((True, {}), cache_models.get_cache_online_user_resource_visitinfo(cache_uid, [resource_type]))
self.assertEqual((True, {resource_type:'db_visit_info_data'}), cache_models.get_cache_offline_user_resource_visitinfo(uid, [resource_type]))
self.assertEqual((True, {}), cache_models.get_cache_online_user_feature_list(cache_uid, [resource_type]))
self.assertEqual((True, {resource_type:'db_feature_list_data'}), cache_models.get_cache_offline_user_feature_list(uid, [resource_type]))
self.assertEqual((True, None), cache_models.get_cache_online_user_feature_reason(cache_uid))
self.assertEqual((True, {}), cache_models.get_cache_online_user_favor_item_list(cache_uid, [resource_type]))
self.assertEqual((True, 'cache_item_id_list_data'), cache_models.get_cache_online_user_push_item_list(cache_uid))
self.assertEqual((True,{resource_type:'db_user_recommend_item_list'}), cache_models.get_cache_offline_user_recommend_item_list(uid, [resource_type]))
self.assertEqual((True,{item_id:'db_item_id_list_data'}), cache_models.get_cache_offline_item_recommend_item_list([item_id]))
self.assertEqual((True,{item_id:'db_item_features_data'}), cache_models.get_cache_offline_item_features([item_id]))
self.assertEqual((True,{feature_name:'db_hot_item_list_data'}), cache_models.get_cache_offline_feature_hot_item_list([feature_name]))
self.assertEqual((True,{resource_type:'db_hot_feature_list_data'}), cache_models.get_cache_offline_global_hot_feature_list([resource_type]))
self.assertEqual((True,{item_id:'db_old_item_id_data'}), cache_models.get_cache_offline_item_id_new_to_old([item_id]))
def test_update_item_features(self):
item_id_list = ['novl1', 'novl2', 'novl3']
prefer_res = {}
for item_id in item_id_list:
prefer_res[item_id] = 'features_%s_cache' % item_id
self.assertEqual( True, cache_models.set_cache_offline_item_features(item_id, prefer_res[item_id]))
res = cache_models.get_cache_offline_item_features(item_id_list)
self.assertEqual(True, res[0])
self.assertEqual(prefer_res, res[1])
db_item_id_list = ['novl3', 'novl4', 'novl5']
update_prefer_res = {}
for item_id in db_item_id_list:
update_prefer_res[item_id] = 'old_feature_%s_db' % item_id
self.assertEqual( True, db_models_update.set_db_offline_item_features(item_id, DataDef.get_resource_type_from_item_id(item_id), update_prefer_res[item_id]))
self.assertEqual(True, cache_update_process.cache_update_offline_item_features(['novl']))
res = cache_models.get_cache_offline_item_features(db_item_id_list)
self.assertEqual(True, res[0])
self.assertEqual(update_prefer_res, res[1])
def test_interface_offline_item_features(self):
db_itemid_list = ['teletiof_item_1', 'teletiof_item_喜剧', 'carttiof_item_3', 'teletiof_item_4']
self.assertEqual(False, cache_models.set_cache_offline_item_features('teletiof_item_1', None))
prefer_res = {}
for itemid in db_itemid_list:
prefer_res[itemid] = 'data_db_%s' % itemid
self.assertEqual(True, db_models_update.set_db_offline_item_features(itemid, DataDef.get_resource_type_from_item_id(itemid), prefer_res[itemid]))
cache_itemid_list = ['teletiof_item_1', 'teletiof_item_喜剧', 'carttiof_item_5', 'teletiof_item_6']
prefer_res = {}
for itemid in db_itemid_list:
prefer_res[itemid] = 'data_cache_%s' % itemid
self.assertEqual(True, cache_models.set_cache_offline_item_features(itemid, prefer_res[itemid]))
itemid_list = ['teletiof_item_1', 'teletiof_item_喜剧', 'carttiof_item_3', 'teletiof_item_4', 'carttiof_item_5', 'teletiof_item_6']
res = model_interface.get_offline_item_features(itemid_list)
self.assertEqual(True, res[0])
self.assertEqual(prefer_res, res[1])
res = cache_models.get_cache_offline_item_features(itemid_list)
self.assertEqual(True, res[0])
for k,v in res[1].items():
if v == DataDef.CACHE_DEFAULT_NONE_VALUE:
res[1].pop(k)
self.assertEqual(prefer_res, res[1])
