Python DropHandler.set_request_hash Examples

Example #1

def test_track_is_saved(mock_pymongo):
    """See if drop as tracked properly."""
    dead = DropHandler(mock_pymongo)
    dead.set_request_hash("127.0.0.1")
    data = {'test': "here"}
    dead.drop(data)

    mock_pymongo.dead.track.insert_one.assert_called_with({
        'key':
        ANY,
        'userHash':
        ANY,
        'createdDate':
        datetime.datetime(2012, 1, 14),
        'pickedUp':
        ANY
    })

Example #2

def test_user_hash_is_truncated(mock_pymongo):
    """See if client_hash is set properly."""
    dead = DropHandler(mock_pymongo)
    ip_addr = "127.0.0.1"
    dead.set_request_hash(ip_addr)
    data = {'test': "here"}
    dead.drop(data)

    salted_ip = dead.salt + ip_addr
    hasher = hashlib.sha256()
    hasher.update(salted_ip.encode('utf-8'))
    client_hash = hasher.hexdigest()[:32]

    mock_pymongo.dead.track.insert_one.assert_called_with({
        'key':
        ANY,
        'userHash':
        client_hash,
        'createdDate':
        datetime.datetime(2012, 1, 14),
        'pickedUp':
        ANY
    })