def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many({"userId": "test_user"}) msg = {"userId": "test_user", "newsId": "test_news", "timestamp": str(datetime.utcnow())} click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId': 'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print 'test_basic passed!'
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many({"userId": "test_user"}) msg = { "userId": "test_user", "newsId": "test_news", "timestamp": str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId': 'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print('test_basic passed!')
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many({'userId':'test_user'}) msg = {'userId':'test_user', 'newsId':TEST_NEWS_DIGEST, 'timestamp':str(datetime.utcnow())} click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId':'test_user'}) print(model) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES
def test_basic(): db = mongodb_client.get_db() db[PREFER_DB_COLLECTION].delete_many({'userId': 'test_user'}) msg = { 'userId': 'test_user', 'newsId': 'test_news', 'timestamp': str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFER_DB_COLLECTION].find_one({'userId': 'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print 'test_basic passed'
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many({'userId': 'test_user'}) msg = { 'userId': 'test_user', 'newsId': '4034f11d616da70a18444ac4c6dabaf5', 'timestamp': str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId': 'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print('test_basic passed!')
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_TABLE_NAME].delete_many({'userId': USERID}) msg = { 'userId': USERID, 'newsId': NEWSID, 'timestamp': str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFERENCE_TABLE_NAME].find_one({'userId': USERID}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print("test_basic passed!")
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many({'userId': 'test_user'}) msg = { 'userId':'test_user', 'newsId':'TRUshhvCK8sTarbZoAQApQ==\n', 'timestamp':str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId':'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES assert math.isclose(model['preference']['U.S.'], 0.2125, rel_tol=1e-10) print('test_basic passed!')
def test_basic(): db = mongodb_client.get_db() db[PREFERENCE_MODEL_TABLE_NAME].delete_many( {"userId": "*****@*****.**"}) msg = { "userId": "*****@*****.**", "newsId": "XHe0M1b3RlpbpU2ttpJgQw==\n", "timestamp": str(datetime.utcnow()) } click_log_processor.handle_message(msg) model = db[PREFERENCE_MODEL_TABLE_NAME].find_one( {'userId': '*****@*****.**'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES print 'test_basic passed!'
def test_basic(): db = mongodb_client.get_db() # delete previous test user db[PREFERENCE_MODEL_TABLE_NAME].delete_many({'userId': 'test_user'}) # if mongodb has no newsId for test, so enter mongo->show dbs-> use demo -> show collections -> db.news.find().preggy()--> copy one digest msg = { 'userId':'test_user', 'newsId':'3RjuEomJo26O1syZbU7OHA==\n', 'timestamp':str(datetime.utcnow()) } click_log_processor.handle_message(msg) # jiancha mongodb actually has a preference_model for test_user model = db[PREFERENCE_MODEL_TABLE_NAME].find_one({'userId':'test_user'}) assert model is not None assert len(model['preference']) == NUM_OF_CLASSES assert math.isclose(model['preference']['U.S.'], 0.2125, rel_tol=1e-10) print('test_basic passed!')