path: root/tests/test_push.py

import pytest
import time
from mastodon.Mastodon import MastodonNotFoundError

def test_decrypt(api):
    priv = {
        'auth': b'\xe7y\x0fp\xb9\x92\xe0\xa0\xc5\xd5~Qr\xd7\xaa\x16',
        'privkey': 86377660131518823691452242868806126400423244879788569019894692873283793307753
    }

    # Yes, I am aware that there is an access token in there. It's not valid anymore.
    encryption_header = "salt=O14vjCdbxxhRUTkrsp98vw"
    crypto_key_header = "dh=BKz_CMlF6kVURVImPDsz3bNbTv-9QTGtXpE4Fd3wQGF44fVTj32-APndoccYdjXY2U-mdTen1PDm_pHacpEmD0M;p256ecdsa=BDSAB3e_l5Qp4X50UYSSrKiZFZWDAgHlWIDhgBjXJuUzb0HrpqoCdFhMCh4o2xYHTqpvyTJ3SfFtrILLiXXWT5k"
    data = b'\x10\\b<\xddi\xacd\x86\xc8J1\xb6}5\x01K\x85;\xd2\xd4WzN\xab\x0b|3D\xe9_YPcsG\x9fh\xae\xfe\xbb:z&\xc4\x8ek\x89\xde\xa2\xdbF\xdc\xdd[p<h\x9e\x95\x8d\xd4\xf0\xd0\xc1\x89\t\x01\xebuV\xb1\xa4Fp\xe3\xbf\x91g\x93\xbe \xe5\xd4\xee\xe2\xb0FaB\x8a\xd0\x00b\xe4Q\x83\xd5\xd9\x83\x9a\x1d\xd5j\xdb"\xc5\xb0\xf5W\xa72r4r]aLs\xa8\x8c\x1a\x19h\xfeX)_t\xd4p\xc9\xd2d\x1b?\x19\xc8X(\x02\xd5\x18\xe4\x93\xe2\xda\x01\xb4b\xe4\xd0F\x08`\x13;>\xc4\x89\xbc\xc3\x8e\xb8\x9bJ~\xc4}]\xdb\xdc\xf1wY\x16g\xf8\x91N\xee\xfd\x92\x1e\xcd\xd2~\xf2\x06\x89\xcd\xa5\xcd\x97\xb7{\xc5\xe1\xe4\xb0\x9f7\xc6\x8a5\xda\xbbm\xce\xc5\x8d\x93`&\x0e\xa9\x83\xa2|p;\xa4\x8b)\xc8\x07\rb!a\x82\xf5E\x92\x00Y{\xd4\x94\xf8\xf0\r\xb5c\x86\xfb\xd0*\xbb\xa1!\x14\xd5\x11\xc8\xafI\xb3j\xca7\xc4\x9c\xe0\x9c0\x12\xc0\xd1\x8a{\xcd\xc4~\\\xc2\x99\xf0d)\x03E\x91;m\xbe\xdb\x86\xef\xd7\xa7>\xd1a\xf1\x83!\xaeB\xaa\xf0\xda\x1b;\x86\xd8;]\x9e\xe3\xfa*!\x07,\t\xbd\xe7\xfc\xa7\xa8\xba[\xcf\x89e\xac\'\xdb\x88g\xd9\\\xe4C\x08Lb\xb6CAT\xcc!\xa4\xce\x92t3\x1c1\x01'
    
    decrypted = api.push_subscription_decrypt_push(
        data, 
        priv, 
        encryption_header,
        crypto_key_header
    )
    
    assert decrypted
    assert decrypted.title == 'You were mentioned by fake halcy'

@pytest.mark.vcr(match_on=['path'])
def test_push_set(api):
    priv, pub = api.push_subscription_generate_keys()
    sub = api.push_subscription_set("example.com", pub, follow_events=True, favourite_events=True, reblog_events=True, mention_events=True, poll_events=True, follow_request_events=True, status_events=True, policy='none')
    assert sub == api.push_subscription()
    assert sub.endpoint == "https://example.com"

    should_throw = False
    try:
        sub = api.push_subscription_set("example.com", pub, follow_events=True, favourite_events=True, reblog_events=True, mention_events=True, poll_events=True, follow_request_events=True, status_events=True, policy='not a valid value')
        should_throw = True
    except:
        pass
    assert not should_throw

@pytest.mark.vcr(match_on=['path'])
def test_push_update(api):
    priv, pub = api.push_subscription_generate_keys()
    sub = api.push_subscription_set("example.com", pub,follow_events=False,
                                    favourite_events=False, reblog_events=False, 
                                    mention_events=False)
    
    sub2 = api.push_subscription_update(follow_events=True, favourite_events=True, 
                                        reblog_events=True, mention_events=True)
    time.sleep(1)
    assert sub2 == api.push_subscription()
    
    sub3 = api.push_subscription_update(follow_events=False, favourite_events=False, 
                                        reblog_events=False, mention_events=False)
    time.sleep(1)
    assert sub3 == api.push_subscription()
    print(sub3)
    print(api.push_subscription())
    
    assert sub3.alerts.follow is False
    assert sub3.alerts.favourite is False
    assert sub3.alerts.reblog is False
    assert sub3.alerts.mention is False
    assert sub2.alerts.follow is True
    assert sub2.alerts.favourite is True
    assert sub2.alerts.reblog is True
    assert sub2.alerts.mention is True
  

@pytest.mark.vcr(match_on=['path'])
def test_push_delete(api):
    priv, pub = api.push_subscription_generate_keys()
    sub = api.push_subscription_set("example.com", pub)
    assert sub 
    
    api.push_subscription_delete()
    with pytest.raises(MastodonNotFoundError):
        api.push_subscription()