Python handle_message示例

示例#1

文件： click_log_processor_test.py 项目： zxkkkkk/Tap-News

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!'

示例#2

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!')

示例#3

文件： click_log_processor_test.py 项目： AlexFlanker/Daily-News-Recommendation-System

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

示例#4

文件： click_log_processor_test.py 项目： liaozh02/news_aggregation

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'

示例#5

文件： click_log_processor_test.py 项目： samallenqing/phoenix_news

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!')

示例#6

文件： click_log_processor_test.py 项目： zhangsquared/News-Scraping-and-Recommendation-System

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!")

示例#7

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!')

示例#8

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!'

示例#9

文件： click_log_processor_test.py 项目： dannieatus/react_news_fetcher

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!')