mirror of https://gitlab.com/bashrc2/epicyon
4389 lines
170 KiB
Python
4389 lines
170 KiB
Python
__filename__ = "tests.py"
|
||
__author__ = "Bob Mottram"
|
||
__license__ = "AGPL3+"
|
||
__version__ = "1.2.0"
|
||
__maintainer__ = "Bob Mottram"
|
||
__email__ = "bob@freedombone.net"
|
||
__status__ = "Production"
|
||
__module_group__ = "Testing"
|
||
|
||
import time
|
||
import os
|
||
import shutil
|
||
import json
|
||
import datetime
|
||
from random import randint
|
||
from time import gmtime, strftime
|
||
from pprint import pprint
|
||
from httpsig import signPostHeaders
|
||
from httpsig import signPostHeadersNew
|
||
from httpsig import verifyPostHeaders
|
||
from httpsig import messageContentDigest
|
||
from cache import storePersonInCache
|
||
from cache import getPersonFromCache
|
||
from threads import threadWithTrace
|
||
from daemon import runDaemon
|
||
from session import createSession
|
||
from posts import removePostInteractions
|
||
from posts import getMentionedPeople
|
||
from posts import validContentWarning
|
||
from posts import deleteAllPosts
|
||
from posts import createPublicPost
|
||
from posts import sendPost
|
||
from posts import noOfFollowersOnDomain
|
||
from posts import groupFollowersByDomain
|
||
from posts import archivePostsForPerson
|
||
from posts import sendPostViaServer
|
||
from follow import clearFollows
|
||
from follow import clearFollowers
|
||
from follow import sendFollowRequestViaServer
|
||
from follow import sendUnfollowRequestViaServer
|
||
from siteactive import siteIsActive
|
||
from utils import validPassword
|
||
from utils import userAgentDomain
|
||
from utils import camelCaseSplit
|
||
from utils import decodedHost
|
||
from utils import getFullDomain
|
||
from utils import validNickname
|
||
from utils import firstParagraphFromString
|
||
from utils import removeIdEnding
|
||
from utils import updateRecentPostsCache
|
||
from utils import followPerson
|
||
from utils import getNicknameFromActor
|
||
from utils import getDomainFromActor
|
||
from utils import copytree
|
||
from utils import loadJson
|
||
from utils import saveJson
|
||
from utils import getStatusNumber
|
||
from utils import getFollowersOfPerson
|
||
from utils import removeHtml
|
||
from utils import dangerousMarkup
|
||
from utils import acctDir
|
||
from pgp import extractPGPPublicKey
|
||
from pgp import pgpPublicKeyUpload
|
||
from utils import containsPGPPublicKey
|
||
from follow import followerOfPerson
|
||
from follow import unfollowAccount
|
||
from follow import unfollowerOfAccount
|
||
from follow import sendFollowRequest
|
||
from person import createPerson
|
||
from person import setDisplayNickname
|
||
from person import setBio
|
||
# from person import generateRSAKey
|
||
from skills import setSkillLevel
|
||
from skills import actorSkillValue
|
||
from skills import setSkillsFromDict
|
||
from skills import actorHasSkill
|
||
from roles import setRolesFromList
|
||
from roles import setRole
|
||
from roles import actorHasRole
|
||
from auth import constantTimeStringCheck
|
||
from auth import createBasicAuthHeader
|
||
from auth import authorizeBasic
|
||
from auth import storeBasicCredentials
|
||
from like import likePost
|
||
from like import sendLikeViaServer
|
||
from announce import announcePublic
|
||
from announce import sendAnnounceViaServer
|
||
from city import parseNogoString
|
||
from city import spoofGeolocation
|
||
from city import pointInNogo
|
||
from media import getImageDimensions
|
||
from media import getMediaPath
|
||
from media import getAttachmentMediaType
|
||
from delete import sendDeleteViaServer
|
||
from inbox import jsonPostAllowsComments
|
||
from inbox import validInbox
|
||
from inbox import validInboxFilenames
|
||
from categories import guessHashtagCategory
|
||
from content import limitRepeatedWords
|
||
from content import switchWords
|
||
from content import extractTextFieldsInPOST
|
||
from content import validHashTag
|
||
from content import htmlReplaceEmailQuote
|
||
from content import htmlReplaceQuoteMarks
|
||
from content import dangerousCSS
|
||
from content import addWebLinks
|
||
from content import replaceEmojiFromTags
|
||
from content import addHtmlTags
|
||
from content import removeLongWords
|
||
from content import replaceContentDuplicates
|
||
from content import removeTextFormatting
|
||
from content import removeHtmlTag
|
||
from theme import updateDefaultThemesList
|
||
from theme import setCSSparam
|
||
from linked_data_sig import generateJsonSignature
|
||
from linked_data_sig import verifyJsonSignature
|
||
from newsdaemon import hashtagRuleTree
|
||
from newsdaemon import hashtagRuleResolve
|
||
from newswire import getNewswireTags
|
||
from newswire import parseFeedDate
|
||
from newswire import limitWordLengths
|
||
from mastoapiv1 import getMastoApiV1IdFromNickname
|
||
from mastoapiv1 import getNicknameFromMastoApiV1Id
|
||
from webapp_post import prepareHtmlPostNickname
|
||
from speaker import speakerReplaceLinks
|
||
from markdown import markdownToHtml
|
||
from languages import setActorLanguages
|
||
from languages import getActorLanguages
|
||
from languages import getLinksFromContent
|
||
from languages import addLinksToContent
|
||
from shares import authorizeSharedItems
|
||
from shares import generateSharedItemFederationTokens
|
||
from shares import createSharedItemFederationToken
|
||
from shares import updateSharedItemFederationToken
|
||
from shares import mergeSharedItemTokens
|
||
|
||
testServerAliceRunning = False
|
||
testServerBobRunning = False
|
||
testServerEveRunning = False
|
||
thrAlice = None
|
||
thrBob = None
|
||
thrEve = None
|
||
|
||
|
||
def _testHttpSigNew():
|
||
print('testHttpSigNew')
|
||
messageBodyJson = {"hello": "world"}
|
||
messageBodyJsonStr = json.dumps(messageBodyJson)
|
||
publicKeyPem = \
|
||
'-----BEGIN RSA PUBLIC KEY-----\n' + \
|
||
'MIIBCgKCAQEAhAKYdtoeoy8zcAcR874L8' + \
|
||
'cnZxKzAGwd7v36APp7Pv6Q2jdsPBRrw\n' + \
|
||
'WEBnez6d0UDKDwGbc6nxfEXAy5mbhgajz' + \
|
||
'rw3MOEt8uA5txSKobBpKDeBLOsdJKFq\n' + \
|
||
'MGmXCQvEG7YemcxDTRPxAleIAgYYRjTSd' + \
|
||
'/QBwVW9OwNFhekro3RtlinV0a75jfZg\n' + \
|
||
'kne/YiktSvLG34lw2zqXBDTC5NHROUqGT' + \
|
||
'lML4PlNZS5Ri2U4aCNx2rUPRcKIlE0P\n' + \
|
||
'uKxI4T+HIaFpv8+rdV6eUgOrB2xeI1dSF' + \
|
||
'Fn/nnv5OoZJEIB+VmuKn3DCUcCZSFlQ\n' + \
|
||
'PSXSfBDiUGhwOw76WuSSsf1D4b/vLoJ10wIDAQAB\n' + \
|
||
'-----END RSA PUBLIC KEY-----\n'
|
||
|
||
privateKeyPem = \
|
||
'-----BEGIN RSA PRIVATE KEY-----\n' + \
|
||
'MIIEqAIBAAKCAQEAhAKYdtoeoy8zcAcR8' + \
|
||
'74L8cnZxKzAGwd7v36APp7Pv6Q2jdsP\n' + \
|
||
'BRrwWEBnez6d0UDKDwGbc6nxfEXAy5mbh' + \
|
||
'gajzrw3MOEt8uA5txSKobBpKDeBLOsd\n' + \
|
||
'JKFqMGmXCQvEG7YemcxDTRPxAleIAgYYR' + \
|
||
'jTSd/QBwVW9OwNFhekro3RtlinV0a75\n' + \
|
||
'jfZgkne/YiktSvLG34lw2zqXBDTC5NHRO' + \
|
||
'UqGTlML4PlNZS5Ri2U4aCNx2rUPRcKI\n' + \
|
||
'lE0PuKxI4T+HIaFpv8+rdV6eUgOrB2xeI' + \
|
||
'1dSFFn/nnv5OoZJEIB+VmuKn3DCUcCZ\n' + \
|
||
'SFlQPSXSfBDiUGhwOw76WuSSsf1D4b/vL' + \
|
||
'oJ10wIDAQABAoIBAG/JZuSWdoVHbi56\n' + \
|
||
'vjgCgkjg3lkO1KrO3nrdm6nrgA9P9qaPj' + \
|
||
'xuKoWaKO1cBQlE1pSWp/cKncYgD5WxE\n' + \
|
||
'CpAnRUXG2pG4zdkzCYzAh1i+c34L6oZoH' + \
|
||
'sirK6oNcEnHveydfzJL5934egm6p8DW\n' + \
|
||
'+m1RQ70yUt4uRc0YSor+q1LGJvGQHReF0' + \
|
||
'WmJBZHrhz5e63Pq7lE0gIwuBqL8SMaA\n' + \
|
||
'yRXtK+JGxZpImTq+NHvEWWCu09SCq0r83' + \
|
||
'8ceQI55SvzmTkwqtC+8AT2zFviMZkKR\n' + \
|
||
'Qo6SPsrqItxZWRty2izawTF0Bf5S2VAx7' + \
|
||
'O+6t3wBsQ1sLptoSgX3QblELY5asI0J\n' + \
|
||
'YFz7LJECgYkAsqeUJmqXE3LP8tYoIjMIA' + \
|
||
'KiTm9o6psPlc8CrLI9CH0UbuaA2JCOM\n' + \
|
||
'cCNq8SyYbTqgnWlB9ZfcAm/cFpA8tYci9' + \
|
||
'm5vYK8HNxQr+8FS3Qo8N9RJ8d0U5Csw\n' + \
|
||
'DzMYfRghAfUGwmlWj5hp1pQzAuhwbOXFt' + \
|
||
'xKHVsMPhz1IBtF9Y8jvgqgYHLbmyiu1\n' + \
|
||
'mwJ5AL0pYF0G7x81prlARURwHo0Yf52kE' + \
|
||
'w1dxpx+JXER7hQRWQki5/NsUEtv+8RT\n' + \
|
||
'qn2m6qte5DXLyn83b1qRscSdnCCwKtKWU' + \
|
||
'ug5q2ZbwVOCJCtmRwmnP131lWRYfj67\n' + \
|
||
'B/xJ1ZA6X3GEf4sNReNAtaucPEelgR2ns' + \
|
||
'N0gKQKBiGoqHWbK1qYvBxX2X3kbPDkv\n' + \
|
||
'9C+celgZd2PW7aGYLCHq7nPbmfDV0yHcW' + \
|
||
'jOhXZ8jRMjmANVR/eLQ2EfsRLdW69bn\n' + \
|
||
'f3ZD7JS1fwGnO3exGmHO3HZG+6AvberKY' + \
|
||
'VYNHahNFEw5TsAcQWDLRpkGybBcxqZo\n' + \
|
||
'81YCqlqidwfeO5YtlO7etx1xLyqa2NsCe' + \
|
||
'G9A86UjG+aeNnXEIDk1PDK+EuiThIUa\n' + \
|
||
'/2IxKzJKWl1BKr2d4xAfR0ZnEYuRrbeDQ' + \
|
||
'YgTImOlfW6/GuYIxKYgEKCFHFqJATAG\n' + \
|
||
'IxHrq1PDOiSwXd2GmVVYyEmhZnbcp8Cxa' + \
|
||
'EMQoevxAta0ssMK3w6UsDtvUvYvF22m\n' + \
|
||
'qQKBiD5GwESzsFPy3Ga0MvZpn3D6EJQLg' + \
|
||
'snrtUPZx+z2Ep2x0xc5orneB5fGyF1P\n' + \
|
||
'WtP+fG5Q6Dpdz3LRfm+KwBCWFKQjg7uTx' + \
|
||
'cjerhBWEYPmEMKYwTJF5PBG9/ddvHLQ\n' + \
|
||
'EQeNC8fHGg4UXU8mhHnSBt3EA10qQJfRD' + \
|
||
's15M38eG2cYwB1PZpDHScDnDA0=\n' + \
|
||
'-----END RSA PRIVATE KEY-----'
|
||
sigInput = \
|
||
'sig1=(date); alg=rsa-sha256; keyId="test-key-b"'
|
||
sig = \
|
||
'sig1=:HtXycCl97RBVkZi66ADKnC9c5eSSlb57GnQ4KFqNZplOpNfxqk62' + \
|
||
'JzZ484jXgLvoOTRaKfR4hwyxlcyb+BWkVasApQovBSdit9Ml/YmN2IvJDPncrlhPD' + \
|
||
'VDv36Z9/DiSO+RNHD7iLXugdXo1+MGRimW1RmYdenl/ITeb7rjfLZ4b9VNnLFtVWw' + \
|
||
'rjhAiwIqeLjodVImzVc5srrk19HMZNuUejK6I3/MyN3+3U8tIRW4LWzx6ZgGZUaEE' + \
|
||
'P0aBlBkt7Fj0Tt5/P5HNW/Sa/m8smxbOHnwzAJDa10PyjzdIbywlnWIIWtZKPPsoV' + \
|
||
'oKVopUWEU3TNhpWmaVhFrUL/O6SN3w==:'
|
||
# "hs2019", using RSASSA-PSS [RFC8017] and SHA-512 [RFC6234]
|
||
# sigInput = \
|
||
# 'sig1=(*request-target, *created, host, date, ' + \
|
||
# 'cache-control, x-empty-header, x-example); keyId="test-key-a"; ' + \
|
||
# 'alg=hs2019; created=1402170695; expires=1402170995'
|
||
# sig = \
|
||
# 'sig1=:K2qGT5srn2OGbOIDzQ6kYT+ruaycnDAAUpKv+ePFfD0RAxn/1BUe' + \
|
||
# 'Zx/Kdrq32DrfakQ6bPsvB9aqZqognNT6be4olHROIkeV879RrsrObury8L9SCEibe' + \
|
||
# 'oHyqU/yCjphSmEdd7WD+zrchK57quskKwRefy2iEC5S2uAH0EPyOZKWlvbKmKu5q4' + \
|
||
# 'CaB8X/I5/+HLZLGvDiezqi6/7p2Gngf5hwZ0lSdy39vyNMaaAT0tKo6nuVw0S1MVg' + \
|
||
# '1Q7MpWYZs0soHjttq0uLIA3DIbQfLiIvK6/l0BdWTU7+2uQj7lBkQAsFZHoA96ZZg' + \
|
||
# 'FquQrXRlmYOh+Hx5D9fJkXcXe5tmAg==:'
|
||
nickname = 'foo'
|
||
boxpath = '/' + nickname
|
||
# headers = {
|
||
# "*request-target": "get " + boxpath,
|
||
# "*created": "1402170695",
|
||
# "host": "example.org",
|
||
# "date": "Tue, 07 Jun 2014 20:51:35 GMT",
|
||
# "cache-control": "max-age=60, must-revalidate",
|
||
# "x-emptyheader": "",
|
||
# "x-example": "Example header with some whitespace.",
|
||
# "x-dictionary": "b=2",
|
||
# "x-dictionary": "a=1",
|
||
# "x-list": "(a, b, c)",
|
||
# "Signature-Input": sigInput,
|
||
# "Signature": sig
|
||
# }
|
||
dateStr = "Tue, 07 Jun 2014 20:51:35 GMT"
|
||
secondsSinceEpoch = 1402174295
|
||
domain = "example.com"
|
||
port = 443
|
||
headers = {
|
||
"*created": str(secondsSinceEpoch),
|
||
"*request-target": "post /foo?param=value&pet=dog",
|
||
"host": domain,
|
||
"date": dateStr,
|
||
"content-type": "application/json",
|
||
"digest": "SHA-256=X48E9qOokqqrvdts8nOJRJN3OWDUoyWxBf7kbu9DBPE=",
|
||
"content-length": "18",
|
||
"Signature-Input": sigInput,
|
||
"Signature": sig
|
||
}
|
||
httpPrefix = 'https'
|
||
debug = False
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, False, None,
|
||
messageBodyJsonStr, debug, True)
|
||
# make a deliberate mistake
|
||
headers['Signature'] = headers['Signature'].replace('V', 'B')
|
||
assert not verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, False, None,
|
||
messageBodyJsonStr, debug, True)
|
||
# test signing
|
||
bodyDigest = messageContentDigest(messageBodyJsonStr)
|
||
contentLength = len(messageBodyJsonStr)
|
||
headers = {
|
||
"host": domain,
|
||
"date": dateStr,
|
||
"digest": f'SHA-256={bodyDigest}',
|
||
"content-type": "application/json",
|
||
"content-length": str(contentLength)
|
||
}
|
||
signatureIndexHeader, signatureHeader = \
|
||
signPostHeadersNew(dateStr, privateKeyPem, nickname,
|
||
domain, port,
|
||
domain, port,
|
||
boxpath, httpPrefix, messageBodyJsonStr,
|
||
'rsa-sha256')
|
||
expectedIndexHeader = \
|
||
'keyId="https://example.com/users/foo#main-key"; ' + \
|
||
'alg=hs2019; created=' + str(secondsSinceEpoch) + '; ' + \
|
||
'sig1=(*request-target, *created, host, date, ' + \
|
||
'digest, content-type, content-length)'
|
||
if signatureIndexHeader != expectedIndexHeader:
|
||
print('Unexpected new http header: ' + signatureIndexHeader)
|
||
print('Should be: ' + expectedIndexHeader)
|
||
assert signatureIndexHeader == expectedIndexHeader
|
||
assert signatureHeader == \
|
||
'sig1=:euX3O1KSTYXN9/oR2qFezswWm9FbrjtRymK7xBpXNQvTs' + \
|
||
'XehtrNdD8nELZKzPXMvMz7PaJd6V+fjzpHoZ9upTdqqQLK2Iwml' + \
|
||
'p4BlHqW6Aopd7sZFCWFq7/Amm5oaizpp3e0jb5XISS5m3cRKuoi' + \
|
||
'LM0x+OudmAoYGi0TEEJk8bpnJAXfVCDfmOyL3XNqQeShQHeOANG' + \
|
||
'okiKktj8ff+KLYLaPTAJkob1k/EhoPIkbw/YzAY8IZjWQNMkf+F' + \
|
||
'JChApQ5HnDCQPwD5xV9eGzBpAf6D0G19xiTmQye4Hn6tAs3fy3V' + \
|
||
'/aYa/GhW2pSrctDnAKIi4imj9joppr3CB8gqgXZOPQ==:'
|
||
|
||
|
||
def _testHttpsigBase(withDigest):
|
||
print('testHttpsig(' + str(withDigest) + ')')
|
||
|
||
baseDir = os.getcwd()
|
||
path = baseDir + '/.testHttpsigBase'
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
os.chdir(path)
|
||
|
||
contentType = 'application/activity+json'
|
||
nickname = 'socrates'
|
||
domain = 'argumentative.social'
|
||
httpPrefix = 'https'
|
||
port = 5576
|
||
password = 'SuperSecretPassword'
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(path, nickname, domain, port, httpPrefix,
|
||
False, False, password)
|
||
assert privateKeyPem
|
||
messageBodyJson = {
|
||
"a key": "a value",
|
||
"another key": "A string",
|
||
"yet another key": "Another string"
|
||
}
|
||
messageBodyJsonStr = json.dumps(messageBodyJson)
|
||
|
||
headersDomain = getFullDomain(domain, port)
|
||
|
||
dateStr = strftime("%a, %d %b %Y %H:%M:%S %Z", gmtime())
|
||
boxpath = '/inbox'
|
||
if not withDigest:
|
||
headers = {
|
||
'host': headersDomain,
|
||
'date': dateStr,
|
||
'content-type': 'application/json'
|
||
}
|
||
signatureHeader = \
|
||
signPostHeaders(dateStr, privateKeyPem, nickname,
|
||
domain, port,
|
||
domain, port,
|
||
boxpath, httpPrefix, None)
|
||
else:
|
||
bodyDigest = messageContentDigest(messageBodyJsonStr)
|
||
contentLength = len(messageBodyJsonStr)
|
||
headers = {
|
||
'host': headersDomain,
|
||
'date': dateStr,
|
||
'digest': f'SHA-256={bodyDigest}',
|
||
'content-type': contentType,
|
||
'content-length': str(contentLength)
|
||
}
|
||
signatureHeader = \
|
||
signPostHeaders(dateStr, privateKeyPem, nickname,
|
||
domain, port,
|
||
domain, port,
|
||
boxpath, httpPrefix, messageBodyJsonStr)
|
||
|
||
headers['signature'] = signatureHeader
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, False, None,
|
||
messageBodyJsonStr, False)
|
||
if withDigest:
|
||
# everything correct except for content-length
|
||
headers['content-length'] = str(contentLength + 2)
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, False, None,
|
||
messageBodyJsonStr, False) is False
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
'/parambulator' + boxpath, False, None,
|
||
messageBodyJsonStr, False) is False
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, True, None,
|
||
messageBodyJsonStr, False) is False
|
||
if not withDigest:
|
||
# fake domain
|
||
headers = {
|
||
'host': 'bogon.domain',
|
||
'date': dateStr,
|
||
'content-type': 'application/json'
|
||
}
|
||
else:
|
||
# correct domain but fake message
|
||
messageBodyJsonStr = \
|
||
'{"a key": "a value", "another key": "Fake GNUs", ' + \
|
||
'"yet another key": "More Fake GNUs"}'
|
||
contentLength = len(messageBodyJsonStr)
|
||
bodyDigest = messageContentDigest(messageBodyJsonStr)
|
||
headers = {
|
||
'host': domain,
|
||
'date': dateStr,
|
||
'digest': f'SHA-256={bodyDigest}',
|
||
'content-type': contentType,
|
||
'content-length': str(contentLength)
|
||
}
|
||
headers['signature'] = signatureHeader
|
||
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
|
||
boxpath, True, None,
|
||
messageBodyJsonStr, False) is False
|
||
|
||
os.chdir(baseDir)
|
||
shutil.rmtree(path)
|
||
|
||
|
||
def _testHttpsig():
|
||
_testHttpsigBase(True)
|
||
_testHttpsigBase(False)
|
||
|
||
|
||
def _testCache():
|
||
print('testCache')
|
||
personUrl = "cat@cardboard.box"
|
||
personJson = {
|
||
"id": 123456,
|
||
"test": "This is a test"
|
||
}
|
||
personCache = {}
|
||
storePersonInCache(None, personUrl, personJson, personCache, True)
|
||
result = getPersonFromCache(None, personUrl, personCache, True)
|
||
assert result['id'] == 123456
|
||
assert result['test'] == 'This is a test'
|
||
|
||
|
||
def _testThreadsFunction(param: str):
|
||
for i in range(10000):
|
||
time.sleep(2)
|
||
|
||
|
||
def _testThreads():
|
||
print('testThreads')
|
||
thr = \
|
||
threadWithTrace(target=_testThreadsFunction,
|
||
args=('test',),
|
||
daemon=True)
|
||
thr.start()
|
||
assert thr.is_alive() is True
|
||
time.sleep(1)
|
||
thr.kill()
|
||
thr.join()
|
||
assert thr.is_alive() is False
|
||
|
||
|
||
def createServerAlice(path: str, domain: str, port: int,
|
||
bobAddress: str, federationList: [],
|
||
hasFollows: bool, hasPosts: bool,
|
||
sendThreads: []):
|
||
print('Creating test server: Alice on port ' + str(port))
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
os.chdir(path)
|
||
sharedItemsFederatedDomains = []
|
||
systemLanguage = 'en'
|
||
nickname = 'alice'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
password = 'alicepass'
|
||
maxReplies = 64
|
||
domainMaxPostsPerDay = 1000
|
||
accountMaxPostsPerDay = 1000
|
||
allowDeletion = True
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(path, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
deleteAllPosts(path, nickname, domain, 'inbox')
|
||
deleteAllPosts(path, nickname, domain, 'outbox')
|
||
assert setSkillLevel(path, nickname, domain, 'hacking', 90)
|
||
assert setRole(path, nickname, domain, 'guru')
|
||
if hasFollows:
|
||
followPerson(path, nickname, domain, 'bob', bobAddress,
|
||
federationList, False)
|
||
followerOfPerson(path, nickname, domain, 'bob', bobAddress,
|
||
federationList, False)
|
||
if hasPosts:
|
||
testFollowersOnly = False
|
||
testSaveToFile = True
|
||
clientToServer = False
|
||
testCommentsEnabled = True
|
||
testAttachImageFilename = None
|
||
testMediaType = None
|
||
testImageDescription = None
|
||
testCity = 'London, England'
|
||
testInReplyTo = None
|
||
testInReplyToAtomUri = None
|
||
testSubject = None
|
||
testSchedulePost = False
|
||
testEventDate = None
|
||
testEventTime = None
|
||
testLocation = None
|
||
testIsArticle = False
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"No wise fish would go anywhere without a porpoise",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"Curiouser and curiouser!",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"In the gardens of memory, in the palace " +
|
||
"of dreams, that is where you and I shall meet",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
global testServerAliceRunning
|
||
testServerAliceRunning = True
|
||
maxMentions = 10
|
||
maxEmoji = 10
|
||
onionDomain = None
|
||
i2pDomain = None
|
||
allowLocalNetworkAccess = True
|
||
maxNewswirePosts = 20
|
||
dormantMonths = 3
|
||
sendThreadsTimeoutMins = 30
|
||
maxFollowers = 10
|
||
verifyAllSignatures = True
|
||
brochMode = False
|
||
showNodeInfoAccounts = True
|
||
showNodeInfoVersion = True
|
||
city = 'London, England'
|
||
logLoginFailures = False
|
||
userAgentsBlocked = []
|
||
print('Server running: Alice')
|
||
runDaemon(sharedItemsFederatedDomains,
|
||
userAgentsBlocked,
|
||
logLoginFailures, city,
|
||
showNodeInfoAccounts,
|
||
showNodeInfoVersion,
|
||
brochMode,
|
||
verifyAllSignatures,
|
||
sendThreadsTimeoutMins,
|
||
dormantMonths, maxNewswirePosts,
|
||
allowLocalNetworkAccess,
|
||
2048, False, True, False, False, True, maxFollowers,
|
||
0, 100, 1024, 5, False,
|
||
0, False, 1, False, False, False,
|
||
5, True, True, 'en', __version__,
|
||
"instanceId", False, path, domain,
|
||
onionDomain, i2pDomain, None, port, port,
|
||
httpPrefix, federationList, maxMentions, maxEmoji, False,
|
||
proxyType, maxReplies,
|
||
domainMaxPostsPerDay, accountMaxPostsPerDay,
|
||
allowDeletion, True, True, False, sendThreads,
|
||
False)
|
||
|
||
|
||
def createServerBob(path: str, domain: str, port: int,
|
||
aliceAddress: str, federationList: [],
|
||
hasFollows: bool, hasPosts: bool,
|
||
sendThreads: []):
|
||
print('Creating test server: Bob on port ' + str(port))
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
os.chdir(path)
|
||
sharedItemsFederatedDomains = []
|
||
systemLanguage = 'en'
|
||
nickname = 'bob'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
clientToServer = False
|
||
password = 'bobpass'
|
||
maxReplies = 64
|
||
domainMaxPostsPerDay = 1000
|
||
accountMaxPostsPerDay = 1000
|
||
allowDeletion = True
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(path, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
deleteAllPosts(path, nickname, domain, 'inbox')
|
||
deleteAllPosts(path, nickname, domain, 'outbox')
|
||
if hasFollows:
|
||
followPerson(path, nickname, domain,
|
||
'alice', aliceAddress, federationList, False)
|
||
followerOfPerson(path, nickname, domain,
|
||
'alice', aliceAddress, federationList, False)
|
||
if hasPosts:
|
||
testFollowersOnly = False
|
||
testSaveToFile = True
|
||
testCommentsEnabled = True
|
||
testAttachImageFilename = None
|
||
testImageDescription = None
|
||
testMediaType = None
|
||
testCity = 'London, England'
|
||
testInReplyTo = None
|
||
testInReplyToAtomUri = None
|
||
testSubject = None
|
||
testSchedulePost = False
|
||
testEventDate = None
|
||
testEventTime = None
|
||
testLocation = None
|
||
testIsArticle = False
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"It's your life, live it your way.",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"One of the things I've realised is that " +
|
||
"I am very simple",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
createPublicPost(path, nickname, domain, port, httpPrefix,
|
||
"Quantum physics is a bit of a passion of mine",
|
||
testFollowersOnly,
|
||
testSaveToFile,
|
||
clientToServer,
|
||
testCommentsEnabled,
|
||
testAttachImageFilename,
|
||
testMediaType,
|
||
testImageDescription, testCity,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
global testServerBobRunning
|
||
testServerBobRunning = True
|
||
maxMentions = 10
|
||
maxEmoji = 10
|
||
onionDomain = None
|
||
i2pDomain = None
|
||
allowLocalNetworkAccess = True
|
||
maxNewswirePosts = 20
|
||
dormantMonths = 3
|
||
sendThreadsTimeoutMins = 30
|
||
maxFollowers = 10
|
||
verifyAllSignatures = True
|
||
brochMode = False
|
||
showNodeInfoAccounts = True
|
||
showNodeInfoVersion = True
|
||
city = 'London, England'
|
||
logLoginFailures = False
|
||
userAgentsBlocked = []
|
||
print('Server running: Bob')
|
||
runDaemon(sharedItemsFederatedDomains,
|
||
userAgentsBlocked,
|
||
logLoginFailures, city,
|
||
showNodeInfoAccounts,
|
||
showNodeInfoVersion,
|
||
brochMode,
|
||
verifyAllSignatures,
|
||
sendThreadsTimeoutMins,
|
||
dormantMonths, maxNewswirePosts,
|
||
allowLocalNetworkAccess,
|
||
2048, False, True, False, False, True, maxFollowers,
|
||
0, 100, 1024, 5, False, 0,
|
||
False, 1, False, False, False,
|
||
5, True, True, 'en', __version__,
|
||
"instanceId", False, path, domain,
|
||
onionDomain, i2pDomain, None, port, port,
|
||
httpPrefix, federationList, maxMentions, maxEmoji, False,
|
||
proxyType, maxReplies,
|
||
domainMaxPostsPerDay, accountMaxPostsPerDay,
|
||
allowDeletion, True, True, False, sendThreads,
|
||
False)
|
||
|
||
|
||
def createServerEve(path: str, domain: str, port: int, federationList: [],
|
||
hasFollows: bool, hasPosts: bool,
|
||
sendThreads: []):
|
||
print('Creating test server: Eve on port ' + str(port))
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
os.chdir(path)
|
||
sharedItemsFederatedDomains = []
|
||
nickname = 'eve'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
password = 'evepass'
|
||
maxReplies = 64
|
||
allowDeletion = True
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(path, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
deleteAllPosts(path, nickname, domain, 'inbox')
|
||
deleteAllPosts(path, nickname, domain, 'outbox')
|
||
global testServerEveRunning
|
||
testServerEveRunning = True
|
||
maxMentions = 10
|
||
maxEmoji = 10
|
||
onionDomain = None
|
||
i2pDomain = None
|
||
allowLocalNetworkAccess = True
|
||
maxNewswirePosts = 20
|
||
dormantMonths = 3
|
||
sendThreadsTimeoutMins = 30
|
||
maxFollowers = 10
|
||
verifyAllSignatures = True
|
||
brochMode = False
|
||
showNodeInfoAccounts = True
|
||
showNodeInfoVersion = True
|
||
city = 'London, England'
|
||
logLoginFailures = False
|
||
userAgentsBlocked = []
|
||
print('Server running: Eve')
|
||
runDaemon(sharedItemsFederatedDomains,
|
||
userAgentsBlocked,
|
||
logLoginFailures, city,
|
||
showNodeInfoAccounts,
|
||
showNodeInfoVersion,
|
||
brochMode,
|
||
verifyAllSignatures,
|
||
sendThreadsTimeoutMins,
|
||
dormantMonths, maxNewswirePosts,
|
||
allowLocalNetworkAccess,
|
||
2048, False, True, False, False, True, maxFollowers,
|
||
0, 100, 1024, 5, False, 0,
|
||
False, 1, False, False, False,
|
||
5, True, True, 'en', __version__,
|
||
"instanceId", False, path, domain,
|
||
onionDomain, i2pDomain, None, port, port,
|
||
httpPrefix, federationList, maxMentions, maxEmoji, False,
|
||
proxyType, maxReplies, allowDeletion, True, True, False,
|
||
sendThreads, False)
|
||
|
||
|
||
def testPostMessageBetweenServers():
|
||
print('Testing sending message from one server to the inbox of another')
|
||
|
||
global testServerAliceRunning
|
||
global testServerBobRunning
|
||
testServerAliceRunning = False
|
||
testServerBobRunning = False
|
||
|
||
systemLanguage = 'en'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
|
||
baseDir = os.getcwd()
|
||
if os.path.isdir(baseDir + '/.tests'):
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
os.mkdir(baseDir + '/.tests')
|
||
|
||
# create the servers
|
||
aliceDir = baseDir + '/.tests/alice'
|
||
aliceDomain = '127.0.0.50'
|
||
alicePort = 61935
|
||
aliceAddress = aliceDomain + ':' + str(alicePort)
|
||
|
||
bobDir = baseDir + '/.tests/bob'
|
||
bobDomain = '127.0.0.100'
|
||
bobPort = 61936
|
||
federationList = [bobDomain, aliceDomain]
|
||
aliceSendThreads = []
|
||
bobSendThreads = []
|
||
bobAddress = bobDomain + ':' + str(bobPort)
|
||
|
||
global thrAlice
|
||
if thrAlice:
|
||
while thrAlice.is_alive():
|
||
thrAlice.stop()
|
||
time.sleep(1)
|
||
thrAlice.kill()
|
||
|
||
thrAlice = \
|
||
threadWithTrace(target=createServerAlice,
|
||
args=(aliceDir, aliceDomain, alicePort, bobAddress,
|
||
federationList, False, False,
|
||
aliceSendThreads),
|
||
daemon=True)
|
||
|
||
global thrBob
|
||
if thrBob:
|
||
while thrBob.is_alive():
|
||
thrBob.stop()
|
||
time.sleep(1)
|
||
thrBob.kill()
|
||
|
||
thrBob = \
|
||
threadWithTrace(target=createServerBob,
|
||
args=(bobDir, bobDomain, bobPort, aliceAddress,
|
||
federationList, False, False,
|
||
bobSendThreads),
|
||
daemon=True)
|
||
|
||
thrAlice.start()
|
||
thrBob.start()
|
||
assert thrAlice.is_alive() is True
|
||
assert thrBob.is_alive() is True
|
||
|
||
# wait for both servers to be running
|
||
while not (testServerAliceRunning and testServerBobRunning):
|
||
time.sleep(1)
|
||
|
||
time.sleep(1)
|
||
|
||
print('\n\n*******************************************************')
|
||
print('Alice sends to Bob')
|
||
os.chdir(aliceDir)
|
||
sessionAlice = createSession(proxyType)
|
||
inReplyTo = None
|
||
inReplyToAtomUri = None
|
||
subject = None
|
||
alicePostLog = []
|
||
followersOnly = False
|
||
saveToFile = True
|
||
clientToServer = False
|
||
ccUrl = None
|
||
alicePersonCache = {}
|
||
aliceCachedWebfingers = {}
|
||
attachedImageFilename = baseDir + '/img/logo.png'
|
||
testImageWidth, testImageHeight = \
|
||
getImageDimensions(attachedImageFilename)
|
||
assert testImageWidth
|
||
assert testImageHeight
|
||
mediaType = getAttachmentMediaType(attachedImageFilename)
|
||
attachedImageDescription = 'Logo'
|
||
isArticle = False
|
||
city = 'London, England'
|
||
# nothing in Alice's outbox
|
||
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 0
|
||
|
||
sendResult = \
|
||
sendPost(__version__,
|
||
sessionAlice, aliceDir, 'alice', aliceDomain, alicePort,
|
||
'bob', bobDomain, bobPort, ccUrl, httpPrefix,
|
||
'Why is a mouse when it spins? ' +
|
||
'यह एक परीक्षण है #sillyquestion',
|
||
followersOnly,
|
||
saveToFile, clientToServer, True,
|
||
attachedImageFilename, mediaType,
|
||
attachedImageDescription, city, federationList,
|
||
aliceSendThreads, alicePostLog, aliceCachedWebfingers,
|
||
alicePersonCache, isArticle, systemLanguage, inReplyTo,
|
||
inReplyToAtomUri, subject)
|
||
print('sendResult: ' + str(sendResult))
|
||
|
||
queuePath = bobDir + '/accounts/bob@' + bobDomain + '/queue'
|
||
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
|
||
mPath = getMediaPath()
|
||
mediaPath = aliceDir + '/' + mPath
|
||
for i in range(30):
|
||
if os.path.isdir(inboxPath):
|
||
if len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) > 0:
|
||
if len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath,
|
||
name))]) == 1:
|
||
if len([name for name in os.listdir(mediaPath)
|
||
if os.path.isfile(os.path.join(mediaPath,
|
||
name))]) > 0:
|
||
if len([name for name in os.listdir(queuePath)
|
||
if os.path.isfile(os.path.join(queuePath,
|
||
name))]) == 0:
|
||
break
|
||
time.sleep(1)
|
||
|
||
# check that a news account exists
|
||
newsActorDir = aliceDir + '/accounts/news@' + aliceDomain
|
||
print("newsActorDir: " + newsActorDir)
|
||
assert os.path.isdir(newsActorDir)
|
||
newsActorFile = newsActorDir + '.json'
|
||
assert os.path.isfile(newsActorFile)
|
||
newsActorJson = loadJson(newsActorFile)
|
||
assert newsActorJson
|
||
assert newsActorJson.get("id")
|
||
# check the id of the news actor
|
||
print('News actor Id: ' + newsActorJson["id"])
|
||
assert (newsActorJson["id"] ==
|
||
httpPrefix + '://' + aliceAddress + '/users/news')
|
||
|
||
# Image attachment created
|
||
assert len([name for name in os.listdir(mediaPath)
|
||
if os.path.isfile(os.path.join(mediaPath, name))]) > 0
|
||
# inbox item created
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
|
||
# queue item removed
|
||
testval = len([name for name in os.listdir(queuePath)
|
||
if os.path.isfile(os.path.join(queuePath, name))])
|
||
print('queuePath: ' + queuePath + ' '+str(testval))
|
||
assert testval == 0
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
print('Check that message received from Alice contains the expected text')
|
||
for name in os.listdir(inboxPath):
|
||
filename = os.path.join(inboxPath, name)
|
||
assert os.path.isfile(filename)
|
||
receivedJson = loadJson(filename, 0)
|
||
if receivedJson:
|
||
pprint(receivedJson['object']['content'])
|
||
assert receivedJson
|
||
assert 'Why is a mouse when it spins?' in \
|
||
receivedJson['object']['content']
|
||
assert 'Why is a mouse when it spins?' in \
|
||
receivedJson['object']['contentMap'][systemLanguage]
|
||
assert 'यह एक परीक्षण है' in receivedJson['object']['content']
|
||
print('Check that message received from Alice contains an attachment')
|
||
assert receivedJson['object']['attachment']
|
||
assert len(receivedJson['object']['attachment']) == 1
|
||
attached = receivedJson['object']['attachment'][0]
|
||
pprint(attached)
|
||
assert attached.get('type')
|
||
assert attached.get('url')
|
||
assert attached['mediaType'] == 'image/png'
|
||
assert '/media/' in attached['url']
|
||
assert attached['url'].endswith('.png')
|
||
assert attached.get('width')
|
||
assert attached.get('height')
|
||
assert attached['width'] > 0
|
||
assert attached['height'] > 0
|
||
|
||
print('\n\n*******************************************************')
|
||
print("Bob likes Alice's post")
|
||
|
||
aliceDomainStr = aliceDomain + ':' + str(alicePort)
|
||
followerOfPerson(bobDir, 'bob', bobDomain, 'alice',
|
||
aliceDomainStr, federationList, False)
|
||
bobDomainStr = bobDomain + ':' + str(bobPort)
|
||
followPerson(aliceDir, 'alice', aliceDomain, 'bob',
|
||
bobDomainStr, federationList, False)
|
||
|
||
sessionBob = createSession(proxyType)
|
||
bobPostLog = []
|
||
bobPersonCache = {}
|
||
bobCachedWebfingers = {}
|
||
statusNumber = None
|
||
outboxPostFilename = None
|
||
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
|
||
for name in os.listdir(outboxPath):
|
||
if '#statuses#' in name:
|
||
statusNumber = \
|
||
int(name.split('#statuses#')[1].replace('.json', ''))
|
||
outboxPostFilename = outboxPath + '/' + name
|
||
assert statusNumber > 0
|
||
assert outboxPostFilename
|
||
assert likePost({}, sessionBob, bobDir, federationList,
|
||
'bob', bobDomain, bobPort, httpPrefix,
|
||
'alice', aliceDomain, alicePort, [],
|
||
statusNumber, False, bobSendThreads, bobPostLog,
|
||
bobPersonCache, bobCachedWebfingers,
|
||
True, __version__)
|
||
|
||
for i in range(20):
|
||
if 'likes' in open(outboxPostFilename).read():
|
||
break
|
||
time.sleep(1)
|
||
|
||
alicePostJson = loadJson(outboxPostFilename, 0)
|
||
if alicePostJson:
|
||
pprint(alicePostJson)
|
||
|
||
assert 'likes' in open(outboxPostFilename).read()
|
||
|
||
print('\n\n*******************************************************')
|
||
print("Bob repeats Alice's post")
|
||
objectUrl = \
|
||
httpPrefix + '://' + aliceDomain + ':' + str(alicePort) + \
|
||
'/users/alice/statuses/' + str(statusNumber)
|
||
inboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/inbox'
|
||
outboxPath = bobDir + '/accounts/bob@' + bobDomain + '/outbox'
|
||
outboxBeforeAnnounceCount = \
|
||
len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))])
|
||
beforeAnnounceCount = \
|
||
len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
print('inbox items before announce: ' + str(beforeAnnounceCount))
|
||
print('outbox items before announce: ' + str(outboxBeforeAnnounceCount))
|
||
assert outboxBeforeAnnounceCount == 0
|
||
assert beforeAnnounceCount == 0
|
||
announcePublic(sessionBob, bobDir, federationList,
|
||
'bob', bobDomain, bobPort, httpPrefix,
|
||
objectUrl,
|
||
False, bobSendThreads, bobPostLog,
|
||
bobPersonCache, bobCachedWebfingers,
|
||
True, __version__)
|
||
announceMessageArrived = False
|
||
outboxMessageArrived = False
|
||
for i in range(10):
|
||
time.sleep(1)
|
||
if not os.path.isdir(inboxPath):
|
||
continue
|
||
if len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) > 0:
|
||
outboxMessageArrived = True
|
||
print('Announce created by Bob')
|
||
if len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) > 0:
|
||
announceMessageArrived = True
|
||
print('Announce message sent to Alice!')
|
||
if announceMessageArrived and outboxMessageArrived:
|
||
break
|
||
afterAnnounceCount = \
|
||
len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
outboxAfterAnnounceCount = \
|
||
len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))])
|
||
print('inbox items after announce: ' + str(afterAnnounceCount))
|
||
print('outbox items after announce: ' + str(outboxAfterAnnounceCount))
|
||
assert afterAnnounceCount == beforeAnnounceCount+1
|
||
assert outboxAfterAnnounceCount == outboxBeforeAnnounceCount + 1
|
||
# stop the servers
|
||
thrAlice.kill()
|
||
thrAlice.join()
|
||
assert thrAlice.is_alive() is False
|
||
|
||
thrBob.kill()
|
||
thrBob.join()
|
||
assert thrBob.is_alive() is False
|
||
|
||
os.chdir(baseDir)
|
||
shutil.rmtree(aliceDir)
|
||
shutil.rmtree(bobDir)
|
||
|
||
|
||
def testFollowBetweenServers():
|
||
print('Testing sending a follow request from one server to another')
|
||
|
||
global testServerAliceRunning
|
||
global testServerBobRunning
|
||
testServerAliceRunning = False
|
||
testServerBobRunning = False
|
||
|
||
systemLanguage = 'en'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
federationList = []
|
||
|
||
baseDir = os.getcwd()
|
||
if os.path.isdir(baseDir + '/.tests'):
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
os.mkdir(baseDir + '/.tests')
|
||
|
||
# create the servers
|
||
aliceDir = baseDir + '/.tests/alice'
|
||
aliceDomain = '127.0.0.47'
|
||
alicePort = 61935
|
||
aliceSendThreads = []
|
||
aliceAddress = aliceDomain + ':' + str(alicePort)
|
||
|
||
bobDir = baseDir + '/.tests/bob'
|
||
bobDomain = '127.0.0.79'
|
||
bobPort = 61936
|
||
bobSendThreads = []
|
||
bobAddress = bobDomain + ':' + str(bobPort)
|
||
|
||
global thrAlice
|
||
if thrAlice:
|
||
while thrAlice.is_alive():
|
||
thrAlice.stop()
|
||
time.sleep(1)
|
||
thrAlice.kill()
|
||
|
||
thrAlice = \
|
||
threadWithTrace(target=createServerAlice,
|
||
args=(aliceDir, aliceDomain, alicePort, bobAddress,
|
||
federationList, False, False,
|
||
aliceSendThreads),
|
||
daemon=True)
|
||
|
||
global thrBob
|
||
if thrBob:
|
||
while thrBob.is_alive():
|
||
thrBob.stop()
|
||
time.sleep(1)
|
||
thrBob.kill()
|
||
|
||
thrBob = \
|
||
threadWithTrace(target=createServerBob,
|
||
args=(bobDir, bobDomain, bobPort, aliceAddress,
|
||
federationList, False, False,
|
||
bobSendThreads),
|
||
daemon=True)
|
||
|
||
thrAlice.start()
|
||
thrBob.start()
|
||
assert thrAlice.is_alive() is True
|
||
assert thrBob.is_alive() is True
|
||
|
||
# wait for all servers to be running
|
||
ctr = 0
|
||
while not (testServerAliceRunning and testServerBobRunning):
|
||
time.sleep(1)
|
||
ctr += 1
|
||
if ctr > 60:
|
||
break
|
||
print('Alice online: ' + str(testServerAliceRunning))
|
||
print('Bob online: ' + str(testServerBobRunning))
|
||
assert ctr <= 60
|
||
time.sleep(1)
|
||
|
||
# In the beginning all was calm and there were no follows
|
||
|
||
print('*********************************************************')
|
||
print('Alice sends a follow request to Bob')
|
||
os.chdir(aliceDir)
|
||
sessionAlice = createSession(proxyType)
|
||
inReplyTo = None
|
||
inReplyToAtomUri = None
|
||
subject = None
|
||
alicePostLog = []
|
||
followersOnly = False
|
||
saveToFile = True
|
||
clientToServer = False
|
||
ccUrl = None
|
||
alicePersonCache = {}
|
||
aliceCachedWebfingers = {}
|
||
alicePostLog = []
|
||
bobActor = httpPrefix + '://' + bobAddress + '/users/bob'
|
||
sendResult = \
|
||
sendFollowRequest(sessionAlice, aliceDir,
|
||
'alice', aliceDomain, alicePort, httpPrefix,
|
||
'bob', bobDomain, bobActor,
|
||
bobPort, httpPrefix,
|
||
clientToServer, federationList,
|
||
aliceSendThreads, alicePostLog,
|
||
aliceCachedWebfingers, alicePersonCache,
|
||
True, __version__)
|
||
print('sendResult: ' + str(sendResult))
|
||
|
||
for t in range(16):
|
||
if os.path.isfile(bobDir + '/accounts/bob@' +
|
||
bobDomain + '/followers.txt'):
|
||
if os.path.isfile(aliceDir + '/accounts/alice@' +
|
||
aliceDomain + '/following.txt'):
|
||
if os.path.isfile(aliceDir + '/accounts/alice@' +
|
||
aliceDomain + '/followingCalendar.txt'):
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
assert 'alice@' + aliceDomain in open(bobDir + '/accounts/bob@' +
|
||
bobDomain + '/followers.txt').read()
|
||
assert 'bob@' + bobDomain in open(aliceDir + '/accounts/alice@' +
|
||
aliceDomain + '/following.txt').read()
|
||
assert 'bob@' + bobDomain in open(aliceDir + '/accounts/alice@' +
|
||
aliceDomain +
|
||
'/followingCalendar.txt').read()
|
||
|
||
print('\n\n*********************************************************')
|
||
print('Alice sends a message to Bob')
|
||
alicePostLog = []
|
||
alicePersonCache = {}
|
||
aliceCachedWebfingers = {}
|
||
alicePostLog = []
|
||
isArticle = False
|
||
city = 'London, England'
|
||
sendResult = \
|
||
sendPost(__version__,
|
||
sessionAlice, aliceDir, 'alice', aliceDomain, alicePort,
|
||
'bob', bobDomain, bobPort, ccUrl,
|
||
httpPrefix, 'Alice message', followersOnly, saveToFile,
|
||
clientToServer, True,
|
||
None, None, None, city, federationList,
|
||
aliceSendThreads, alicePostLog, aliceCachedWebfingers,
|
||
alicePersonCache, isArticle, systemLanguage, inReplyTo,
|
||
inReplyToAtomUri, subject)
|
||
print('sendResult: ' + str(sendResult))
|
||
|
||
queuePath = bobDir + '/accounts/bob@' + bobDomain + '/queue'
|
||
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
|
||
aliceMessageArrived = False
|
||
for i in range(20):
|
||
time.sleep(1)
|
||
if os.path.isdir(inboxPath):
|
||
if len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) > 0:
|
||
aliceMessageArrived = True
|
||
print('Alice message sent to Bob!')
|
||
break
|
||
|
||
assert aliceMessageArrived is True
|
||
print('Message from Alice to Bob succeeded')
|
||
|
||
# stop the servers
|
||
thrAlice.kill()
|
||
thrAlice.join()
|
||
assert thrAlice.is_alive() is False
|
||
|
||
thrBob.kill()
|
||
thrBob.join()
|
||
assert thrBob.is_alive() is False
|
||
|
||
# queue item removed
|
||
time.sleep(4)
|
||
assert len([name for name in os.listdir(queuePath)
|
||
if os.path.isfile(os.path.join(queuePath, name))]) == 0
|
||
|
||
os.chdir(baseDir)
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
|
||
|
||
def _testFollowersOfPerson():
|
||
print('testFollowersOfPerson')
|
||
currDir = os.getcwd()
|
||
nickname = 'mxpop'
|
||
domain = 'diva.domain'
|
||
password = 'birb'
|
||
port = 80
|
||
httpPrefix = 'https'
|
||
federationList = []
|
||
baseDir = currDir + '/.tests_followersofperson'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
createPerson(baseDir, nickname, domain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'maxboardroom', domain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'ultrapancake', domain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'drokk', domain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'sausagedog', domain, port,
|
||
httpPrefix, True, False, password)
|
||
|
||
clearFollows(baseDir, nickname, domain)
|
||
followPerson(baseDir, nickname, domain, 'maxboardroom', domain,
|
||
federationList, False)
|
||
followPerson(baseDir, 'drokk', domain, 'ultrapancake', domain,
|
||
federationList, False)
|
||
# deliberate duplication
|
||
followPerson(baseDir, 'drokk', domain, 'ultrapancake', domain,
|
||
federationList, False)
|
||
followPerson(baseDir, 'sausagedog', domain, 'ultrapancake', domain,
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'ultrapancake', domain,
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'someother', 'randodomain.net',
|
||
federationList, False)
|
||
|
||
followList = getFollowersOfPerson(baseDir, 'ultrapancake', domain)
|
||
assert len(followList) == 3
|
||
assert 'mxpop@' + domain in followList
|
||
assert 'drokk@' + domain in followList
|
||
assert 'sausagedog@' + domain in followList
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def _testNoOfFollowersOnDomain():
|
||
print('testNoOfFollowersOnDomain')
|
||
currDir = os.getcwd()
|
||
nickname = 'mxpop'
|
||
domain = 'diva.domain'
|
||
otherdomain = 'soup.dragon'
|
||
password = 'birb'
|
||
port = 80
|
||
httpPrefix = 'https'
|
||
federationList = []
|
||
baseDir = currDir + '/.tests_nooffollowersOndomain'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
createPerson(baseDir, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
createPerson(baseDir, 'maxboardroom', otherdomain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'ultrapancake', otherdomain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'drokk', otherdomain, port,
|
||
httpPrefix, True, False, password)
|
||
createPerson(baseDir, 'sausagedog', otherdomain, port,
|
||
httpPrefix, True, False, password)
|
||
|
||
followPerson(baseDir, 'drokk', otherdomain, nickname, domain,
|
||
federationList, False)
|
||
followPerson(baseDir, 'sausagedog', otherdomain, nickname, domain,
|
||
federationList, False)
|
||
followPerson(baseDir, 'maxboardroom', otherdomain, nickname, domain,
|
||
federationList, False)
|
||
|
||
followerOfPerson(baseDir, nickname, domain,
|
||
'cucumber', 'sandwiches.party',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain,
|
||
'captainsensible', 'damned.zone',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'pilchard', 'zombies.attack',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'drokk', otherdomain,
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'sausagedog', otherdomain,
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'maxboardroom', otherdomain,
|
||
federationList, False)
|
||
|
||
followersOnOtherDomain = \
|
||
noOfFollowersOnDomain(baseDir, nickname + '@' + domain, otherdomain)
|
||
assert followersOnOtherDomain == 3
|
||
|
||
unfollowerOfAccount(baseDir, nickname, domain, 'sausagedog', otherdomain)
|
||
followersOnOtherDomain = \
|
||
noOfFollowersOnDomain(baseDir, nickname + '@' + domain, otherdomain)
|
||
assert followersOnOtherDomain == 2
|
||
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def _testGroupFollowers():
|
||
print('testGroupFollowers')
|
||
|
||
currDir = os.getcwd()
|
||
nickname = 'test735'
|
||
domain = 'mydomain.com'
|
||
password = 'somepass'
|
||
port = 80
|
||
httpPrefix = 'https'
|
||
federationList = []
|
||
baseDir = currDir + '/.tests_testgroupfollowers'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
createPerson(baseDir, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
|
||
clearFollowers(baseDir, nickname, domain)
|
||
followerOfPerson(baseDir, nickname, domain, 'badger', 'wild.domain',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'squirrel', 'wild.domain',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'rodent', 'wild.domain',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'utterly', 'clutterly.domain',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'zonked', 'zzz.domain',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'nap', 'zzz.domain',
|
||
federationList, False)
|
||
|
||
grouped = groupFollowersByDomain(baseDir, nickname, domain)
|
||
assert len(grouped.items()) == 3
|
||
assert grouped.get('zzz.domain')
|
||
assert grouped.get('clutterly.domain')
|
||
assert grouped.get('wild.domain')
|
||
assert len(grouped['zzz.domain']) == 2
|
||
assert len(grouped['wild.domain']) == 3
|
||
assert len(grouped['clutterly.domain']) == 1
|
||
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def _testFollows():
|
||
print('testFollows')
|
||
currDir = os.getcwd()
|
||
nickname = 'test529'
|
||
domain = 'testdomain.com'
|
||
password = 'mypass'
|
||
port = 80
|
||
httpPrefix = 'https'
|
||
federationList = ['wild.com', 'mesh.com']
|
||
baseDir = currDir + '/.tests_testfollows'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
createPerson(baseDir, nickname, domain, port, httpPrefix, True,
|
||
False, password)
|
||
|
||
clearFollows(baseDir, nickname, domain)
|
||
followPerson(baseDir, nickname, domain, 'badger', 'wild.com',
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'squirrel', 'secret.com',
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'rodent', 'drainpipe.com',
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'batman', 'mesh.com',
|
||
federationList, False)
|
||
followPerson(baseDir, nickname, domain, 'giraffe', 'trees.com',
|
||
federationList, False)
|
||
|
||
accountDir = acctDir(baseDir, nickname, domain)
|
||
f = open(accountDir + '/following.txt', 'r')
|
||
domainFound = False
|
||
for followingDomain in f:
|
||
testDomain = followingDomain.split('@')[1]
|
||
testDomain = testDomain.replace('\n', '').replace('\r', '')
|
||
if testDomain == 'mesh.com':
|
||
domainFound = True
|
||
if testDomain not in federationList:
|
||
print(testDomain)
|
||
assert(False)
|
||
|
||
assert(domainFound)
|
||
unfollowAccount(baseDir, nickname, domain, 'batman', 'mesh.com')
|
||
|
||
domainFound = False
|
||
for followingDomain in f:
|
||
testDomain = followingDomain.split('@')[1]
|
||
testDomain = testDomain.replace('\n', '').replace('\r', '')
|
||
if testDomain == 'mesh.com':
|
||
domainFound = True
|
||
assert(domainFound is False)
|
||
|
||
clearFollowers(baseDir, nickname, domain)
|
||
followerOfPerson(baseDir, nickname, domain, 'badger', 'wild.com',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'squirrel', 'secret.com',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'rodent', 'drainpipe.com',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'batman', 'mesh.com',
|
||
federationList, False)
|
||
followerOfPerson(baseDir, nickname, domain, 'giraffe', 'trees.com',
|
||
federationList, False)
|
||
|
||
accountDir = acctDir(baseDir, nickname, domain)
|
||
f = open(accountDir + '/followers.txt', 'r')
|
||
for followerDomain in f:
|
||
testDomain = followerDomain.split('@')[1]
|
||
testDomain = testDomain.replace('\n', '').replace('\r', '')
|
||
if testDomain not in federationList:
|
||
print(testDomain)
|
||
assert(False)
|
||
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def _testCreatePerson():
|
||
print('testCreatePerson')
|
||
systemLanguage = 'en'
|
||
currDir = os.getcwd()
|
||
nickname = 'test382'
|
||
domain = 'badgerdomain.com'
|
||
password = 'mypass'
|
||
port = 80
|
||
httpPrefix = 'https'
|
||
clientToServer = False
|
||
baseDir = currDir + '/.tests_createperson'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(baseDir, nickname, domain, port,
|
||
httpPrefix, True, False, password)
|
||
assert os.path.isfile(baseDir + '/accounts/passwords')
|
||
deleteAllPosts(baseDir, nickname, domain, 'inbox')
|
||
deleteAllPosts(baseDir, nickname, domain, 'outbox')
|
||
setDisplayNickname(baseDir, nickname, domain, 'badger')
|
||
setBio(baseDir, nickname, domain, 'Randomly roaming in your backyard')
|
||
archivePostsForPerson(nickname, domain, baseDir, 'inbox', None, {}, 4)
|
||
archivePostsForPerson(nickname, domain, baseDir, 'outbox', None, {}, 4)
|
||
testInReplyTo = None
|
||
testInReplyToAtomUri = None
|
||
testSubject = None
|
||
testSchedulePost = False
|
||
testEventDate = None
|
||
testEventTime = None
|
||
testLocation = None
|
||
testIsArticle = False
|
||
content = "G'day world!"
|
||
followersOnly = False
|
||
saveToFile = True
|
||
commentsEnabled = True
|
||
attachImageFilename = None
|
||
mediaType = None
|
||
createPublicPost(baseDir, nickname, domain, port, httpPrefix,
|
||
content, followersOnly, saveToFile, clientToServer,
|
||
commentsEnabled, attachImageFilename, mediaType,
|
||
'Not suitable for Vogons', 'London, England',
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def _testAuthentication():
|
||
print('testAuthentication')
|
||
currDir = os.getcwd()
|
||
nickname = 'test8743'
|
||
password = 'SuperSecretPassword12345'
|
||
|
||
baseDir = currDir + '/.tests_authentication'
|
||
if os.path.isdir(baseDir):
|
||
shutil.rmtree(baseDir)
|
||
os.mkdir(baseDir)
|
||
os.chdir(baseDir)
|
||
|
||
assert storeBasicCredentials(baseDir, 'othernick', 'otherpass')
|
||
assert storeBasicCredentials(baseDir, 'bad:nick', 'otherpass') is False
|
||
assert storeBasicCredentials(baseDir, 'badnick', 'otherpa:ss') is False
|
||
assert storeBasicCredentials(baseDir, nickname, password)
|
||
|
||
authHeader = createBasicAuthHeader(nickname, password)
|
||
assert authorizeBasic(baseDir, '/users/' + nickname + '/inbox',
|
||
authHeader, False)
|
||
assert authorizeBasic(baseDir, '/users/' + nickname,
|
||
authHeader, False) is False
|
||
assert authorizeBasic(baseDir, '/users/othernick/inbox',
|
||
authHeader, False) is False
|
||
|
||
authHeader = createBasicAuthHeader(nickname, password + '1')
|
||
assert authorizeBasic(baseDir, '/users/' + nickname + '/inbox',
|
||
authHeader, False) is False
|
||
|
||
password = 'someOtherPassword'
|
||
assert storeBasicCredentials(baseDir, nickname, password)
|
||
|
||
authHeader = createBasicAuthHeader(nickname, password)
|
||
assert authorizeBasic(baseDir, '/users/' + nickname + '/inbox',
|
||
authHeader, False)
|
||
|
||
os.chdir(currDir)
|
||
shutil.rmtree(baseDir)
|
||
|
||
|
||
def testClientToServer():
|
||
print('Testing sending a post via c2s')
|
||
|
||
global testServerAliceRunning
|
||
global testServerBobRunning
|
||
testServerAliceRunning = False
|
||
testServerBobRunning = False
|
||
|
||
systemLanguage = 'en'
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
federationList = []
|
||
|
||
baseDir = os.getcwd()
|
||
if os.path.isdir(baseDir + '/.tests'):
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
os.mkdir(baseDir + '/.tests')
|
||
|
||
# create the servers
|
||
aliceDir = baseDir + '/.tests/alice'
|
||
aliceDomain = '127.0.0.42'
|
||
alicePort = 61935
|
||
aliceSendThreads = []
|
||
aliceAddress = aliceDomain + ':' + str(alicePort)
|
||
|
||
bobDir = baseDir + '/.tests/bob'
|
||
bobDomain = '127.0.0.64'
|
||
bobPort = 61936
|
||
bobSendThreads = []
|
||
bobAddress = bobDomain + ':' + str(bobPort)
|
||
|
||
global thrAlice
|
||
if thrAlice:
|
||
while thrAlice.is_alive():
|
||
thrAlice.stop()
|
||
time.sleep(1)
|
||
thrAlice.kill()
|
||
|
||
thrAlice = \
|
||
threadWithTrace(target=createServerAlice,
|
||
args=(aliceDir, aliceDomain, alicePort, bobAddress,
|
||
federationList, False, False,
|
||
aliceSendThreads),
|
||
daemon=True)
|
||
|
||
global thrBob
|
||
if thrBob:
|
||
while thrBob.is_alive():
|
||
thrBob.stop()
|
||
time.sleep(1)
|
||
thrBob.kill()
|
||
|
||
thrBob = \
|
||
threadWithTrace(target=createServerBob,
|
||
args=(bobDir, bobDomain, bobPort, aliceAddress,
|
||
federationList, False, False,
|
||
bobSendThreads),
|
||
daemon=True)
|
||
|
||
thrAlice.start()
|
||
thrBob.start()
|
||
assert thrAlice.is_alive() is True
|
||
assert thrBob.is_alive() is True
|
||
|
||
# wait for both servers to be running
|
||
ctr = 0
|
||
while not (testServerAliceRunning and testServerBobRunning):
|
||
time.sleep(1)
|
||
ctr += 1
|
||
if ctr > 60:
|
||
break
|
||
print('Alice online: ' + str(testServerAliceRunning))
|
||
print('Bob online: ' + str(testServerBobRunning))
|
||
|
||
time.sleep(1)
|
||
|
||
print('\n\n*******************************************************')
|
||
print('Alice sends to Bob via c2s')
|
||
|
||
sessionAlice = createSession(proxyType)
|
||
followersOnly = False
|
||
attachedImageFilename = baseDir + '/img/logo.png'
|
||
mediaType = getAttachmentMediaType(attachedImageFilename)
|
||
attachedImageDescription = 'Logo'
|
||
city = 'London, England'
|
||
isArticle = False
|
||
cachedWebfingers = {}
|
||
personCache = {}
|
||
password = 'alicepass'
|
||
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
|
||
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 0
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 0
|
||
sendResult = \
|
||
sendPostViaServer(__version__,
|
||
aliceDir, sessionAlice, 'alice', password,
|
||
aliceDomain, alicePort,
|
||
'bob', bobDomain, bobPort, None,
|
||
httpPrefix, 'Sent from my ActivityPub client',
|
||
followersOnly, True,
|
||
attachedImageFilename, mediaType,
|
||
attachedImageDescription, city,
|
||
cachedWebfingers, personCache, isArticle,
|
||
systemLanguage, True, None, None, None)
|
||
print('sendResult: ' + str(sendResult))
|
||
|
||
for i in range(30):
|
||
if os.path.isdir(outboxPath):
|
||
if len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 1:
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 1
|
||
print(">>> c2s post arrived in Alice's outbox")
|
||
|
||
for i in range(30):
|
||
if os.path.isdir(inboxPath):
|
||
if len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1:
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
|
||
print(">>> s2s post arrived in Bob's inbox")
|
||
print("c2s send success")
|
||
|
||
print('\n\nGetting message id for the post')
|
||
statusNumber = 0
|
||
outboxPostFilename = None
|
||
outboxPostId = None
|
||
for name in os.listdir(outboxPath):
|
||
if '#statuses#' in name:
|
||
statusNumber = name.split('#statuses#')[1].replace('.json', '')
|
||
statusNumber = int(statusNumber.replace('#activity', ''))
|
||
outboxPostFilename = outboxPath + '/' + name
|
||
postJsonObject = loadJson(outboxPostFilename, 0)
|
||
if postJsonObject:
|
||
outboxPostId = removeIdEnding(postJsonObject['id'])
|
||
assert outboxPostId
|
||
print('message id obtained: ' + outboxPostId)
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
|
||
print('\n\nAlice follows Bob')
|
||
sendFollowRequestViaServer(aliceDir, sessionAlice,
|
||
'alice', password,
|
||
aliceDomain, alicePort,
|
||
'bob', bobDomain, bobPort,
|
||
httpPrefix,
|
||
cachedWebfingers, personCache,
|
||
True, __version__)
|
||
alicePetnamesFilename = aliceDir + '/accounts/' + \
|
||
'alice@' + aliceDomain + '/petnames.txt'
|
||
aliceFollowingFilename = \
|
||
aliceDir + '/accounts/alice@' + aliceDomain + '/following.txt'
|
||
bobFollowersFilename = \
|
||
bobDir + '/accounts/bob@' + bobDomain + '/followers.txt'
|
||
for t in range(10):
|
||
if os.path.isfile(bobFollowersFilename):
|
||
if 'alice@' + aliceDomain + ':' + str(alicePort) in \
|
||
open(bobFollowersFilename).read():
|
||
if os.path.isfile(aliceFollowingFilename) and \
|
||
os.path.isfile(alicePetnamesFilename):
|
||
if 'bob@' + bobDomain + ':' + str(bobPort) in \
|
||
open(aliceFollowingFilename).read():
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert os.path.isfile(bobFollowersFilename)
|
||
assert os.path.isfile(aliceFollowingFilename)
|
||
assert os.path.isfile(alicePetnamesFilename)
|
||
assert 'bob bob@' + bobDomain in \
|
||
open(alicePetnamesFilename).read()
|
||
print('alice@' + aliceDomain + ':' + str(alicePort) + ' in ' +
|
||
bobFollowersFilename)
|
||
assert 'alice@' + aliceDomain + ':' + str(alicePort) in \
|
||
open(bobFollowersFilename).read()
|
||
print('bob@' + bobDomain + ':' + str(bobPort) + ' in ' +
|
||
aliceFollowingFilename)
|
||
assert 'bob@' + bobDomain + ':' + str(bobPort) in \
|
||
open(aliceFollowingFilename).read()
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
|
||
print('\n\nBob follows Alice')
|
||
sendFollowRequestViaServer(aliceDir, sessionAlice,
|
||
'bob', 'bobpass',
|
||
bobDomain, bobPort,
|
||
'alice', aliceDomain, alicePort,
|
||
httpPrefix,
|
||
cachedWebfingers, personCache,
|
||
True, __version__)
|
||
for t in range(10):
|
||
if os.path.isfile(aliceDir + '/accounts/alice@' + aliceDomain +
|
||
'/followers.txt'):
|
||
if 'bob@' + bobDomain + ':' + str(bobPort) in \
|
||
open(aliceDir + '/accounts/alice@' + aliceDomain +
|
||
'/followers.txt').read():
|
||
if os.path.isfile(bobDir + '/accounts/bob@' + bobDomain +
|
||
'/following.txt'):
|
||
aliceHandleStr = \
|
||
'alice@' + aliceDomain + ':' + str(alicePort)
|
||
if aliceHandleStr in \
|
||
open(bobDir + '/accounts/bob@' + bobDomain +
|
||
'/following.txt').read():
|
||
if os.path.isfile(bobDir + '/accounts/bob@' +
|
||
bobDomain +
|
||
'/followingCalendar.txt'):
|
||
if aliceHandleStr in \
|
||
open(bobDir + '/accounts/bob@' + bobDomain +
|
||
'/followingCalendar.txt').read():
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert os.path.isfile(aliceDir + '/accounts/alice@' + aliceDomain +
|
||
'/followers.txt')
|
||
assert os.path.isfile(bobDir + '/accounts/bob@' + bobDomain +
|
||
'/following.txt')
|
||
assert 'bob@' + bobDomain + ':' + str(bobPort) in \
|
||
open(aliceDir + '/accounts/alice@' + aliceDomain +
|
||
'/followers.txt').read()
|
||
assert 'alice@' + aliceDomain + ':' + str(alicePort) in \
|
||
open(bobDir + '/accounts/bob@' + bobDomain + '/following.txt').read()
|
||
|
||
print('\n\nBob likes the post')
|
||
sessionBob = createSession(proxyType)
|
||
password = 'bobpass'
|
||
outboxPath = bobDir + '/accounts/bob@' + bobDomain + '/outbox'
|
||
inboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/inbox'
|
||
print(str(len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))])))
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 1
|
||
print(str(len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])))
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
|
||
sendLikeViaServer(bobDir, sessionBob,
|
||
'bob', 'bobpass',
|
||
bobDomain, bobPort,
|
||
httpPrefix, outboxPostId,
|
||
cachedWebfingers, personCache,
|
||
True, __version__)
|
||
for i in range(20):
|
||
if os.path.isdir(outboxPath) and os.path.isdir(inboxPath):
|
||
if len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 2:
|
||
test = len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
if test == 1:
|
||
break
|
||
time.sleep(1)
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 2
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
|
||
print('Post liked')
|
||
|
||
print('\n\nBob repeats the post')
|
||
print(str(len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))])))
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 2
|
||
print(str(len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])))
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
|
||
sendAnnounceViaServer(bobDir, sessionBob, 'bob', password,
|
||
bobDomain, bobPort,
|
||
httpPrefix, outboxPostId,
|
||
cachedWebfingers,
|
||
personCache, True, __version__)
|
||
for i in range(20):
|
||
if os.path.isdir(outboxPath) and os.path.isdir(inboxPath):
|
||
if len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 3:
|
||
if len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath,
|
||
name))]) == 2:
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 3
|
||
assert len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))]) == 2
|
||
print('Post repeated')
|
||
|
||
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
|
||
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
|
||
postsBefore = \
|
||
len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
print('\n\nAlice deletes her post: ' + outboxPostId + ' ' +
|
||
str(postsBefore))
|
||
password = 'alicepass'
|
||
sendDeleteViaServer(aliceDir, sessionAlice, 'alice', password,
|
||
aliceDomain, alicePort,
|
||
httpPrefix, outboxPostId,
|
||
cachedWebfingers, personCache,
|
||
True, __version__)
|
||
for i in range(30):
|
||
if os.path.isdir(inboxPath):
|
||
test = len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
if test == postsBefore-1:
|
||
break
|
||
time.sleep(1)
|
||
|
||
test = len([name for name in os.listdir(inboxPath)
|
||
if os.path.isfile(os.path.join(inboxPath, name))])
|
||
assert test == postsBefore - 1
|
||
print(">>> post deleted from Alice's outbox and Bob's inbox")
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
|
||
print('\n\nAlice unfollows Bob')
|
||
password = 'alicepass'
|
||
sendUnfollowRequestViaServer(baseDir, sessionAlice,
|
||
'alice', password,
|
||
aliceDomain, alicePort,
|
||
'bob', bobDomain, bobPort,
|
||
httpPrefix,
|
||
cachedWebfingers, personCache,
|
||
True, __version__)
|
||
for t in range(10):
|
||
if 'alice@' + aliceDomain + ':' + str(alicePort) not in \
|
||
open(bobFollowersFilename).read():
|
||
if 'bob@' + bobDomain + ':' + str(bobPort) not in \
|
||
open(aliceFollowingFilename).read():
|
||
break
|
||
time.sleep(1)
|
||
|
||
assert os.path.isfile(bobFollowersFilename)
|
||
assert os.path.isfile(aliceFollowingFilename)
|
||
assert 'alice@' + aliceDomain + ':' + str(alicePort) \
|
||
not in open(bobFollowersFilename).read()
|
||
assert 'bob@' + bobDomain + ':' + str(bobPort) \
|
||
not in open(aliceFollowingFilename).read()
|
||
assert validInbox(bobDir, 'bob', bobDomain)
|
||
assert validInboxFilenames(bobDir, 'bob', bobDomain,
|
||
aliceDomain, alicePort)
|
||
assert validInbox(aliceDir, 'alice', aliceDomain)
|
||
assert validInboxFilenames(aliceDir, 'alice', aliceDomain,
|
||
bobDomain, bobPort)
|
||
|
||
# stop the servers
|
||
thrAlice.kill()
|
||
thrAlice.join()
|
||
assert thrAlice.is_alive() is False
|
||
|
||
thrBob.kill()
|
||
thrBob.join()
|
||
assert thrBob.is_alive() is False
|
||
|
||
os.chdir(baseDir)
|
||
# shutil.rmtree(aliceDir)
|
||
# shutil.rmtree(bobDir)
|
||
|
||
|
||
def _testActorParsing():
|
||
print('testActorParsing')
|
||
actor = 'https://mydomain:72/users/mynick'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'mydomain'
|
||
assert port == 72
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'mynick'
|
||
|
||
actor = 'https://element/accounts/badger'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'element'
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'badger'
|
||
|
||
actor = 'egg@chicken.com'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'chicken.com'
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'egg'
|
||
|
||
actor = '@waffle@cardboard'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'cardboard'
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'waffle'
|
||
|
||
actor = 'https://astral/channel/sky'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'astral'
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'sky'
|
||
|
||
actor = 'https://randomain/users/rando'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'randomain'
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'rando'
|
||
|
||
actor = 'https://otherdomain:49/@othernick'
|
||
domain, port = getDomainFromActor(actor)
|
||
assert domain == 'otherdomain'
|
||
assert port == 49
|
||
nickname = getNicknameFromActor(actor)
|
||
assert nickname == 'othernick'
|
||
|
||
|
||
def _testWebLinks():
|
||
print('testWebLinks')
|
||
|
||
exampleText = \
|
||
"<p>Aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" + \
|
||
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" + \
|
||
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" + \
|
||
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" + \
|
||
" <a href=\"https://domain.ugh/tags/turbot\" class=\"mention " + \
|
||
"hashtag\" rel=\"tag\">#<span>turbot</span></a> <a href=\"" + \
|
||
"https://domain.ugh/tags/haddock\" class=\"mention hashtag\"" + \
|
||
" rel=\"tag\">#<span>haddock</span></a></p>"
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == "<p>Aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" + \
|
||
" <a href=\"https://domain.ugh/tags/turbot\" class=\"mention " + \
|
||
"hashtag\" rel=\"tag\">#<span>turbot</span></a> " + \
|
||
"<a href=\"https://domain.ugh/tags/haddock\" " + \
|
||
"class=\"mention hashtag\" rel=\"tag\">#<span>haddock</span></a></p>"
|
||
|
||
exampleText = \
|
||
'<p><span class=\"h-card\"><a href=\"https://something/@orother' + \
|
||
'\" class=\"u-url mention\">@<span>foo</span></a></span> Some ' + \
|
||
'random text.</p><p>AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' + \
|
||
'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' + \
|
||
'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' + \
|
||
'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' + \
|
||
'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA' + \
|
||
'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA</p>'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == \
|
||
'<p><span class="h-card"><a href="https://something/@orother"' + \
|
||
' class="u-url mention">@<span>foo</span></a></span> ' + \
|
||
'Some random text.</p>'
|
||
|
||
exampleText = \
|
||
'This post has a web links https://somesite.net\n\nAnd some other text'
|
||
linkedText = addWebLinks(exampleText)
|
||
assert \
|
||
'<a href="https://somesite.net" rel="nofollow noopener noreferrer"' + \
|
||
' target="_blank"><span class="invisible">https://' + \
|
||
'</span><span class="ellipsis">somesite.net</span></a' in linkedText
|
||
|
||
exampleText = \
|
||
'This post has a very long web link\n\nhttp://' + \
|
||
'cbwebewuvfuftdiudbqd33dddbbyuef23fyug3bfhcyu2fct2' + \
|
||
'cuyqbcbucuwvckiwyfgewfvqejbchevbhwevuevwbqebqekve' + \
|
||
'qvuvjfkf.onion\n\nAnd some other text'
|
||
linkedText = addWebLinks(exampleText)
|
||
assert 'ellipsis' in linkedText
|
||
|
||
exampleText = \
|
||
'<p>1. HAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAH' + \
|
||
'AHAHAHHAHAHAHAHAHAHAHAHAHAHAHAHHAHAHAHAHAHAHAHAH</p>'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == '<p>1. HAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHAHA</p>'
|
||
|
||
exampleText = \
|
||
'<p>Tox address is 88AB9DED6F9FBEF43E105FB72060A2D89F9B93C74' + \
|
||
'4E8C45AB3C5E42C361C837155AFCFD9D448 </p>'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == exampleText
|
||
|
||
exampleText = \
|
||
'some.incredibly.long.and.annoying.word.which.should.be.removed: ' + \
|
||
'The remaining text'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == \
|
||
'some.incredibly.long.and.annoying.word.w\n' + \
|
||
'hich.should.be.removed: The remaining text'
|
||
|
||
exampleText = \
|
||
'<p>Tox address is 88AB9DED6F9FBEF43E105FB72060A2D89F9B93C74' + \
|
||
'4E8C45AB3C5E42C361C837155AFCFD9D448</p>'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == \
|
||
'<p>Tox address is 88AB9DED6F9FBEF43E105FB72060A2D89F9B93C7\n' + \
|
||
'44E8C45AB3C5E42C361C837155AFCFD9D448</p>'
|
||
|
||
exampleText = \
|
||
'<p>ABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCA' + \
|
||
'BCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCAB' + \
|
||
'CABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABC' + \
|
||
'ABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCA' + \
|
||
'BCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCAB' + \
|
||
'CABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABC' + \
|
||
'ABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCA' + \
|
||
'BCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCABCAB' + \
|
||
'CABCABCABCABCABCABCABCABC</p>'
|
||
resultText = removeLongWords(exampleText, 40, [])
|
||
assert resultText == r'<p>ABCABCABCABCABCABCABCABCABCABCABCABCABCA<\p>'
|
||
|
||
exampleText = \
|
||
'"the nucleus of mutual-support institutions, habits, and customs ' + \
|
||
'remains alive with the millions; it keeps them together; and ' + \
|
||
'they prefer to cling to their customs, beliefs, and traditions ' + \
|
||
'rather than to accept the teachings of a war of each ' + \
|
||
'against all"\n\n--Peter Kropotkin'
|
||
testFnStr = addWebLinks(exampleText)
|
||
resultText = removeLongWords(testFnStr, 40, [])
|
||
assert resultText == exampleText
|
||
assert 'ellipsis' not in resultText
|
||
|
||
exampleText = \
|
||
'<p>ｆｉｌｅｐｏｐｏｕｔ＝' + \
|
||
'ＴｅｍｐｌａｔｅＡｔｔａｃｈｍｅｎｔＲｉｃｈＰｏｐｏｕｔ<<\\p>'
|
||
resultText = replaceContentDuplicates(exampleText)
|
||
assert resultText == \
|
||
'<p>ｆｉｌｅｐｏｐｏｕｔ＝' + \
|
||
'ＴｅｍｐｌａｔｅＡｔｔａｃｈｍｅｎｔＲｉｃｈＰｏｐｏｕｔ'
|
||
|
||
exampleText = \
|
||
'<p>Test1 test2 #YetAnotherExcessivelyLongwindedAndBoringHashtag</p>'
|
||
testFnStr = addWebLinks(exampleText)
|
||
resultText = removeLongWords(testFnStr, 40, [])
|
||
assert(resultText ==
|
||
'<p>Test1 test2 '
|
||
'#YetAnotherExcessivelyLongwindedAndBorin\ngHashtag</p>')
|
||
|
||
exampleText = \
|
||
"<p>Don't remove a p2p link " + \
|
||
"rad:git:hwd1yrerc3mcgn8ga9rho3dqi4w33nep7kxmqezss4topyfgmexihp" + \
|
||
"33xcw</p>"
|
||
testFnStr = addWebLinks(exampleText)
|
||
resultText = removeLongWords(testFnStr, 40, [])
|
||
assert resultText == exampleText
|
||
|
||
|
||
def _testAddEmoji():
|
||
print('testAddEmoji')
|
||
content = "Emoji :lemon: :strawberry: :banana:"
|
||
httpPrefix = 'http'
|
||
nickname = 'testuser'
|
||
domain = 'testdomain.net'
|
||
port = 3682
|
||
recipients = []
|
||
hashtags = {}
|
||
baseDir = os.getcwd()
|
||
baseDirOriginal = os.getcwd()
|
||
path = baseDir + '/.tests'
|
||
if not os.path.isdir(path):
|
||
os.mkdir(path)
|
||
path = baseDir + '/.tests/emoji'
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
baseDir = path
|
||
path = baseDir + '/emoji'
|
||
if os.path.isdir(path):
|
||
shutil.rmtree(path)
|
||
os.mkdir(path)
|
||
copytree(baseDirOriginal + '/emoji', baseDir + '/emoji')
|
||
os.chdir(baseDir)
|
||
privateKeyPem, publicKeyPem, person, wfEndpoint = \
|
||
createPerson(baseDir, nickname, domain, port,
|
||
httpPrefix, True, False, 'password')
|
||
contentModified = \
|
||
addHtmlTags(baseDir, httpPrefix,
|
||
nickname, domain, content,
|
||
recipients, hashtags, True)
|
||
assert ':lemon:' in contentModified
|
||
assert contentModified.startswith('<p>')
|
||
assert contentModified.endswith('</p>')
|
||
tags = []
|
||
for tagName, tag in hashtags.items():
|
||
tags.append(tag)
|
||
content = contentModified
|
||
contentModified = replaceEmojiFromTags(content, tags, 'content')
|
||
# print('contentModified: ' + contentModified)
|
||
assert contentModified == '<p>Emoji 🍋 🍓 🍌</p>'
|
||
|
||
os.chdir(baseDirOriginal)
|
||
shutil.rmtree(baseDirOriginal + '/.tests')
|
||
|
||
|
||
def _testGetStatusNumber():
|
||
print('testGetStatusNumber')
|
||
prevStatusNumber = None
|
||
for i in range(1, 20):
|
||
statusNumber, published = getStatusNumber()
|
||
if prevStatusNumber:
|
||
assert len(statusNumber) == 18
|
||
assert int(statusNumber) > prevStatusNumber
|
||
prevStatusNumber = int(statusNumber)
|
||
|
||
|
||
def _testJsonString() -> None:
|
||
print('testJsonString')
|
||
filename = '.epicyon_tests_testJsonString.json'
|
||
messageStr = "Crème brûlée यह एक परीक्षण ह"
|
||
testJson = {
|
||
"content": messageStr
|
||
}
|
||
assert saveJson(testJson, filename)
|
||
receivedJson = loadJson(filename, 0)
|
||
assert receivedJson
|
||
assert receivedJson['content'] == messageStr
|
||
encodedStr = json.dumps(testJson, ensure_ascii=False)
|
||
assert messageStr in encodedStr
|
||
os.remove(filename)
|
||
|
||
|
||
def _testSaveLoadJson():
|
||
print('testSaveLoadJson')
|
||
testJson = {
|
||
"param1": 3,
|
||
"param2": '"Crème brûlée यह एक परीक्षण ह"'
|
||
}
|
||
testFilename = '.epicyon_tests_testSaveLoadJson.json'
|
||
if os.path.isfile(testFilename):
|
||
os.remove(testFilename)
|
||
assert saveJson(testJson, testFilename)
|
||
assert os.path.isfile(testFilename)
|
||
testLoadJson = loadJson(testFilename)
|
||
assert(testLoadJson)
|
||
assert testLoadJson.get('param1')
|
||
assert testLoadJson.get('param2')
|
||
assert testLoadJson['param1'] == 3
|
||
assert testLoadJson['param2'] == '"Crème brûlée यह एक परीक्षण ह"'
|
||
os.remove(testFilename)
|
||
|
||
|
||
def _testTheme():
|
||
print('testTheme')
|
||
css = 'somestring --background-value: 24px; --foreground-value: 24px;'
|
||
result = setCSSparam(css, 'background-value', '32px')
|
||
assert result == \
|
||
'somestring --background-value: 32px; --foreground-value: 24px;'
|
||
css = \
|
||
'somestring --background-value: 24px; --foreground-value: 24px; ' + \
|
||
'--background-value: 24px;'
|
||
result = setCSSparam(css, 'background-value', '32px')
|
||
assert result == \
|
||
'somestring --background-value: 32px; --foreground-value: 24px; ' + \
|
||
'--background-value: 32px;'
|
||
css = '--background-value: 24px; --foreground-value: 24px;'
|
||
result = setCSSparam(css, 'background-value', '32px')
|
||
assert result == '--background-value: 32px; --foreground-value: 24px;'
|
||
|
||
|
||
def _testRecentPostsCache():
|
||
print('testRecentPostsCache')
|
||
recentPostsCache = {}
|
||
maxRecentPosts = 3
|
||
htmlStr = '<html></html>'
|
||
for i in range(5):
|
||
postJsonObject = {
|
||
"id": "https://somesite.whatever/users/someuser/statuses/" + str(i)
|
||
}
|
||
updateRecentPostsCache(recentPostsCache, maxRecentPosts,
|
||
postJsonObject, htmlStr)
|
||
assert len(recentPostsCache['index']) == maxRecentPosts
|
||
assert len(recentPostsCache['json'].items()) == maxRecentPosts
|
||
assert len(recentPostsCache['html'].items()) == maxRecentPosts
|
||
|
||
|
||
def _testRemoveTextFormatting():
|
||
print('testRemoveTextFormatting')
|
||
testStr = '<p>Text without formatting</p>'
|
||
resultStr = removeTextFormatting(testStr)
|
||
assert(resultStr == testStr)
|
||
testStr = '<p>Text <i>with</i> <h3>formatting</h3></p>'
|
||
resultStr = removeTextFormatting(testStr)
|
||
assert(resultStr == '<p>Text with formatting</p>')
|
||
|
||
|
||
def _testJsonld():
|
||
print("testJsonld")
|
||
|
||
jldDocument = {
|
||
"@context": "https://www.w3.org/ns/activitystreams",
|
||
"actor": "https://somesite.net/users/gerbil",
|
||
"description": "My json document",
|
||
"numberField": 83582,
|
||
"object": {
|
||
"content": "valid content"
|
||
}
|
||
}
|
||
# privateKeyPem, publicKeyPem = generateRSAKey()
|
||
privateKeyPem = '-----BEGIN RSA PRIVATE KEY-----\n' \
|
||
'MIIEowIBAAKCAQEAod9iHfIn4ugY/2byFrFjUprrFLkkH5bCrjiBq2/MdHFg99IQ\n' \
|
||
'7li2x2mg5fkBMhU5SJIxlN8kiZMFq7JUXSA97Yo4puhVubqTSHihIh6Xn2mTjTgs\n' \
|
||
'zNo9SBbmN3YiyBPTcr0rF4jGWZAduJ8u6i7Eky2QH+UBKyUNRZrcfoVq+7grHUIA\n' \
|
||
'45pE7vAfEEWtgRiw32Nwlx55N3hayHax0y8gMdKEF/vfYKRLcM7rZgEASMtlCpgy\n' \
|
||
'fsyHwFCDzl/BP8AhP9u3dM+SEundeAvF58AiXx1pKvBpxqttDNAsKWCRQ06/WI/W\n' \
|
||
'2Rwihl9yCjobqRoFsZ/cTEi6FG9AbDAds5YjTwIDAQABAoIBAERL3rbpy8Bl0t43\n' \
|
||
'jh7a+yAIMvVMZBxb3InrV3KAug/LInGNFQ2rKnsaawN8uu9pmwCuhfLc7yqIeJUH\n' \
|
||
'qaadCuPlNJ/fWQQC309tbfbaV3iv78xejjBkSATZfIqb8nLeQpGflMXaNG3na1LQ\n' \
|
||
'/tdZoiDC0ZNTaNnOSTo765oKKqhHUTQkwkGChrwG3Js5jekV4zpPMLhUafXk6ksd\n' \
|
||
'8XLlZdCF3RUnuguXAg2xP/duxMYmTCx3eeGPkXBPQl0pahu8/6OtBoYvBrqNdQcx\n' \
|
||
'jnEtYX9PCqDY3hAXW9GWsxNfu02DKhWigFHFNRUQtMI++438+QIfzXPslE2bTQIt\n' \
|
||
'0OXUlwECgYEAxTKUZ7lwIBb5XKPJq53RQmX66M3ArxI1RzFSKm1+/CmxvYiN0c+5\n' \
|
||
'2Aq62WEIauX6hoZ7yQb4zhdeNRzinLR7rsmBvIcP12FidXG37q9v3Vu70KmHniJE\n' \
|
||
'TPbt5lHQ0bNACFxkar4Ab/JZN4CkMRgJdlcZ5boYNmcGOYCvw9izuM8CgYEA0iQ1\n' \
|
||
'khIFZ6fCiXwVRGvEHmqSnkBmBHz8MY8fczv2Z4Gzfq3Tlh9VxpigK2F2pFt7keWc\n' \
|
||
'53HerYFHFpf5otDhEyRwA1LyIcwbj5HopumxsB2WG+/M2as45lLfWa6KO73OtPpU\n' \
|
||
'wGZYW+i/otdk9eFphceYtw19mxI+3lYoeI8EjYECgYBxOtTKJkmCs45lqkp/d3QT\n' \
|
||
'2zjSempcXGkpQuG6KPtUUaCUgxdj1RISQj792OCbeQh8PDZRvOYaeIKInthkQKIQ\n' \
|
||
'P/Z1yVvIQUvmwfBqZmQmR6k1bFLJ80UiqFr7+BiegH2RD3Q9cnIP1aly3DPrWLD+\n' \
|
||
'OY9OQKfsfQWu+PxzyTeRMwKBgD8Zjlh5PtQ8RKcB8mTkMzSq7bHFRpzsZtH+1wPE\n' \
|
||
'Kp40DRDp41H9wMTsiZPdJUH/EmDh4LaCs8nHuu/m3JfuPtd/pn7pBjntzwzSVFji\n' \
|
||
'bW+jwrJK1Gk8B87pbZXBWlLMEOi5Dn/je37Fqd2c7f0DHauFHq9AxsmsteIPXwGs\n' \
|
||
'eEKBAoGBAIzJX/5yFp3ObkPracIfOJ/U/HF1UdP6Y8qmOJBZOg5s9Y+JAdY76raK\n' \
|
||
'0SbZPsOpuFUdTiRkSI3w/p1IuM5dPxgCGH9MHqjqogU5QwXr3vLF+a/PFhINkn1x\n' \
|
||
'lozRZjDcF1y6xHfExotPC973UZnKEviq9/FqOsovZpvSQkzAYSZF\n' \
|
||
'-----END RSA PRIVATE KEY-----'
|
||
publicKeyPem = '-----BEGIN PUBLIC KEY-----\n' \
|
||
'MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAod9iHfIn4ugY/2byFrFj\n' \
|
||
'UprrFLkkH5bCrjiBq2/MdHFg99IQ7li2x2mg5fkBMhU5SJIxlN8kiZMFq7JUXSA9\n' \
|
||
'7Yo4puhVubqTSHihIh6Xn2mTjTgszNo9SBbmN3YiyBPTcr0rF4jGWZAduJ8u6i7E\n' \
|
||
'ky2QH+UBKyUNRZrcfoVq+7grHUIA45pE7vAfEEWtgRiw32Nwlx55N3hayHax0y8g\n' \
|
||
'MdKEF/vfYKRLcM7rZgEASMtlCpgyfsyHwFCDzl/BP8AhP9u3dM+SEundeAvF58Ai\n' \
|
||
'Xx1pKvBpxqttDNAsKWCRQ06/WI/W2Rwihl9yCjobqRoFsZ/cTEi6FG9AbDAds5Yj\n' \
|
||
'TwIDAQAB\n' \
|
||
'-----END PUBLIC KEY-----'
|
||
|
||
signedDocument = jldDocument.copy()
|
||
generateJsonSignature(signedDocument, privateKeyPem)
|
||
assert(signedDocument)
|
||
assert(signedDocument.get('signature'))
|
||
assert(signedDocument['signature'].get('signatureValue'))
|
||
assert(signedDocument['signature'].get('type'))
|
||
assert(len(signedDocument['signature']['signatureValue']) > 50)
|
||
# print(str(signedDocument['signature']))
|
||
assert(signedDocument['signature']['type'] == 'RsaSignature2017')
|
||
assert(verifyJsonSignature(signedDocument, publicKeyPem))
|
||
|
||
# alter the signed document
|
||
signedDocument['object']['content'] = 'forged content'
|
||
assert(not verifyJsonSignature(signedDocument, publicKeyPem))
|
||
|
||
jldDocument2 = {
|
||
"@context": "https://www.w3.org/ns/activitystreams",
|
||
"actor": "https://somesite.net/users/gerbil",
|
||
"description": "Another json document",
|
||
"numberField": 13353,
|
||
"object": {
|
||
"content": "More content"
|
||
}
|
||
}
|
||
signedDocument2 = jldDocument2.copy()
|
||
generateJsonSignature(signedDocument2, privateKeyPem)
|
||
assert(signedDocument2)
|
||
assert(signedDocument2.get('signature'))
|
||
assert(signedDocument2['signature'].get('signatureValue'))
|
||
# changed signature on different document
|
||
if signedDocument['signature']['signatureValue'] == \
|
||
signedDocument2['signature']['signatureValue']:
|
||
print('json signature has not changed for different documents')
|
||
assert '.' not in str(signedDocument['signature']['signatureValue'])
|
||
assert len(str(signedDocument['signature']['signatureValue'])) > 340
|
||
assert(signedDocument['signature']['signatureValue'] !=
|
||
signedDocument2['signature']['signatureValue'])
|
||
|
||
|
||
def _testSiteIsActive():
|
||
print('testSiteIsActive')
|
||
assert(siteIsActive('https://archive.org'))
|
||
assert(siteIsActive('https://mastodon.social'))
|
||
assert(not siteIsActive('https://notarealwebsite.a.b.c'))
|
||
|
||
|
||
def _testRemoveHtml():
|
||
print('testRemoveHtml')
|
||
testStr = 'This string has no html.'
|
||
assert(removeHtml(testStr) == testStr)
|
||
testStr = 'This string <a href="1234.567">has html</a>.'
|
||
assert(removeHtml(testStr) == 'This string has html.')
|
||
testStr = '<label>This string has.</label><label>Two labels.</label>'
|
||
assert(removeHtml(testStr) == 'This string has. Two labels.')
|
||
testStr = '<p>This string has.</p><p>Two paragraphs.</p>'
|
||
assert(removeHtml(testStr) == 'This string has.\n\nTwo paragraphs.')
|
||
testStr = 'This string has.<br>A new line.'
|
||
assert(removeHtml(testStr) == 'This string has.\nA new line.')
|
||
testStr = '<p>This string contains a url http://somesite.or.other</p>'
|
||
assert(removeHtml(testStr) ==
|
||
'This string contains a url http://somesite.or.other')
|
||
|
||
|
||
def _testDangerousCSS():
|
||
print('testDangerousCSS')
|
||
baseDir = os.getcwd()
|
||
for subdir, dirs, files in os.walk(baseDir):
|
||
for f in files:
|
||
if not f.endswith('.css'):
|
||
continue
|
||
assert not dangerousCSS(baseDir + '/' + f, False)
|
||
break
|
||
|
||
|
||
def _testDangerousMarkup():
|
||
print('testDangerousMarkup')
|
||
allowLocalNetworkAccess = False
|
||
content = '<p>This is a valid message</p>'
|
||
assert(not dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = 'This is a valid message without markup'
|
||
assert(not dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This is a valid-looking message. But wait... ' + \
|
||
'<script>document.getElementById("concentrated")' + \
|
||
'.innerHTML = "evil";</script></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This is a valid-looking message. But wait... ' + \
|
||
'<script>document.getElementById("concentrated")' + \
|
||
'.innerHTML = "evil";</script></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This html contains more than you expected... ' + \
|
||
'<script language="javascript">document.getElementById("abc")' + \
|
||
'.innerHTML = "def";</script></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This is a valid-looking message. But wait... ' + \
|
||
'<script src="https://evilsite/payload.js" /></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message embeds an evil frame.' + \
|
||
'<iframe src="somesite"></iframe></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message tries to obfuscate an evil frame.' + \
|
||
'< iframe src = "somesite"></ iframe ></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message is not necessarily evil, but annoying.' + \
|
||
'<hr><br><br><br><br><br><br><br><hr><hr></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message contans a ' + \
|
||
'<a href="https://validsite/index.html">valid link.</a></p>'
|
||
assert(not dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message contans a ' + \
|
||
'<a href="https://validsite/iframe.html">' + \
|
||
'valid link having invalid but harmless name.</a></p>'
|
||
assert(not dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message which <a href="127.0.0.1:8736">' + \
|
||
'tries to access the local network</a></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>This message which <a href="http://192.168.5.10:7235">' + \
|
||
'tries to access the local network</a></p>'
|
||
assert(dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
content = '<p>127.0.0.1 This message which does not access ' + \
|
||
'the local network</a></p>'
|
||
assert(not dangerousMarkup(content, allowLocalNetworkAccess))
|
||
|
||
|
||
def _runHtmlReplaceQuoteMarks():
|
||
print('htmlReplaceQuoteMarks')
|
||
testStr = 'The "cat" "sat" on the mat'
|
||
result = htmlReplaceQuoteMarks(testStr)
|
||
assert result == 'The “cat” “sat” on the mat'
|
||
|
||
testStr = 'The cat sat on the mat'
|
||
result = htmlReplaceQuoteMarks(testStr)
|
||
assert result == 'The cat sat on the mat'
|
||
|
||
testStr = '"hello"'
|
||
result = htmlReplaceQuoteMarks(testStr)
|
||
assert result == '“hello”'
|
||
|
||
testStr = '"hello" <a href="somesite.html">"test" html</a>'
|
||
result = htmlReplaceQuoteMarks(testStr)
|
||
assert result == '“hello” <a href="somesite.html">“test” html</a>'
|
||
|
||
|
||
def _testJsonPostAllowsComments():
|
||
print('testJsonPostAllowsComments')
|
||
postJsonObject = {
|
||
"id": "123"
|
||
}
|
||
assert jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"commentsEnabled": False
|
||
}
|
||
assert not jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"rejectReplies": False
|
||
}
|
||
assert jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"rejectReplies": True
|
||
}
|
||
assert not jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"commentsEnabled": True
|
||
}
|
||
assert jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"object": {
|
||
"commentsEnabled": True
|
||
}
|
||
}
|
||
assert jsonPostAllowsComments(postJsonObject)
|
||
postJsonObject = {
|
||
"id": "123",
|
||
"object": {
|
||
"commentsEnabled": False
|
||
}
|
||
}
|
||
assert not jsonPostAllowsComments(postJsonObject)
|
||
|
||
|
||
def _testRemoveIdEnding():
|
||
print('testRemoveIdEnding')
|
||
testStr = 'https://activitypub.somedomain.net'
|
||
resultStr = removeIdEnding(testStr)
|
||
assert resultStr == 'https://activitypub.somedomain.net'
|
||
|
||
testStr = \
|
||
'https://activitypub.somedomain.net/users/foo/' + \
|
||
'statuses/34544814814/activity'
|
||
resultStr = removeIdEnding(testStr)
|
||
assert resultStr == \
|
||
'https://activitypub.somedomain.net/users/foo/statuses/34544814814'
|
||
|
||
testStr = \
|
||
'https://undo.somedomain.net/users/foo/statuses/34544814814/undo'
|
||
resultStr = removeIdEnding(testStr)
|
||
assert resultStr == \
|
||
'https://undo.somedomain.net/users/foo/statuses/34544814814'
|
||
|
||
testStr = \
|
||
'https://event.somedomain.net/users/foo/statuses/34544814814/event'
|
||
resultStr = removeIdEnding(testStr)
|
||
assert resultStr == \
|
||
'https://event.somedomain.net/users/foo/statuses/34544814814'
|
||
|
||
|
||
def _testValidContentWarning():
|
||
print('testValidContentWarning')
|
||
resultStr = validContentWarning('Valid content warning')
|
||
assert resultStr == 'Valid content warning'
|
||
|
||
resultStr = validContentWarning('Invalid #content warning')
|
||
assert resultStr == 'Invalid content warning'
|
||
|
||
resultStr = \
|
||
validContentWarning('Invalid <a href="somesite">content warning</a>')
|
||
assert resultStr == 'Invalid content warning'
|
||
|
||
|
||
def _testTranslations():
|
||
print('testTranslations')
|
||
languagesStr = ('ar', 'ca', 'cy', 'de', 'es', 'fr', 'ga',
|
||
'hi', 'it', 'ja', 'oc', 'pt', 'ru', 'zh')
|
||
|
||
# load all translations into a dict
|
||
langDict = {}
|
||
for lang in languagesStr:
|
||
langJson = loadJson('translations/' + lang + '.json')
|
||
if not langJson:
|
||
print('Missing language file ' +
|
||
'translations/' + lang + '.json')
|
||
assert langJson
|
||
langDict[lang] = langJson
|
||
|
||
# load english translations
|
||
translationsJson = loadJson('translations/en.json')
|
||
# test each english string exists in the other language files
|
||
for englishStr, translatedStr in translationsJson.items():
|
||
for lang in languagesStr:
|
||
langJson = langDict[lang]
|
||
if not langJson.get(englishStr):
|
||
print(englishStr + ' is missing from ' + lang + '.json')
|
||
assert langJson.get(englishStr)
|
||
|
||
|
||
def _testConstantTimeStringCheck():
|
||
print('testConstantTimeStringCheck')
|
||
assert constantTimeStringCheck('testing', 'testing')
|
||
assert not constantTimeStringCheck('testing', '1234')
|
||
assert not constantTimeStringCheck('testing', '1234567')
|
||
|
||
itterations = 256
|
||
|
||
start = time.time()
|
||
for timingTest in range(itterations):
|
||
constantTimeStringCheck('nnjfbefefbsnjsdnvbcueftqfeuqfbqefnjeniwufgy',
|
||
'nnjfbefefbsnjsdnvbcueftqfeuqfbqefnjeniwufgy')
|
||
end = time.time()
|
||
avTime1 = ((end - start) * 1000000 / itterations)
|
||
|
||
# change a single character and observe timing difference
|
||
start = time.time()
|
||
for timingTest in range(itterations):
|
||
constantTimeStringCheck('nnjfbefefbsnjsdnvbcueftqfeuqfbqefnjeniwufgy',
|
||
'nnjfbefefbsnjsdnvbcueftqfeuqfbqeznjeniwufgy')
|
||
end = time.time()
|
||
avTime2 = ((end - start) * 1000000 / itterations)
|
||
timeDiffMicroseconds = abs(avTime2 - avTime1)
|
||
# time difference should be less than 10uS
|
||
assert int(timeDiffMicroseconds) < 10
|
||
|
||
# change multiple characters and observe timing difference
|
||
start = time.time()
|
||
for timingTest in range(itterations):
|
||
constantTimeStringCheck('nnjfbefefbsnjsdnvbcueftqfeuqfbqefnjeniwufgy',
|
||
'ano1befffbsn7sd3vbluef6qseuqfpqeznjgni9bfgi')
|
||
end = time.time()
|
||
avTime2 = ((end - start) * 1000000 / itterations)
|
||
timeDiffMicroseconds = abs(avTime2 - avTime1)
|
||
# time difference should be less than 10uS
|
||
assert int(timeDiffMicroseconds) < 10
|
||
|
||
|
||
def _testReplaceEmailQuote():
|
||
print('testReplaceEmailQuote')
|
||
testStr = '<p>This content has no quote.</p>'
|
||
assert htmlReplaceEmailQuote(testStr) == testStr
|
||
|
||
testStr = '<p>This content has no quote.</p>' + \
|
||
'<p>With multiple</p><p>lines</p>'
|
||
assert htmlReplaceEmailQuote(testStr) == testStr
|
||
|
||
testStr = '<p>"This is a quoted paragraph."</p>'
|
||
assert htmlReplaceEmailQuote(testStr) == \
|
||
'<p><blockquote>This is a quoted paragraph.</blockquote></p>'
|
||
|
||
testStr = "<p><span class=\"h-card\">" + \
|
||
"<a href=\"https://somewebsite/@nickname\" " + \
|
||
"class=\"u-url mention\">@<span>nickname</span></a></span> " + \
|
||
"<br />> This is a quote</p><p>Some other text.</p>"
|
||
expectedStr = "<p><span class=\"h-card\">" + \
|
||
"<a href=\"https://somewebsite/@nickname\" " + \
|
||
"class=\"u-url mention\">@<span>nickname</span></a></span> " + \
|
||
"<br /><blockquote>This is a quote</blockquote></p>" + \
|
||
"<p>Some other text.</p>"
|
||
resultStr = htmlReplaceEmailQuote(testStr)
|
||
if resultStr != expectedStr:
|
||
print('Result: ' + str(resultStr))
|
||
print('Expect: ' + expectedStr)
|
||
assert resultStr == expectedStr
|
||
|
||
testStr = "<p>Some text:</p><p>> first line->second line</p>" + \
|
||
"<p>Some question?</p>"
|
||
expectedStr = "<p>Some text:</p><p><blockquote>first line-<br>" + \
|
||
"second line</blockquote></p><p>Some question?</p>"
|
||
resultStr = htmlReplaceEmailQuote(testStr)
|
||
if resultStr != expectedStr:
|
||
print('Result: ' + str(resultStr))
|
||
print('Expect: ' + expectedStr)
|
||
assert resultStr == expectedStr
|
||
|
||
testStr = "<p><span class=\"h-card\">" + \
|
||
"<a href=\"https://somedomain/@somenick\" " + \
|
||
"class=\"u-url mention\">@<span>somenick</span>" + \
|
||
"</a></span> </p><p>> Text1.<br />> <br />" + \
|
||
"> Text2<br />> <br />> Text3<br />" + \
|
||
"><br />> Text4<br />> <br />> " + \
|
||
"Text5<br />> <br />> Text6</p><p>Text7</p>"
|
||
expectedStr = "<p><span class=\"h-card\">" + \
|
||
"<a href=\"https://somedomain/@somenick\" " + \
|
||
"class=\"u-url mention\">@<span>somenick</span></a>" + \
|
||
"</span> </p><p><blockquote> Text1.<br /><br />" + \
|
||
"Text2<br /><br />Text3<br />><br />Text4<br />" + \
|
||
"<br />Text5<br /><br />Text6</blockquote></p><p>Text7</p>"
|
||
resultStr = htmlReplaceEmailQuote(testStr)
|
||
if resultStr != expectedStr:
|
||
print('Result: ' + str(resultStr))
|
||
print('Expect: ' + expectedStr)
|
||
assert resultStr == expectedStr
|
||
|
||
|
||
def _testRemoveHtmlTag():
|
||
print('testRemoveHtmlTag')
|
||
testStr = "<p><img width=\"864\" height=\"486\" " + \
|
||
"src=\"https://somesiteorother.com/image.jpg\"></p>"
|
||
resultStr = removeHtmlTag(testStr, 'width')
|
||
assert resultStr == "<p><img height=\"486\" " + \
|
||
"src=\"https://somesiteorother.com/image.jpg\"></p>"
|
||
|
||
|
||
def _testHashtagRuleTree():
|
||
print('testHashtagRuleTree')
|
||
operators = ('not', 'and', 'or', 'xor', 'from', 'contains')
|
||
|
||
url = 'testsite.com'
|
||
moderated = True
|
||
conditionsStr = \
|
||
'contains "Cat" or contains "Corvid" or ' + \
|
||
'contains "Dormouse" or contains "Buzzard"'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['or', ['contains', ['"Cat"']],
|
||
['contains', ['"Corvid"']],
|
||
['contains', ['"Dormouse"']],
|
||
['contains', ['"Buzzard"']]])
|
||
|
||
content = 'This is a test'
|
||
moderated = True
|
||
conditionsStr = '#foo or #bar'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['or', ['#foo'], ['#bar']])
|
||
assert str(tagsInConditions) == str(['#foo', '#bar'])
|
||
hashtags = ['#foo']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#carrot', '#stick']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
content = 'This is a test'
|
||
url = 'https://testsite.com/something'
|
||
moderated = True
|
||
conditionsStr = '#foo and from "testsite.com"'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['and', ['#foo'], ['from', ['"testsite.com"']]])
|
||
assert str(tagsInConditions) == str(['#foo'])
|
||
hashtags = ['#foo']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content,
|
||
'othersite.net')
|
||
|
||
content = 'This is a test'
|
||
moderated = True
|
||
conditionsStr = 'contains "is a" and #foo or #bar'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == \
|
||
str(['and', ['contains', ['"is a"']],
|
||
['or', ['#foo'], ['#bar']]])
|
||
assert str(tagsInConditions) == str(['#foo', '#bar'])
|
||
hashtags = ['#foo']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#carrot', '#stick']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
moderated = False
|
||
conditionsStr = 'not moderated and #foo or #bar'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == \
|
||
str(['not', ['and', ['moderated'], ['or', ['#foo'], ['#bar']]]])
|
||
assert str(tagsInConditions) == str(['#foo', '#bar'])
|
||
hashtags = ['#foo']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#carrot', '#stick']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
moderated = True
|
||
conditionsStr = 'moderated and #foo or #bar'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == \
|
||
str(['and', ['moderated'], ['or', ['#foo'], ['#bar']]])
|
||
assert str(tagsInConditions) == str(['#foo', '#bar'])
|
||
hashtags = ['#foo']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#carrot', '#stick']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
conditionsStr = 'x'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert tree is None
|
||
assert tagsInConditions == []
|
||
hashtags = ['#foo']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
conditionsStr = '#x'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['#x'])
|
||
assert str(tagsInConditions) == str(['#x'])
|
||
hashtags = ['#x']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#y', '#z']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
conditionsStr = 'not #b'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['not', ['#b']])
|
||
assert str(tagsInConditions) == str(['#b'])
|
||
hashtags = ['#y', '#z']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#a', '#b', '#c']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
conditionsStr = '#foo or #bar and #a'
|
||
tagsInConditions = []
|
||
tree = hashtagRuleTree(operators, conditionsStr,
|
||
tagsInConditions, moderated)
|
||
assert str(tree) == str(['and', ['or', ['#foo'], ['#bar']], ['#a']])
|
||
assert str(tagsInConditions) == str(['#foo', '#bar', '#a'])
|
||
hashtags = ['#foo', '#bar', '#a']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#bar', '#a']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#foo', '#a']
|
||
assert hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
hashtags = ['#x', '#a']
|
||
assert not hashtagRuleResolve(tree, hashtags, moderated, content, url)
|
||
|
||
|
||
def _testGetNewswireTags():
|
||
print('testGetNewswireTags')
|
||
rssDescription = '<img src="https://somesite/someimage.jpg" ' + \
|
||
'class="misc-stuff" alt="#ExcitingHashtag" ' + \
|
||
'srcset="https://somesite/someimage.jpg" ' + \
|
||
'sizes="(max-width: 864px) 100vw, 864px" />' + \
|
||
'Compelling description with #ExcitingHashtag, which is ' + \
|
||
'being posted in #BoringForum'
|
||
tags = getNewswireTags(rssDescription, 10)
|
||
assert len(tags) == 2
|
||
assert '#BoringForum' in tags
|
||
assert '#ExcitingHashtag' in tags
|
||
|
||
|
||
def _testFirstParagraphFromString():
|
||
print('testFirstParagraphFromString')
|
||
testStr = \
|
||
'<p><a href="https://somesite.com/somepath">This is a test</a></p>' + \
|
||
'<p>This is another paragraph</p>'
|
||
resultStr = firstParagraphFromString(testStr)
|
||
if resultStr != 'This is a test':
|
||
print(resultStr)
|
||
assert resultStr == 'This is a test'
|
||
|
||
testStr = 'Testing without html'
|
||
resultStr = firstParagraphFromString(testStr)
|
||
assert resultStr == testStr
|
||
|
||
|
||
def _testParseFeedDate():
|
||
print('testParseFeedDate')
|
||
|
||
pubDate = "2020-12-14T00:08:06+00:00"
|
||
publishedDate = parseFeedDate(pubDate)
|
||
assert publishedDate == "2020-12-14 00:08:06+00:00"
|
||
|
||
pubDate = "Tue, 08 Dec 2020 06:24:38 -0600"
|
||
publishedDate = parseFeedDate(pubDate)
|
||
assert publishedDate == "2020-12-08 12:24:38+00:00"
|
||
|
||
pubDate = "2020-08-27T16:12:34+00:00"
|
||
publishedDate = parseFeedDate(pubDate)
|
||
assert publishedDate == "2020-08-27 16:12:34+00:00"
|
||
|
||
pubDate = "Sun, 22 Nov 2020 19:51:33 +0100"
|
||
publishedDate = parseFeedDate(pubDate)
|
||
assert publishedDate == "2020-11-22 18:51:33+00:00"
|
||
|
||
|
||
def _testValidNickname():
|
||
print('testValidNickname')
|
||
domain = 'somedomain.net'
|
||
|
||
nickname = 'myvalidnick'
|
||
assert validNickname(domain, nickname)
|
||
|
||
nickname = 'my.invalid.nick'
|
||
assert not validNickname(domain, nickname)
|
||
|
||
nickname = 'myinvalidnick?'
|
||
assert not validNickname(domain, nickname)
|
||
|
||
nickname = 'my invalid nick?'
|
||
assert not validNickname(domain, nickname)
|
||
|
||
|
||
def _testGuessHashtagCategory() -> None:
|
||
print('testGuessHashtagCategory')
|
||
hashtagCategories = {
|
||
"foo": ["swan", "goose"],
|
||
"bar": ["cats", "mouse"]
|
||
}
|
||
guess = guessHashtagCategory("unspecifiedgoose", hashtagCategories)
|
||
assert guess == "foo"
|
||
|
||
guess = guessHashtagCategory("mastocats", hashtagCategories)
|
||
assert guess == "bar"
|
||
|
||
|
||
def _testGetMentionedPeople() -> None:
|
||
print('testGetMentionedPeople')
|
||
baseDir = os.getcwd()
|
||
|
||
content = "@dragon@cave.site @bat@cave.site This is a test."
|
||
actors = getMentionedPeople(baseDir, 'https',
|
||
content,
|
||
'mydomain', False)
|
||
assert actors
|
||
assert len(actors) == 2
|
||
assert actors[0] == "https://cave.site/users/dragon"
|
||
assert actors[1] == "https://cave.site/users/bat"
|
||
|
||
|
||
def _testReplyToPublicPost() -> None:
|
||
baseDir = os.getcwd()
|
||
systemLanguage = 'en'
|
||
nickname = 'test7492362'
|
||
domain = 'other.site'
|
||
port = 443
|
||
httpPrefix = 'https'
|
||
postId = httpPrefix + '://rat.site/users/ninjarodent/statuses/63746173435'
|
||
content = "@ninjarodent@rat.site This is a test."
|
||
followersOnly = False
|
||
saveToFile = False
|
||
clientToServer = False
|
||
commentsEnabled = True
|
||
attachImageFilename = None
|
||
mediaType = None
|
||
imageDescription = 'Some description'
|
||
city = 'London, England'
|
||
testInReplyToAtomUri = None
|
||
testSubject = None
|
||
testSchedulePost = False
|
||
testEventDate = None
|
||
testEventTime = None
|
||
testLocation = None
|
||
testIsArticle = False
|
||
reply = \
|
||
createPublicPost(baseDir, nickname, domain, port, httpPrefix,
|
||
content, followersOnly, saveToFile,
|
||
clientToServer, commentsEnabled,
|
||
attachImageFilename, mediaType,
|
||
imageDescription, city, postId,
|
||
testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
# print(str(reply))
|
||
assert reply['object']['content'] == \
|
||
'<p><span class=\"h-card\">' + \
|
||
'<a href=\"https://rat.site/@ninjarodent\" ' + \
|
||
'class=\"u-url mention\">@<span>ninjarodent</span>' + \
|
||
'</a></span> This is a test.</p>'
|
||
reply['object']['contentMap'][systemLanguage] = reply['object']['content']
|
||
assert reply['object']['tag'][0]['type'] == 'Mention'
|
||
assert reply['object']['tag'][0]['name'] == '@ninjarodent@rat.site'
|
||
assert reply['object']['tag'][0]['href'] == \
|
||
'https://rat.site/users/ninjarodent'
|
||
assert len(reply['object']['to']) == 1
|
||
assert reply['object']['to'][0].endswith('#Public')
|
||
assert len(reply['object']['cc']) >= 1
|
||
assert reply['object']['cc'][0].endswith(nickname + '/followers')
|
||
assert len(reply['object']['tag']) == 1
|
||
assert len(reply['object']['cc']) == 2
|
||
assert reply['object']['cc'][1] == \
|
||
httpPrefix + '://rat.site/users/ninjarodent'
|
||
|
||
|
||
def _getFunctionCallArgs(name: str, lines: [], startLineCtr: int) -> []:
|
||
"""Returns the arguments of a function call given lines
|
||
of source code and a starting line number
|
||
"""
|
||
argsStr = lines[startLineCtr].split(name + '(')[1]
|
||
if ')' in argsStr:
|
||
argsStr = argsStr.split(')')[0].replace(' ', '').split(',')
|
||
return argsStr
|
||
for lineCtr in range(startLineCtr + 1, len(lines)):
|
||
if ')' not in lines[lineCtr]:
|
||
argsStr += lines[lineCtr]
|
||
continue
|
||
else:
|
||
argsStr += lines[lineCtr].split(')')[0]
|
||
break
|
||
return argsStr.replace('\n', '').replace(' ', '').split(',')
|
||
|
||
|
||
def getFunctionCalls(name: str, lines: [], startLineCtr: int,
|
||
functionProperties: {}) -> []:
|
||
"""Returns the functions called by the given one,
|
||
Starting with the given source code at the given line
|
||
"""
|
||
callsFunctions = []
|
||
functionContentStr = ''
|
||
for lineCtr in range(startLineCtr + 1, len(lines)):
|
||
lineStr = lines[lineCtr].strip()
|
||
if lineStr.startswith('def '):
|
||
break
|
||
if lineStr.startswith('class '):
|
||
break
|
||
functionContentStr += lines[lineCtr]
|
||
for funcName, properties in functionProperties.items():
|
||
if funcName + '(' in functionContentStr:
|
||
callsFunctions.append(funcName)
|
||
return callsFunctions
|
||
|
||
|
||
def _functionArgsMatch(callArgs: [], funcArgs: []):
|
||
"""Do the function artuments match the function call arguments
|
||
"""
|
||
if len(callArgs) == len(funcArgs):
|
||
return True
|
||
|
||
# count non-optional arguments
|
||
callArgsCtr = 0
|
||
for a in callArgs:
|
||
if a == 'self':
|
||
continue
|
||
if '=' not in a or a.startswith("'"):
|
||
callArgsCtr += 1
|
||
|
||
funcArgsCtr = 0
|
||
for a in funcArgs:
|
||
if a == 'self':
|
||
continue
|
||
if '=' not in a or a.startswith("'"):
|
||
funcArgsCtr += 1
|
||
|
||
return callArgsCtr >= funcArgsCtr
|
||
|
||
|
||
def _moduleInGroups(modName: str, includeGroups: [], modGroups: {}) -> bool:
|
||
"""Is the given module within the included groups list?
|
||
"""
|
||
for groupName in includeGroups:
|
||
if modName in modGroups[groupName]:
|
||
return True
|
||
return False
|
||
|
||
|
||
def _diagramGroups(includeGroups: [],
|
||
excludeExtraModules: [],
|
||
modules: {}, modGroups: {},
|
||
maxModuleCalls: int) -> None:
|
||
"""Draws a dot diagram containing only the given module groups
|
||
"""
|
||
callGraphStr = 'digraph EpicyonGroups {\n\n'
|
||
callGraphStr += ' graph [fontsize=10 fontname="Verdana" compound=true];\n'
|
||
callGraphStr += ' node [fontsize=10 fontname="Verdana"];\n\n'
|
||
excludeModulesFromDiagram = [
|
||
'setup', 'tests', '__init__', 'pyjsonld'
|
||
]
|
||
excludeModulesFromDiagram += excludeExtraModules
|
||
# colors of modules nodes
|
||
for modName, modProperties in modules.items():
|
||
if modName in excludeModulesFromDiagram:
|
||
continue
|
||
if not _moduleInGroups(modName, includeGroups, modGroups):
|
||
continue
|
||
if not modProperties.get('calls'):
|
||
callGraphStr += ' "' + modName + \
|
||
'" [fillcolor=yellow style=filled];\n'
|
||
continue
|
||
if len(modProperties['calls']) <= int(maxModuleCalls / 8):
|
||
callGraphStr += ' "' + modName + \
|
||
'" [fillcolor=green style=filled];\n'
|
||
elif len(modProperties['calls']) < int(maxModuleCalls / 4):
|
||
callGraphStr += ' "' + modName + \
|
||
'" [fillcolor=orange style=filled];\n'
|
||
else:
|
||
callGraphStr += ' "' + modName + \
|
||
'" [fillcolor=red style=filled];\n'
|
||
callGraphStr += '\n'
|
||
# connections between modules
|
||
for modName, modProperties in modules.items():
|
||
if modName in excludeModulesFromDiagram:
|
||
continue
|
||
if not _moduleInGroups(modName, includeGroups, modGroups):
|
||
continue
|
||
if not modProperties.get('calls'):
|
||
continue
|
||
for modCall in modProperties['calls']:
|
||
if modCall in excludeModulesFromDiagram:
|
||
continue
|
||
if not _moduleInGroups(modCall, includeGroups, modGroups):
|
||
continue
|
||
callGraphStr += ' "' + modName + '" -> "' + modCall + '";\n'
|
||
# module groups/clusters
|
||
clusterCtr = 1
|
||
for groupName, groupModules in modGroups.items():
|
||
if groupName not in includeGroups:
|
||
continue
|
||
callGraphStr += '\n'
|
||
callGraphStr += \
|
||
' subgraph cluster_' + str(clusterCtr) + ' {\n'
|
||
callGraphStr += ' node [style=filled];\n'
|
||
for modName in groupModules:
|
||
if modName not in excludeModulesFromDiagram:
|
||
callGraphStr += ' ' + modName + ';\n'
|
||
callGraphStr += ' label = "' + groupName + '";\n'
|
||
callGraphStr += ' color = blue;\n'
|
||
callGraphStr += ' }\n'
|
||
clusterCtr += 1
|
||
callGraphStr += '\n}\n'
|
||
filename = 'epicyon_groups'
|
||
for groupName in includeGroups:
|
||
filename += '_' + groupName.replace(' ', '-')
|
||
filename += '.dot'
|
||
with open(filename, 'w+') as fp:
|
||
fp.write(callGraphStr)
|
||
print('Graph saved to ' + filename)
|
||
print('Plot using: ' +
|
||
'sfdp -x -Goverlap=false -Goverlap_scaling=2 ' +
|
||
'-Gsep=+100 -Tx11 epicyon_modules.dot')
|
||
|
||
|
||
def _testFunctions():
|
||
print('testFunctions')
|
||
function = {}
|
||
functionProperties = {}
|
||
modules = {}
|
||
modGroups = {}
|
||
methodLOC = []
|
||
|
||
for subdir, dirs, files in os.walk('.'):
|
||
for sourceFile in files:
|
||
if not sourceFile.endswith('.py'):
|
||
continue
|
||
if sourceFile.startswith('.#'):
|
||
continue
|
||
modName = sourceFile.replace('.py', '')
|
||
modules[modName] = {
|
||
'functions': []
|
||
}
|
||
sourceStr = ''
|
||
with open(sourceFile, 'r') as f:
|
||
sourceStr = f.read()
|
||
modules[modName]['source'] = sourceStr
|
||
with open(sourceFile, 'r') as f:
|
||
lines = f.readlines()
|
||
modules[modName]['lines'] = lines
|
||
lineCount = 0
|
||
prevLine = 'start'
|
||
methodName = ''
|
||
for line in lines:
|
||
if '__module_group__' in line:
|
||
if '=' in line:
|
||
groupName = line.split('=')[1].strip()
|
||
groupName = groupName.replace('"', '')
|
||
groupName = groupName.replace("'", '')
|
||
modules[modName]['group'] = groupName
|
||
if not modGroups.get(groupName):
|
||
modGroups[groupName] = [modName]
|
||
else:
|
||
if modName not in modGroups[groupName]:
|
||
modGroups[groupName].append(modName)
|
||
if not line.strip().startswith('def '):
|
||
if lineCount > 0:
|
||
lineCount += 1
|
||
# add LOC count for this function
|
||
if len(prevLine.strip()) == 0 and \
|
||
len(line.strip()) == 0 and \
|
||
lineCount > 2:
|
||
lineCount -= 2
|
||
if lineCount > 80:
|
||
locStr = str(lineCount) + ';' + methodName
|
||
if lineCount < 1000:
|
||
locStr = '0' + locStr
|
||
if lineCount < 100:
|
||
locStr = '0' + locStr
|
||
if lineCount < 10:
|
||
locStr = '0' + locStr
|
||
if locStr not in methodLOC:
|
||
methodLOC.append(locStr)
|
||
lineCount = 0
|
||
prevLine = line
|
||
continue
|
||
prevLine = line
|
||
lineCount = 1
|
||
methodName = line.split('def ', 1)[1].split('(')[0]
|
||
methodArgs = \
|
||
sourceStr.split('def ' + methodName + '(')[1]
|
||
methodArgs = methodArgs.split(')')[0]
|
||
methodArgs = methodArgs.replace(' ', '').split(',')
|
||
if function.get(modName):
|
||
function[modName].append(methodName)
|
||
else:
|
||
function[modName] = [methodName]
|
||
if methodName not in modules[modName]['functions']:
|
||
modules[modName]['functions'].append(methodName)
|
||
functionProperties[methodName] = {
|
||
"args": methodArgs,
|
||
"module": modName,
|
||
"calledInModule": []
|
||
}
|
||
# LOC count for the last function
|
||
if lineCount > 2:
|
||
lineCount -= 2
|
||
if lineCount > 80:
|
||
locStr = str(lineCount) + ';' + methodName
|
||
if lineCount < 1000:
|
||
locStr = '0' + locStr
|
||
if lineCount < 100:
|
||
locStr = '0' + locStr
|
||
if lineCount < 10:
|
||
locStr = '0' + locStr
|
||
if locStr not in methodLOC:
|
||
methodLOC.append(locStr)
|
||
break
|
||
|
||
print('LOC counts:')
|
||
methodLOC.sort()
|
||
for locStr in methodLOC:
|
||
print(locStr.split(';')[0] + ' ' + locStr.split(';')[1])
|
||
|
||
excludeFuncArgs = [
|
||
'pyjsonld'
|
||
]
|
||
excludeFuncs = [
|
||
'link',
|
||
'set',
|
||
'get'
|
||
]
|
||
# which modules is each function used within?
|
||
for modName, modProperties in modules.items():
|
||
print('Module: ' + modName + ' ✓')
|
||
for name, properties in functionProperties.items():
|
||
lineCtr = 0
|
||
for line in modules[modName]['lines']:
|
||
lineStr = line.strip()
|
||
if lineStr.startswith('def '):
|
||
lineCtr += 1
|
||
continue
|
||
if lineStr.startswith('class '):
|
||
lineCtr += 1
|
||
continue
|
||
if name + '(' in line:
|
||
modList = \
|
||
functionProperties[name]['calledInModule']
|
||
if modName not in modList:
|
||
modList.append(modName)
|
||
if modName in excludeFuncArgs:
|
||
lineCtr += 1
|
||
continue
|
||
if name in excludeFuncs:
|
||
lineCtr += 1
|
||
continue
|
||
callArgs = \
|
||
_getFunctionCallArgs(name,
|
||
modules[modName]['lines'],
|
||
lineCtr)
|
||
funcArgs = functionProperties[name]['args']
|
||
if not _functionArgsMatch(callArgs, funcArgs):
|
||
print('Call to function ' + name +
|
||
' does not match its arguments')
|
||
print('def args: ' +
|
||
str(len(functionProperties[name]['args'])) +
|
||
'\n' + str(functionProperties[name]['args']))
|
||
print('Call args: ' + str(len(callArgs)) + '\n' +
|
||
str(callArgs))
|
||
print('module ' + modName + ' line ' + str(lineCtr))
|
||
assert False
|
||
lineCtr += 1
|
||
|
||
# don't check these functions, because they are procedurally called
|
||
exclusions = [
|
||
'do_GET',
|
||
'do_POST',
|
||
'do_HEAD',
|
||
'__run',
|
||
'globaltrace',
|
||
'localtrace',
|
||
'kill',
|
||
'clone',
|
||
'unregister_rdf_parser',
|
||
'set_document_loader',
|
||
'has_property',
|
||
'has_value',
|
||
'add_value',
|
||
'get_values',
|
||
'remove_property',
|
||
'remove_value',
|
||
'normalize',
|
||
'get_document_loader',
|
||
'runInboxQueueWatchdog',
|
||
'runInboxQueue',
|
||
'runPostSchedule',
|
||
'runPostScheduleWatchdog',
|
||
'str2bool',
|
||
'runNewswireDaemon',
|
||
'runNewswireWatchdog',
|
||
'runFederatedSharesWatchdog',
|
||
'runFederatedSharesDaemon',
|
||
'threadSendPost',
|
||
'sendToFollowers',
|
||
'expireCache',
|
||
'getMutualsOfPerson',
|
||
'runPostsQueue',
|
||
'runSharesExpire',
|
||
'runPostsWatchdog',
|
||
'runSharesExpireWatchdog',
|
||
'getThisWeeksEvents',
|
||
'getAvailability',
|
||
'_testThreadsFunction',
|
||
'createServerAlice',
|
||
'createServerBob',
|
||
'createServerEve',
|
||
'E2EEremoveDevice',
|
||
'setOrganizationScheme',
|
||
'fill_headers',
|
||
'_nothing',
|
||
'sharesCatalogAccountEndpoint'
|
||
]
|
||
excludeImports = [
|
||
'link',
|
||
'start'
|
||
]
|
||
excludeLocal = [
|
||
'pyjsonld',
|
||
'daemon',
|
||
'tests'
|
||
]
|
||
excludeMods = [
|
||
'pyjsonld'
|
||
]
|
||
# check that functions are called somewhere
|
||
for name, properties in functionProperties.items():
|
||
if name.startswith('__'):
|
||
if name.endswith('__'):
|
||
continue
|
||
if name in exclusions:
|
||
continue
|
||
if properties['module'] in excludeMods:
|
||
continue
|
||
isLocalFunction = False
|
||
if not properties['calledInModule']:
|
||
print('function ' + name +
|
||
' in module ' + properties['module'] +
|
||
' is not called anywhere')
|
||
assert properties['calledInModule']
|
||
|
||
if len(properties['calledInModule']) == 1:
|
||
modName = properties['calledInModule'][0]
|
||
if modName not in excludeLocal and \
|
||
modName == properties['module']:
|
||
isLocalFunction = True
|
||
if not name.startswith('_'):
|
||
print('Local function ' + name +
|
||
' in ' + modName + '.py does not begin with _')
|
||
assert False
|
||
|
||
if name not in excludeImports:
|
||
for modName in properties['calledInModule']:
|
||
if modName == properties['module']:
|
||
continue
|
||
importStr = 'from ' + properties['module'] + ' import ' + name
|
||
if importStr not in modules[modName]['source']:
|
||
print(importStr + ' not found in ' + modName + '.py')
|
||
assert False
|
||
|
||
if not isLocalFunction:
|
||
if name.startswith('_'):
|
||
excludePublic = [
|
||
'pyjsonld',
|
||
'daemon',
|
||
'tests'
|
||
]
|
||
modName = properties['module']
|
||
if modName not in excludePublic:
|
||
print('Public function ' + name + ' in ' +
|
||
modName + '.py begins with _')
|
||
assert False
|
||
print('Function: ' + name + ' ✓')
|
||
|
||
print('Constructing function call graph')
|
||
moduleColors = ('red', 'green', 'yellow', 'orange', 'purple', 'cyan',
|
||
'darkgoldenrod3', 'darkolivegreen1', 'darkorange1',
|
||
'darkorchid1', 'darkseagreen', 'darkslategray4',
|
||
'deeppink1', 'deepskyblue1', 'dimgrey', 'gold1',
|
||
'goldenrod', 'burlywood2', 'bisque1', 'brown1',
|
||
'chartreuse2', 'cornsilk', 'darksalmon')
|
||
maxModuleCalls = 1
|
||
maxFunctionCalls = 1
|
||
colorCtr = 0
|
||
for modName, modProperties in modules.items():
|
||
lineCtr = 0
|
||
modules[modName]['color'] = moduleColors[colorCtr]
|
||
colorCtr += 1
|
||
if colorCtr >= len(moduleColors):
|
||
colorCtr = 0
|
||
for line in modules[modName]['lines']:
|
||
if line.strip().startswith('def '):
|
||
name = line.split('def ')[1].split('(')[0]
|
||
callsList = \
|
||
getFunctionCalls(name, modules[modName]['lines'],
|
||
lineCtr, functionProperties)
|
||
functionProperties[name]['calls'] = callsList.copy()
|
||
if len(callsList) > maxFunctionCalls:
|
||
maxFunctionCalls = len(callsList)
|
||
# keep track of which module calls which other module
|
||
for fn in callsList:
|
||
modCall = functionProperties[fn]['module']
|
||
if modCall != modName:
|
||
if modules[modName].get('calls'):
|
||
if modCall not in modules[modName]['calls']:
|
||
modules[modName]['calls'].append(modCall)
|
||
if len(modules[modName]['calls']) > \
|
||
maxModuleCalls:
|
||
maxModuleCalls = \
|
||
len(modules[modName]['calls'])
|
||
else:
|
||
modules[modName]['calls'] = [modCall]
|
||
lineCtr += 1
|
||
|
||
_diagramGroups(['Commandline Interface', 'ActivityPub'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Commandline Interface', 'Core'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Timeline', 'Core'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Web Interface', 'Core'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Web Interface Columns', 'Core'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Core'], [],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['ActivityPub'], [],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['ActivityPub', 'Core'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['ActivityPub', 'Security'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Core', 'Security'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Timeline', 'Security'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Web Interface', 'Accessibility'],
|
||
['utils', 'webapp_utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
_diagramGroups(['Core', 'Accessibility'], ['utils'],
|
||
modules, modGroups, maxModuleCalls)
|
||
|
||
|
||
def _testLinksWithinPost() -> None:
|
||
baseDir = os.getcwd()
|
||
systemLanguage = 'en'
|
||
nickname = 'test27636'
|
||
domain = 'rando.site'
|
||
port = 443
|
||
httpPrefix = 'https'
|
||
content = 'This is a test post with links.\n\n' + \
|
||
'ftp://ftp.ncdc.noaa.gov/pub/data/ghcn/v4/\n\nhttps://freedombone.net'
|
||
followersOnly = False
|
||
saveToFile = False
|
||
clientToServer = False
|
||
commentsEnabled = True
|
||
attachImageFilename = None
|
||
mediaType = None
|
||
imageDescription = None
|
||
city = 'London, England'
|
||
testInReplyTo = None
|
||
testInReplyToAtomUri = None
|
||
testSubject = None
|
||
testSchedulePost = False
|
||
testEventDate = None
|
||
testEventTime = None
|
||
testLocation = None
|
||
testIsArticle = False
|
||
|
||
postJsonObject = \
|
||
createPublicPost(baseDir, nickname, domain, port, httpPrefix,
|
||
content, followersOnly, saveToFile,
|
||
clientToServer, commentsEnabled,
|
||
attachImageFilename, mediaType,
|
||
imageDescription, city,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
|
||
assert postJsonObject['object']['content'] == \
|
||
'<p>This is a test post with links.<br><br>' + \
|
||
'<a href="ftp://ftp.ncdc.noaa.gov/pub/data/ghcn/v4/" ' + \
|
||
'rel="nofollow noopener noreferrer" target="_blank">' + \
|
||
'<span class="invisible">ftp://</span>' + \
|
||
'<span class="ellipsis">' + \
|
||
'ftp.ncdc.noaa.gov/pub/data/ghcn/v4/</span>' + \
|
||
'</a><br><br><a href="https://freedombone.net" ' + \
|
||
'rel="nofollow noopener noreferrer" target="_blank">' + \
|
||
'<span class="invisible">https://</span>' + \
|
||
'<span class="ellipsis">freedombone.net</span></a></p>'
|
||
assert postJsonObject['object']['content'] == \
|
||
postJsonObject['object']['contentMap'][systemLanguage]
|
||
|
||
content = "<p>Some text</p><p>Other text</p><p>More text</p>" + \
|
||
"<pre><code>Errno::EOHNOES (No such file or rodent @ " + \
|
||
"ik_right - /tmp/blah.png)<br></code></pre><p>" + \
|
||
"(<a href=\"https://welllookeyhere.maam/error.txt\" " + \
|
||
"rel=\"nofollow noopener noreferrer\" target=\"_blank\">" + \
|
||
"wuh</a>)</p><p>Oh yeah like for sure</p>" + \
|
||
"<p>Ground sloth tin opener</p>" + \
|
||
"<p><a href=\"https://whocodedthis.huh/tags/" + \
|
||
"taggedthing\" class=\"mention hashtag\" rel=\"tag\" " + \
|
||
"target=\"_blank\">#<span>taggedthing</span></a></p>"
|
||
postJsonObject = \
|
||
createPublicPost(baseDir, nickname, domain, port, httpPrefix,
|
||
content,
|
||
False, False,
|
||
False, True,
|
||
None, None,
|
||
False, None,
|
||
testInReplyTo, testInReplyToAtomUri,
|
||
testSubject, testSchedulePost,
|
||
testEventDate, testEventTime, testLocation,
|
||
testIsArticle, systemLanguage)
|
||
assert postJsonObject['object']['content'] == content
|
||
assert postJsonObject['object']['contentMap'][systemLanguage] == content
|
||
|
||
|
||
def _testMastoApi():
|
||
print('testMastoApi')
|
||
nickname = 'ThisIsATestNickname'
|
||
mastoId = getMastoApiV1IdFromNickname(nickname)
|
||
assert(mastoId)
|
||
nickname2 = getNicknameFromMastoApiV1Id(mastoId)
|
||
if nickname2 != nickname:
|
||
print(nickname + ' != ' + nickname2)
|
||
assert nickname2 == nickname
|
||
|
||
|
||
def _testDomainHandling():
|
||
print('testDomainHandling')
|
||
testDomain = 'localhost'
|
||
assert decodedHost(testDomain) == testDomain
|
||
testDomain = '127.0.0.1:60'
|
||
assert decodedHost(testDomain) == testDomain
|
||
testDomain = '192.168.5.153'
|
||
assert decodedHost(testDomain) == testDomain
|
||
testDomain = 'xn--espaa-rta.icom.museum'
|
||
assert decodedHost(testDomain) == "españa.icom.museum"
|
||
|
||
|
||
def _testPrepareHtmlPostNickname():
|
||
print('testPrepareHtmlPostNickname')
|
||
postHtml = '<a class="imageAnchor" href="/users/bob?replyfollowers='
|
||
postHtml += '<a class="imageAnchor" href="/users/bob?repeatprivate='
|
||
result = prepareHtmlPostNickname('alice', postHtml)
|
||
assert result == postHtml.replace('/bob?', '/alice?')
|
||
|
||
postHtml = '<a class="imageAnchor" href="/users/bob?replyfollowers='
|
||
postHtml += '<a class="imageAnchor" href="/users/bob;repeatprivate='
|
||
expectedHtml = '<a class="imageAnchor" href="/users/alice?replyfollowers='
|
||
expectedHtml += '<a class="imageAnchor" href="/users/bob;repeatprivate='
|
||
result = prepareHtmlPostNickname('alice', postHtml)
|
||
assert result == expectedHtml
|
||
|
||
|
||
def _testValidHashTag():
|
||
print('testValidHashTag')
|
||
assert validHashTag('ThisIsValid')
|
||
assert validHashTag('ThisIsValid12345')
|
||
assert validHashTag('ThisIsVälid')
|
||
assert validHashTag('यहमान्यहै')
|
||
assert not validHashTag('ThisIsNotValid!')
|
||
assert not validHashTag('#ThisIsAlsoNotValid')
|
||
assert not validHashTag('#यहमान्यहै')
|
||
assert not validHashTag('ThisIsAlso&NotValid')
|
||
assert not validHashTag('ThisIsAlsoNotValid"')
|
||
assert not validHashTag('This Is Also Not Valid"')
|
||
assert not validHashTag('This=IsAlsoNotValid"')
|
||
|
||
|
||
def _testMarkdownToHtml():
|
||
print('testMarkdownToHtml')
|
||
markdown = 'This is just plain text'
|
||
assert markdownToHtml(markdown) == markdown
|
||
|
||
markdown = 'This is a quotation:\n' + \
|
||
'> Some quote or other'
|
||
assert markdownToHtml(markdown) == 'This is a quotation:<br>' + \
|
||
'<blockquote><i>Some quote or other</i></blockquote>'
|
||
|
||
markdown = 'This is a multi-line quotation:\n' + \
|
||
'> The first line\n' + \
|
||
'> The second line'
|
||
assert markdownToHtml(markdown) == \
|
||
'This is a multi-line quotation:<br>' + \
|
||
'<blockquote><i>The first line The second line</i></blockquote>'
|
||
|
||
markdown = 'This is **bold**'
|
||
assert markdownToHtml(markdown) == 'This is <b>bold</b>'
|
||
|
||
markdown = 'This is *italic*'
|
||
assert markdownToHtml(markdown) == 'This is <i>italic</i>'
|
||
|
||
markdown = 'This is _underlined_'
|
||
assert markdownToHtml(markdown) == 'This is <ul>underlined</ul>'
|
||
|
||
markdown = 'This is **just** plain text'
|
||
assert markdownToHtml(markdown) == 'This is <b>just</b> plain text'
|
||
|
||
markdown = '# Title1\n### Title3\n## Title2\n'
|
||
assert markdownToHtml(markdown) == \
|
||
'<h1>Title1</h1><h3>Title3</h3><h2>Title2</h2>'
|
||
|
||
markdown = \
|
||
'This is [a link](https://something.somewhere) to something.\n' + \
|
||
'And [something else](https://cat.pic).\n' + \
|
||
'Or .'
|
||
assert markdownToHtml(markdown) == \
|
||
'This is <a href="https://something.somewhere" ' + \
|
||
'target="_blank" rel="nofollow noopener noreferrer">' + \
|
||
'a link</a> to something.<br>' + \
|
||
'And <a href="https://cat.pic" ' + \
|
||
'target="_blank" rel="nofollow noopener noreferrer">' + \
|
||
'something else</a>.<br>' + \
|
||
'Or <img class="markdownImage" src="/cat.jpg" alt="pounce" />.'
|
||
|
||
|
||
def _testExtractTextFieldsInPOST():
|
||
print('testExtractTextFieldsInPOST')
|
||
boundary = '-----------------------------116202748023898664511855843036'
|
||
formData = '-----------------------------116202748023898664511855' + \
|
||
'843036\r\nContent-Disposition: form-data; name="submitPost"' + \
|
||
'\r\n\r\nSubmit\r\n-----------------------------116202748023' + \
|
||
'898664511855843036\r\nContent-Disposition: form-data; name=' + \
|
||
'"subject"\r\n\r\n\r\n-----------------------------116202748' + \
|
||
'023898664511855843036\r\nContent-Disposition: form-data; na' + \
|
||
'me="message"\r\n\r\nThis is a ; test\r\n-------------------' + \
|
||
'----------116202748023898664511855843036\r\nContent-Disposi' + \
|
||
'tion: form-data; name="commentsEnabled"\r\n\r\non\r\n------' + \
|
||
'-----------------------116202748023898664511855843036\r\nCo' + \
|
||
'ntent-Disposition: form-data; name="eventDate"\r\n\r\n\r\n' + \
|
||
'-----------------------------116202748023898664511855843036' + \
|
||
'\r\nContent-Disposition: form-data; name="eventTime"\r\n\r' + \
|
||
'\n\r\n-----------------------------116202748023898664511855' + \
|
||
'843036\r\nContent-Disposition: form-data; name="location"' + \
|
||
'\r\n\r\n\r\n-----------------------------116202748023898664' + \
|
||
'511855843036\r\nContent-Disposition: form-data; name=' + \
|
||
'"imageDescription"\r\n\r\n\r\n-----------------------------' + \
|
||
'116202748023898664511855843036\r\nContent-Disposition: ' + \
|
||
'form-data; name="attachpic"; filename=""\r\nContent-Type: ' + \
|
||
'application/octet-stream\r\n\r\n\r\n----------------------' + \
|
||
'-------116202748023898664511855843036--\r\n'
|
||
debug = False
|
||
fields = extractTextFieldsInPOST(None, boundary, debug, formData)
|
||
assert fields['submitPost'] == 'Submit'
|
||
assert fields['subject'] == ''
|
||
assert fields['commentsEnabled'] == 'on'
|
||
assert fields['eventDate'] == ''
|
||
assert fields['eventTime'] == ''
|
||
assert fields['location'] == ''
|
||
assert fields['imageDescription'] == ''
|
||
assert fields['message'] == 'This is a ; test'
|
||
|
||
|
||
def _testSpeakerReplaceLinks():
|
||
print('testSpeakerReplaceLinks')
|
||
text = 'The Tor Project: For Snowflake volunteers: If you use ' + \
|
||
'Firefox, Brave, or Chrome, our Snowflake extension turns ' + \
|
||
'your browser into a proxy that connects Tor users in ' + \
|
||
'censored regions to the Tor network. Note: you should ' + \
|
||
'not run more than one snowflake in the same ' + \
|
||
'network.https://support.torproject.org/censorship/' + \
|
||
'how-to-help-running-snowflake/'
|
||
detectedLinks = []
|
||
result = speakerReplaceLinks(text, {'Linked': 'Web link'}, detectedLinks)
|
||
assert len(detectedLinks) == 1
|
||
assert detectedLinks[0] == \
|
||
'https://support.torproject.org/censorship/' + \
|
||
'how-to-help-running-snowflake/'
|
||
assert 'Web link support.torproject.org' in result
|
||
|
||
|
||
def _testCamelCaseSplit():
|
||
print('testCamelCaseSplit')
|
||
testStr = 'ThisIsCamelCase'
|
||
assert camelCaseSplit(testStr) == 'This Is Camel Case'
|
||
|
||
testStr = 'Notcamelcase test'
|
||
assert camelCaseSplit(testStr) == 'Notcamelcase test'
|
||
|
||
|
||
def _testEmojiImages():
|
||
print('testEmojiImages')
|
||
emojiFilename = 'emoji/default_emoji.json'
|
||
assert os.path.isfile(emojiFilename)
|
||
emojiJson = loadJson(emojiFilename)
|
||
assert emojiJson
|
||
for emojiName, emojiImage in emojiJson.items():
|
||
emojiImageFilename = 'emoji/' + emojiImage + '.png'
|
||
if not os.path.isfile(emojiImageFilename):
|
||
print('Missing emoji image ' + emojiName + ' ' +
|
||
emojiImage + '.png')
|
||
assert os.path.isfile(emojiImageFilename)
|
||
|
||
|
||
def _testExtractPGPPublicKey():
|
||
print('testExtractPGPPublicKey')
|
||
pubKey = \
|
||
'-----BEGIN PGP PUBLIC KEY BLOCK-----\n\n' + \
|
||
'mDMEWZBueBYJKwYBBAHaRw8BAQdAKx1t6wL0RTuU6/' + \
|
||
'IBjngMbVJJ3Wg/3UW73/PV\n' + \
|
||
'I47xKTS0IUJvYiBNb3R0cmFtIDxib2JAZnJlZWRvb' + \
|
||
'WJvbmUubmV0PoiQBBMWCAA4\n' + \
|
||
'FiEEmruCwAq/OfgmgEh9zCU2GR+nwz8FAlmQbngCG' + \
|
||
'wMFCwkIBwMFFQoJCAsFFgID\n' + \
|
||
'AQACHgECF4AACgkQzCU2GR+nwz/9sAD/YgsHnVszH' + \
|
||
'Nz1zlVc5EgY1ByDupiJpHj0\n' + \
|
||
'XsLYk3AbNRgBALn45RqgD4eWHpmOriH09H5Rc5V9i' + \
|
||
'N4+OiGUn2AzJ6oHuDgEWZBu\n' + \
|
||
'eBIKKwYBBAGXVQEFAQEHQPRBG2ZQJce475S3e0Dxe' + \
|
||
'b0Fz5WdEu2q3GYLo4QG+4Ry\n' + \
|
||
'AwEIB4h4BBgWCAAgFiEEmruCwAq/OfgmgEh9zCU2G' + \
|
||
'R+nwz8FAlmQbngCGwwACgkQ\n' + \
|
||
'zCU2GR+nwz+OswD+JOoyBku9FzuWoVoOevU2HH+bP' + \
|
||
'OMDgY2OLnST9ZSyHkMBAMcK\n' + \
|
||
'fnaZ2Wi050483Sj2RmQRpb99Dod7rVZTDtCqXk0J\n' + \
|
||
'=gv5G\n' + \
|
||
'-----END PGP PUBLIC KEY BLOCK-----'
|
||
testStr = "Some introduction\n\n" + pubKey + "\n\nSome message."
|
||
assert containsPGPPublicKey(testStr)
|
||
assert not containsPGPPublicKey('String without a pgp key')
|
||
result = extractPGPPublicKey(testStr)
|
||
assert result
|
||
assert result == pubKey
|
||
|
||
|
||
def testUpdateActor():
|
||
print('Testing update of actor properties')
|
||
|
||
global testServerAliceRunning
|
||
testServerAliceRunning = False
|
||
|
||
httpPrefix = 'http'
|
||
proxyType = None
|
||
federationList = []
|
||
|
||
baseDir = os.getcwd()
|
||
if os.path.isdir(baseDir + '/.tests'):
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
os.mkdir(baseDir + '/.tests')
|
||
|
||
# create the server
|
||
aliceDir = baseDir + '/.tests/alice'
|
||
aliceDomain = '127.0.0.11'
|
||
alicePort = 61792
|
||
aliceSendThreads = []
|
||
bobAddress = '127.0.0.84:6384'
|
||
|
||
global thrAlice
|
||
if thrAlice:
|
||
while thrAlice.is_alive():
|
||
thrAlice.stop()
|
||
time.sleep(1)
|
||
thrAlice.kill()
|
||
|
||
thrAlice = \
|
||
threadWithTrace(target=createServerAlice,
|
||
args=(aliceDir, aliceDomain, alicePort, bobAddress,
|
||
federationList, False, False,
|
||
aliceSendThreads),
|
||
daemon=True)
|
||
|
||
thrAlice.start()
|
||
assert thrAlice.is_alive() is True
|
||
|
||
# wait for server to be running
|
||
ctr = 0
|
||
while not testServerAliceRunning:
|
||
time.sleep(1)
|
||
ctr += 1
|
||
if ctr > 60:
|
||
break
|
||
print('Alice online: ' + str(testServerAliceRunning))
|
||
|
||
print('\n\n*******************************************************')
|
||
print('Alice updates her PGP key')
|
||
|
||
sessionAlice = createSession(proxyType)
|
||
cachedWebfingers = {}
|
||
personCache = {}
|
||
password = 'alicepass'
|
||
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
|
||
actorFilename = aliceDir + '/accounts/' + 'alice@' + aliceDomain + '.json'
|
||
assert os.path.isfile(actorFilename)
|
||
assert len([name for name in os.listdir(outboxPath)
|
||
if os.path.isfile(os.path.join(outboxPath, name))]) == 0
|
||
pubKey = \
|
||
'-----BEGIN PGP PUBLIC KEY BLOCK-----\n\n' + \
|
||
'mDMEWZBueBYJKwYBBAHaRw8BAQdAKx1t6wL0RTuU6/' + \
|
||
'IBjngMbVJJ3Wg/3UW73/PV\n' + \
|
||
'I47xKTS0IUJvYiBNb3R0cmFtIDxib2JAZnJlZWRvb' + \
|
||
'WJvbmUubmV0PoiQBBMWCAA4\n' + \
|
||
'FiEEmruCwAq/OfgmgEh9zCU2GR+nwz8FAlmQbngCG' + \
|
||
'wMFCwkIBwMFFQoJCAsFFgID\n' + \
|
||
'AQACHgECF4AACgkQzCU2GR+nwz/9sAD/YgsHnVszH' + \
|
||
'Nz1zlVc5EgY1ByDupiJpHj0\n' + \
|
||
'XsLYk3AbNRgBALn45RqgD4eWHpmOriH09H5Rc5V9i' + \
|
||
'N4+OiGUn2AzJ6oHuDgEWZBu\n' + \
|
||
'eBIKKwYBBAGXVQEFAQEHQPRBG2ZQJce475S3e0Dxe' + \
|
||
'b0Fz5WdEu2q3GYLo4QG+4Ry\n' + \
|
||
'AwEIB4h4BBgWCAAgFiEEmruCwAq/OfgmgEh9zCU2G' + \
|
||
'R+nwz8FAlmQbngCGwwACgkQ\n' + \
|
||
'zCU2GR+nwz+OswD+JOoyBku9FzuWoVoOevU2HH+bP' + \
|
||
'OMDgY2OLnST9ZSyHkMBAMcK\n' + \
|
||
'fnaZ2Wi050483Sj2RmQRpb99Dod7rVZTDtCqXk0J\n' + \
|
||
'=gv5G\n' + \
|
||
'-----END PGP PUBLIC KEY BLOCK-----'
|
||
actorUpdate = \
|
||
pgpPublicKeyUpload(aliceDir, sessionAlice,
|
||
'alice', password,
|
||
aliceDomain, alicePort,
|
||
httpPrefix,
|
||
cachedWebfingers, personCache,
|
||
True, pubKey)
|
||
print('actor update result: ' + str(actorUpdate))
|
||
assert actorUpdate
|
||
|
||
# load alice actor
|
||
print('Loading actor: ' + actorFilename)
|
||
actorJson = loadJson(actorFilename)
|
||
assert actorJson
|
||
if len(actorJson['attachment']) == 0:
|
||
print("actorJson['attachment'] has no contents")
|
||
assert len(actorJson['attachment']) > 0
|
||
propertyFound = False
|
||
for propertyValue in actorJson['attachment']:
|
||
if propertyValue['name'] == 'PGP':
|
||
print('PGP property set within attachment')
|
||
assert pubKey in propertyValue['value']
|
||
propertyFound = True
|
||
assert propertyFound
|
||
|
||
# stop the server
|
||
thrAlice.kill()
|
||
thrAlice.join()
|
||
assert thrAlice.is_alive() is False
|
||
|
||
os.chdir(baseDir)
|
||
if os.path.isdir(baseDir + '/.tests'):
|
||
shutil.rmtree(baseDir + '/.tests')
|
||
|
||
|
||
def _testRemovePostInteractions() -> None:
|
||
print('testRemovePostInteractions')
|
||
postJsonObject = {
|
||
"type": "Create",
|
||
"object": {
|
||
"to": ["#Public"],
|
||
"likes": {
|
||
"items": ["a", "b", "c"]
|
||
},
|
||
"replies": {
|
||
"replyStuff": ["a", "b", "c"]
|
||
},
|
||
"shares": {
|
||
"sharesStuff": ["a", "b", "c"]
|
||
},
|
||
"bookmarks": {
|
||
"bookmarksStuff": ["a", "b", "c"]
|
||
},
|
||
"ignores": {
|
||
"ignoresStuff": ["a", "b", "c"]
|
||
}
|
||
}
|
||
}
|
||
removePostInteractions(postJsonObject, True)
|
||
assert postJsonObject['object']['likes']['items'] == []
|
||
assert postJsonObject['object']['replies'] == {}
|
||
assert postJsonObject['object']['shares'] == {}
|
||
assert postJsonObject['object']['bookmarks'] == {}
|
||
assert postJsonObject['object']['ignores'] == {}
|
||
assert not removePostInteractions(postJsonObject, False)
|
||
|
||
|
||
def _testSpoofGeolocation() -> None:
|
||
print('testSpoofGeolocation')
|
||
nogoLine = \
|
||
'NEW YORK, USA: 73.951W,40.879, 73.974W,40.83, ' + \
|
||
'74.029W,40.756, 74.038W,40.713, 74.056W,40.713, ' + \
|
||
'74.127W,40.647, 74.038W,40.629, 73.995W,40.667, ' + \
|
||
'74.014W,40.676, 73.994W,40.702, 73.967W,40.699, ' + \
|
||
'73.958W,40.729, 73.956W,40.745, 73.918W,40.781, ' + \
|
||
'73.937W,40.793, 73.946W,40.782, 73.977W,40.738, ' + \
|
||
'73.98W,40.713, 74.012W,40.705, 74.006W,40.752, ' + \
|
||
'73.955W,40.824'
|
||
polygon = parseNogoString(nogoLine)
|
||
assert len(polygon) > 0
|
||
assert polygon[0][1] == -73.951
|
||
assert polygon[0][0] == 40.879
|
||
citiesList = [
|
||
'NEW YORK, USA:40.7127281:W74.0060152:784',
|
||
'LOS ANGELES, USA:34.0536909:W118.242766:1214',
|
||
'SAN FRANCISCO, USA:37.74594738515095:W122.44299445520019:121',
|
||
'HOUSTON, USA:29.6072:W95.1586:1553',
|
||
'MANCHESTER, ENGLAND:53.4794892:W2.2451148:1276',
|
||
'BERLIN, GERMANY:52.5170365:13.3888599:891',
|
||
'ANKARA, TURKEY:39.93:32.85:24521',
|
||
'LONDON, ENGLAND:51.5073219:W0.1276474:1738',
|
||
'SEATTLE, USA:47.59840153253106:W122.31143714060059:217'
|
||
]
|
||
testSquare = [
|
||
[[0.03, 0.01], [0.02, 10], [10.01, 10.02], [10.03, 0.02]]
|
||
]
|
||
assert pointInNogo(testSquare, 5, 5)
|
||
assert pointInNogo(testSquare, 2, 3)
|
||
assert not pointInNogo(testSquare, 20, 5)
|
||
assert not pointInNogo(testSquare, 11, 6)
|
||
assert not pointInNogo(testSquare, 5, -5)
|
||
assert not pointInNogo(testSquare, 5, 11)
|
||
assert not pointInNogo(testSquare, -5, -5)
|
||
assert not pointInNogo(testSquare, -5, 5)
|
||
nogoList = []
|
||
currTime = datetime.datetime.utcnow()
|
||
decoySeed = 7634681
|
||
cityRadius = 0.1
|
||
coords = spoofGeolocation('', 'los angeles', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
assert coords[0] >= 34.0536909 - cityRadius
|
||
assert coords[0] <= 34.0536909 + cityRadius
|
||
assert coords[1] >= 118.242766 - cityRadius
|
||
assert coords[1] <= 118.242766 + cityRadius
|
||
assert coords[2] == 'N'
|
||
assert coords[3] == 'W'
|
||
assert len(coords[4]) > 4
|
||
assert len(coords[5]) > 4
|
||
assert coords[6] > 0
|
||
nogoList = []
|
||
coords = spoofGeolocation('', 'unknown', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
assert coords[0] >= 51.8744 - cityRadius
|
||
assert coords[0] <= 51.8744 + cityRadius
|
||
assert coords[1] >= 0.368333 - cityRadius
|
||
assert coords[1] <= 0.368333 + cityRadius
|
||
assert coords[2] == 'N'
|
||
assert coords[3] == 'W'
|
||
assert len(coords[4]) == 0
|
||
assert len(coords[5]) == 0
|
||
assert coords[6] == 0
|
||
kmlStr = '<?xml version="1.0" encoding="UTF-8"?>\n'
|
||
kmlStr += '<kml xmlns="http://www.opengis.net/kml/2.2">\n'
|
||
kmlStr += '<Document>\n'
|
||
nogoLine2 = \
|
||
'NEW YORK, USA: 74.115W,40.663, 74.065W,40.602, ' + \
|
||
'74.118W,40.555, 74.047W,40.516, 73.882W,40.547, ' + \
|
||
'73.909W,40.618, 73.978W,40.579, 74.009W,40.602, ' + \
|
||
'74.033W,40.61, 74.039W,40.623, 74.032W,40.641, ' + \
|
||
'73.996W,40.665'
|
||
polygon2 = parseNogoString(nogoLine2)
|
||
nogoList = [polygon, polygon2]
|
||
for i in range(1000):
|
||
dayNumber = randint(10, 30)
|
||
hour = randint(1, 23)
|
||
hourStr = str(hour)
|
||
if hour < 10:
|
||
hourStr = '0' + hourStr
|
||
dateTimeStr = "2021-05-" + str(dayNumber) + " " + hourStr + ":14"
|
||
currTime = datetime.datetime.strptime(dateTimeStr, "%Y-%m-%d %H:%M")
|
||
coords = spoofGeolocation('', 'new york, usa', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
longitude = coords[1]
|
||
if coords[3] == 'W':
|
||
longitude = -coords[1]
|
||
kmlStr += '<Placemark id="' + str(i) + '">\n'
|
||
kmlStr += ' <name>' + str(i) + '</name>\n'
|
||
kmlStr += ' <Point>\n'
|
||
kmlStr += ' <coordinates>' + str(longitude) + ',' + \
|
||
str(coords[0]) + ',0</coordinates>\n'
|
||
kmlStr += ' </Point>\n'
|
||
kmlStr += '</Placemark>\n'
|
||
|
||
nogoLine = \
|
||
'LONDON, ENGLAND: 0.23888E,51.459, 0.1216E,51.5, ' + \
|
||
'0.016E,51.479, 0.097W,51.502, 0.126W,51.482, ' + \
|
||
'0.196W,51.457, 0.292W,51.465, 0.309W,51.49, ' + \
|
||
'0.226W,51.495, 0.198W,51.47, 0.174W,51.488, ' + \
|
||
'0.136W,51.489, 0.1189W,51.515, 0.038E,51.513, ' + \
|
||
'0.0692E,51.51, 0.12833E,51.526, 0.3289E,51.475'
|
||
polygon = parseNogoString(nogoLine)
|
||
nogoLine2 = \
|
||
'LONDON, ENGLAND: 0.054W,51.535, 0.044W,51.53, ' + \
|
||
'0.008W,51.55, 0.0429W,51.57, 0.038W,51.6, ' + \
|
||
'0.0209W,51.603, 0.032W,51.613, 0.00191E,51.66, ' + \
|
||
'0.024W,51.666, 0.0313W,51.659, 0.0639W,51.579, ' + \
|
||
'0.059W,51.568, 0.0329W,51.552'
|
||
polygon2 = parseNogoString(nogoLine2)
|
||
nogoList = [polygon, polygon2]
|
||
for i in range(1000):
|
||
dayNumber = randint(10, 30)
|
||
hour = randint(1, 23)
|
||
hourStr = str(hour)
|
||
if hour < 10:
|
||
hourStr = '0' + hourStr
|
||
dateTimeStr = "2021-05-" + str(dayNumber) + " " + hourStr + ":14"
|
||
currTime = datetime.datetime.strptime(dateTimeStr, "%Y-%m-%d %H:%M")
|
||
coords = spoofGeolocation('', 'london, england', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
longitude = coords[1]
|
||
if coords[3] == 'W':
|
||
longitude = -coords[1]
|
||
kmlStr += '<Placemark id="' + str(i) + '">\n'
|
||
kmlStr += ' <name>' + str(i) + '</name>\n'
|
||
kmlStr += ' <Point>\n'
|
||
kmlStr += ' <coordinates>' + str(longitude) + ',' + \
|
||
str(coords[0]) + ',0</coordinates>\n'
|
||
kmlStr += ' </Point>\n'
|
||
kmlStr += '</Placemark>\n'
|
||
|
||
nogoLine = \
|
||
'SAN FRANCISCO, USA: 121.988W,37.408, 121.924W,37.452, ' + \
|
||
'121.951W,37.498, 121.992W,37.505, 122.056W,37.54, ' + \
|
||
'122.077W,37.578, 122.098W,37.618, 122.131W,37.637, ' + \
|
||
'122.189W,37.706, 122.227W,37.775, 122.279W,37.798, ' + \
|
||
'122.315W,37.802, 122.291W,37.832, 122.309W,37.902, ' + \
|
||
'122.382W,37.915, 122.368W,37.927, 122.514W,37.882, ' + \
|
||
'122.473W,37.83, 122.481W,37.788, 122.394W,37.796, ' + \
|
||
'122.384W,37.729, 122.4W,37.688, 122.382W,37.654, ' + \
|
||
'122.406W,37.637, 122.392W,37.612, 122.356W,37.586, ' + \
|
||
'122.332W,37.586, 122.275W,37.529, 122.228W,37.488, ' + \
|
||
'122.181W,37.482, 122.134W,37.48, 122.128W,37.471, ' + \
|
||
'122.122W,37.448, 122.095W,37.428, 122.07W,37.413, ' + \
|
||
'122.036W,37.402, 122.035W,37.421'
|
||
polygon = parseNogoString(nogoLine)
|
||
nogoLine2 = \
|
||
'SAN FRANCISCO, USA: 122.446W,37.794, 122.511W,37.778, ' + \
|
||
'122.51W,37.771, 122.454W,37.775, 122.452W,37.766, ' + \
|
||
'122.510W,37.763, 122.506W,37.735, 122.498W,37.733, ' + \
|
||
'122.496W,37.729, 122.491W,37.729, 122.475W,37.73, ' + \
|
||
'122.474W,37.72, 122.484W,37.72, 122.485W,37.703, ' + \
|
||
'122.495W,37.702, 122.493W,37.679, 122.486W,37.667, ' + \
|
||
'122.492W,37.664, 122.493W,37.629, 122.456W,37.625, ' + \
|
||
'122.450W,37.617, 122.455W,37.621, 122.41W,37.586, ' + \
|
||
'122.383W,37.561, 122.335W,37.509, 122.655W,37.48, ' + \
|
||
'122.67W,37.9, 122.272W,37.93, 122.294W,37.801, ' + \
|
||
'122.448W,37.804'
|
||
polygon2 = parseNogoString(nogoLine2)
|
||
nogoList = [polygon, polygon2]
|
||
for i in range(1000):
|
||
dayNumber = randint(10, 30)
|
||
hour = randint(1, 23)
|
||
hourStr = str(hour)
|
||
if hour < 10:
|
||
hourStr = '0' + hourStr
|
||
dateTimeStr = "2021-05-" + str(dayNumber) + " " + hourStr + ":14"
|
||
currTime = datetime.datetime.strptime(dateTimeStr, "%Y-%m-%d %H:%M")
|
||
coords = spoofGeolocation('', 'SAN FRANCISCO, USA', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
longitude = coords[1]
|
||
if coords[3] == 'W':
|
||
longitude = -coords[1]
|
||
kmlStr += '<Placemark id="' + str(i) + '">\n'
|
||
kmlStr += ' <name>' + str(i) + '</name>\n'
|
||
kmlStr += ' <Point>\n'
|
||
kmlStr += ' <coordinates>' + str(longitude) + ',' + \
|
||
str(coords[0]) + ',0</coordinates>\n'
|
||
kmlStr += ' </Point>\n'
|
||
kmlStr += '</Placemark>\n'
|
||
|
||
nogoLine = \
|
||
'SEATTLE, USA: 122.247W,47.918, 122.39W,47.802, ' + \
|
||
'122.389W,47.769, 122.377W,47.758, 122.371W,47.726, ' + \
|
||
'122.379W,47.706, 122.4W,47.696, 122.405W,47.673, ' + \
|
||
'122.416W,47.65, 122.414W,47.642, 122.391W,47.632, ' + \
|
||
'122.373W,47.633, 122.336W,47.602, 122.288W,47.501, ' + \
|
||
'122.299W,47.503, 122.386W,47.592, 122.412W,47.574, ' + \
|
||
'122.394W,47.549, 122.388W,47.507, 122.35W,47.481, ' + \
|
||
'122.365W,47.459, 122.33W,47.406, 122.323W,47.392, ' + \
|
||
'122.321W,47.346, 122.441W,47.302, 122.696W,47.085, ' + \
|
||
'122.926W,47.066, 122.929W,48.383'
|
||
polygon = parseNogoString(nogoLine)
|
||
nogoLine2 = \
|
||
'SEATTLE, USA: 122.267W,47.758, 122.29W,47.471, ' + \
|
||
'122.272W,47.693, 122.256W,47.672, 122.278W,47.652, ' + \
|
||
'122.29W,47.583, 122.262W,47.548, 122.265W,47.52, ' + \
|
||
'122.218W,47.498, 122.194W,47.501, 122.193W,47.55, ' + \
|
||
'122.173W,47.58, 122.22W,47.617, 122.238W,47.617, ' + \
|
||
'122.239W,47.637, 122.2W,47.644, 122.207W,47.703, ' + \
|
||
'122.22W,47.705, 122.231W,47.699, 122.255W,47.751'
|
||
polygon2 = parseNogoString(nogoLine2)
|
||
nogoLine3 = \
|
||
'SEATTLE, USA: 122.347W,47.675, 122.344W,47.681, ' + \
|
||
'122.337W,47.685, 122.324W,47.679, 122.331W,47.677, ' + \
|
||
'122.34W,47.669, 122.34W,47.664, 122.348W,47.665'
|
||
polygon3 = parseNogoString(nogoLine3)
|
||
nogoLine4 = \
|
||
'SEATTLE, USA: 122.423W,47.669, 122.345W,47.641, ' + \
|
||
'122.34W,47.625, 122.327W,47.626, 122.274W,47.64, ' + \
|
||
'122.268W,47.654, 122.327W,47.654, 122.336W,47.647, ' + \
|
||
'122.429W,47.684'
|
||
polygon4 = parseNogoString(nogoLine4)
|
||
nogoList = [polygon, polygon2, polygon3, polygon4]
|
||
for i in range(1000):
|
||
dayNumber = randint(10, 30)
|
||
hour = randint(1, 23)
|
||
hourStr = str(hour)
|
||
if hour < 10:
|
||
hourStr = '0' + hourStr
|
||
dateTimeStr = "2021-05-" + str(dayNumber) + " " + hourStr + ":14"
|
||
currTime = datetime.datetime.strptime(dateTimeStr, "%Y-%m-%d %H:%M")
|
||
coords = spoofGeolocation('', 'SEATTLE, USA', currTime,
|
||
decoySeed, citiesList, nogoList)
|
||
longitude = coords[1]
|
||
if coords[3] == 'W':
|
||
longitude = -coords[1]
|
||
kmlStr += '<Placemark id="' + str(i) + '">\n'
|
||
kmlStr += ' <name>' + str(i) + '</name>\n'
|
||
kmlStr += ' <Point>\n'
|
||
kmlStr += ' <coordinates>' + str(longitude) + ',' + \
|
||
str(coords[0]) + ',0</coordinates>\n'
|
||
kmlStr += ' </Point>\n'
|
||
kmlStr += '</Placemark>\n'
|
||
|
||
kmlStr += '</Document>\n'
|
||
kmlStr += '</kml>'
|
||
with open('unittest_decoy.kml', 'w+') as kmlFile:
|
||
kmlFile.write(kmlStr)
|
||
|
||
|
||
def _testSkills() -> None:
|
||
print('testSkills')
|
||
actorJson = {
|
||
'hasOccupation': [
|
||
{
|
||
'@type': 'Occupation',
|
||
'name': "Sysop",
|
||
"occupationLocation": {
|
||
"@type": "City",
|
||
"name": "Fediverse"
|
||
},
|
||
'skills': []
|
||
}
|
||
]
|
||
}
|
||
skillsDict = {
|
||
'bakery': 40,
|
||
'gardening': 70
|
||
}
|
||
setSkillsFromDict(actorJson, skillsDict)
|
||
assert actorHasSkill(actorJson, 'bakery')
|
||
assert actorHasSkill(actorJson, 'gardening')
|
||
assert actorSkillValue(actorJson, 'bakery') == 40
|
||
assert actorSkillValue(actorJson, 'gardening') == 70
|
||
|
||
|
||
def _testRoles() -> None:
|
||
print('testRoles')
|
||
actorJson = {
|
||
'hasOccupation': [
|
||
{
|
||
'@type': 'Occupation',
|
||
'name': "Sysop",
|
||
'occupationLocation': {
|
||
'@type': 'City',
|
||
'name': 'Fediverse'
|
||
},
|
||
'skills': []
|
||
}
|
||
]
|
||
}
|
||
testRolesList = ["admin", "moderator"]
|
||
setRolesFromList(actorJson, testRolesList)
|
||
assert actorHasRole(actorJson, "admin")
|
||
assert actorHasRole(actorJson, "moderator")
|
||
assert not actorHasRole(actorJson, "editor")
|
||
assert not actorHasRole(actorJson, "counselor")
|
||
assert not actorHasRole(actorJson, "artist")
|
||
|
||
|
||
def _testUserAgentDomain() -> None:
|
||
print('testUserAgentDomain')
|
||
userAgent = \
|
||
'http.rb/4.4.1 (Mastodon/9.10.11; +https://mastodon.something/)'
|
||
assert userAgentDomain(userAgent, False) == 'mastodon.something'
|
||
userAgent = \
|
||
'Mozilla/70.0 (X11; Linux x86_64; rv:1.0) Gecko/20450101 Firefox/1.0'
|
||
assert userAgentDomain(userAgent, False) is None
|
||
|
||
|
||
def _testSwitchWords() -> None:
|
||
print('testSwitchWords')
|
||
rules = [
|
||
"rock -> hamster",
|
||
"orange -> lemon"
|
||
]
|
||
baseDir = os.getcwd()
|
||
nickname = 'testuser'
|
||
domain = 'testdomain.com'
|
||
|
||
content = 'This is a test'
|
||
result = switchWords(baseDir, nickname, domain, content, rules)
|
||
assert result == content
|
||
|
||
content = 'This is orange test'
|
||
result = switchWords(baseDir, nickname, domain, content, rules)
|
||
assert result == 'This is lemon test'
|
||
|
||
content = 'This is a test rock'
|
||
result = switchWords(baseDir, nickname, domain, content, rules)
|
||
assert result == 'This is a test hamster'
|
||
|
||
|
||
def _testLimitWordLengths() -> None:
|
||
print('testLimitWordLengths')
|
||
maxWordLength = 13
|
||
text = "This is a test"
|
||
result = limitWordLengths(text, maxWordLength)
|
||
assert result == text
|
||
|
||
text = "This is an exceptionallylongword test"
|
||
result = limitWordLengths(text, maxWordLength)
|
||
assert result == "This is an exceptionally test"
|
||
|
||
|
||
def _testLimitRepetedWords() -> None:
|
||
print('limitRepeatedWords')
|
||
text = \
|
||
"This is a preamble.\n\n" + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same\n\n" + \
|
||
"Some other text."
|
||
expected = \
|
||
"This is a preamble.\n\n" + \
|
||
"Same Same Same Same Same Same\n\n" + \
|
||
"Some other text."
|
||
result = limitRepeatedWords(text, 6)
|
||
assert result == expected
|
||
|
||
text = \
|
||
"This is other preamble.\n\n" + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same " + \
|
||
"Same Same Same Same Same Same Same Same Same Same"
|
||
expected = \
|
||
"This is other preamble.\n\n" + \
|
||
"Same Same Same Same Same Same"
|
||
result = limitRepeatedWords(text, 6)
|
||
assert result == expected
|
||
|
||
|
||
def _testSetActorLanguages():
|
||
print('testSetActorLanguages')
|
||
actorJson = {
|
||
"attachment": []
|
||
}
|
||
setActorLanguages(None, actorJson, 'es, fr, en')
|
||
assert len(actorJson['attachment']) == 1
|
||
assert actorJson['attachment'][0]['name'] == 'Languages'
|
||
assert actorJson['attachment'][0]['type'] == 'PropertyValue'
|
||
assert len(actorJson['attachment'][0]['value']) == 3
|
||
assert 'en' in actorJson['attachment'][0]['value']
|
||
assert 'fr' in actorJson['attachment'][0]['value']
|
||
assert 'es' in actorJson['attachment'][0]['value']
|
||
languagesStr = getActorLanguages(actorJson)
|
||
assert languagesStr == 'en / es / fr'
|
||
|
||
|
||
def _testGetLinksFromContent():
|
||
print('testGetLinksFromContent')
|
||
content = 'This text has no links'
|
||
links = getLinksFromContent(content)
|
||
assert not links
|
||
|
||
link1 = 'https://somewebsite.net'
|
||
link2 = 'http://somewhere.or.other'
|
||
content = \
|
||
'This is <a href="' + link1 + '">@linked</a>. ' + \
|
||
'And <a href="' + link2 + '">another</a>.'
|
||
links = getLinksFromContent(content)
|
||
assert len(links.items()) == 2
|
||
assert links.get('@linked')
|
||
assert links['@linked'] == link1
|
||
assert links.get('another')
|
||
assert links['another'] == link2
|
||
|
||
contentPlain = '<p>' + removeHtml(content) + '</p>'
|
||
assert '>@linked</a>' not in contentPlain
|
||
content = addLinksToContent(contentPlain, links)
|
||
assert '>@linked</a>' in content
|
||
|
||
|
||
def _testValidPassword():
|
||
print('testValidPassword')
|
||
assert not validPassword('123')
|
||
assert not validPassword('')
|
||
assert not validPassword('Abcdefg1?23456')
|
||
assert validPassword('Abcdef!g123456')
|
||
|
||
|
||
def _testAuthorizeSharedItems():
|
||
print('testAuthorizeSharedItems')
|
||
sharedItemsFederatedDomains = \
|
||
['dog.domain', 'cat.domain', 'birb.domain']
|
||
tokensJson = \
|
||
generateSharedItemFederationTokens(sharedItemsFederatedDomains, None)
|
||
tokensJson = \
|
||
createSharedItemFederationToken(None, 'cat.domain', tokensJson)
|
||
assert tokensJson
|
||
assert not tokensJson.get('dog.domain')
|
||
assert tokensJson.get('cat.domain')
|
||
assert not tokensJson.get('birb.domain')
|
||
assert len(tokensJson['dog.domain']) == 0
|
||
assert len(tokensJson['cat.domain']) >= 64
|
||
assert len(tokensJson['birb.domain']) == 0
|
||
assert not authorizeSharedItems(sharedItemsFederatedDomains, None,
|
||
'cat.domain', 'M' * 86,
|
||
False, tokensJson)
|
||
assert authorizeSharedItems(sharedItemsFederatedDomains, None,
|
||
'cat.domain', tokensJson['cat.domain'],
|
||
False, tokensJson)
|
||
tokensJson = \
|
||
updateSharedItemFederationToken(None,
|
||
'dog.domain', 'testToken', tokensJson)
|
||
assert tokensJson['dog.domain'] == 'testToken'
|
||
|
||
# the shared item federation list changes
|
||
sharedItemsFederatedDomains = \
|
||
['possum.domain', 'cat.domain', 'birb.domain']
|
||
tokensJson = mergeSharedItemTokens(None, '',
|
||
sharedItemsFederatedDomains,
|
||
tokensJson)
|
||
assert 'dog.domain' not in tokensJson
|
||
assert 'cat.domain' in tokensJson
|
||
assert len(tokensJson['cat.domain']) >= 64
|
||
assert 'birb.domain' in tokensJson
|
||
assert 'possum.domain' in tokensJson
|
||
assert len(tokensJson['birb.domain']) == 0
|
||
assert len(tokensJson['possum.domain']) == 0
|
||
|
||
|
||
def runAllTests():
|
||
print('Running tests...')
|
||
updateDefaultThemesList(os.getcwd())
|
||
_testAuthorizeSharedItems()
|
||
_testValidPassword()
|
||
_testGetLinksFromContent()
|
||
_testSetActorLanguages()
|
||
_testLimitRepetedWords()
|
||
_testLimitWordLengths()
|
||
_testSwitchWords()
|
||
_testFunctions()
|
||
_testUserAgentDomain()
|
||
_testRoles()
|
||
_testSkills()
|
||
_testSpoofGeolocation()
|
||
_testRemovePostInteractions()
|
||
_testExtractPGPPublicKey()
|
||
_testEmojiImages()
|
||
_testCamelCaseSplit()
|
||
_testSpeakerReplaceLinks()
|
||
_testExtractTextFieldsInPOST()
|
||
_testMarkdownToHtml()
|
||
_testValidHashTag()
|
||
_testPrepareHtmlPostNickname()
|
||
_testDomainHandling()
|
||
_testMastoApi()
|
||
_testLinksWithinPost()
|
||
_testReplyToPublicPost()
|
||
_testGetMentionedPeople()
|
||
_testGuessHashtagCategory()
|
||
_testValidNickname()
|
||
_testParseFeedDate()
|
||
_testFirstParagraphFromString()
|
||
_testGetNewswireTags()
|
||
_testHashtagRuleTree()
|
||
_testRemoveHtmlTag()
|
||
_testReplaceEmailQuote()
|
||
_testConstantTimeStringCheck()
|
||
_testTranslations()
|
||
_testValidContentWarning()
|
||
_testRemoveIdEnding()
|
||
_testJsonPostAllowsComments()
|
||
_runHtmlReplaceQuoteMarks()
|
||
_testDangerousCSS()
|
||
_testDangerousMarkup()
|
||
_testRemoveHtml()
|
||
_testSiteIsActive()
|
||
_testJsonld()
|
||
_testRemoveTextFormatting()
|
||
_testWebLinks()
|
||
_testRecentPostsCache()
|
||
_testTheme()
|
||
_testSaveLoadJson()
|
||
_testJsonString()
|
||
_testGetStatusNumber()
|
||
_testAddEmoji()
|
||
_testActorParsing()
|
||
_testHttpsig()
|
||
_testHttpSigNew()
|
||
_testCache()
|
||
_testThreads()
|
||
_testCreatePerson()
|
||
_testAuthentication()
|
||
_testFollowersOfPerson()
|
||
_testNoOfFollowersOnDomain()
|
||
_testFollows()
|
||
_testGroupFollowers()
|
||
print('Tests succeeded\n')
|