epicyon/tests.py

1943 lines
74 KiB
Python
Raw Normal View History

2020-04-05 13:25:47 +00:00
__filename__ = "tests.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.1.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"
2019-06-30 20:14:03 +00:00
import time
2020-04-05 13:25:47 +00:00
import os
2019-06-30 21:20:02 +00:00
import shutil
2019-08-16 10:36:41 +00:00
import json
2019-08-15 18:21:43 +00:00
from time import gmtime, strftime
2019-07-09 14:20:23 +00:00
from pprint import pprint
2019-06-30 20:14:03 +00:00
from httpsig import signPostHeaders
from httpsig import verifyPostHeaders
from httpsig import messageContentDigest
2019-06-30 20:14:03 +00:00
from cache import storePersonInCache
from cache import getPersonFromCache
from threads import threadWithTrace
2019-06-30 21:20:02 +00:00
from daemon import runDaemon
from session import createSession
from posts import deleteAllPosts
from posts import createPublicPost
2019-06-30 22:56:37 +00:00
from posts import sendPost
2019-07-05 14:39:24 +00:00
from posts import noOfFollowersOnDomain
from posts import groupFollowersByDomain
2019-07-12 22:29:10 +00:00
from posts import archivePostsForPerson
2019-07-16 10:19:04 +00:00
from posts import sendPostViaServer
2019-07-01 11:09:09 +00:00
from follow import clearFollows
from follow import clearFollowers
2019-07-16 21:38:06 +00:00
from follow import sendFollowRequestViaServer
2019-07-17 11:54:13 +00:00
from follow import sendUnfollowRequestViaServer
2020-06-22 16:55:19 +00:00
from utils import siteIsActive
from utils import updateRecentPostsCache
2019-07-06 19:24:52 +00:00
from utils import followPerson
2019-08-21 16:35:46 +00:00
from utils import getNicknameFromActor
from utils import getDomainFromActor
2019-09-29 18:48:34 +00:00
from utils import copytree
from utils import loadJson
from utils import saveJson
2019-10-12 12:45:53 +00:00
from utils import getStatusNumber
2019-06-30 21:20:02 +00:00
from follow import followerOfPerson
2019-07-01 11:09:09 +00:00
from follow import unfollowPerson
from follow import unfollowerOfPerson
from follow import getFollowersOfPerson
2019-07-06 13:49:25 +00:00
from follow import sendFollowRequest
2019-07-03 10:04:23 +00:00
from person import createPerson
from person import setDisplayNickname
2019-07-03 10:04:23 +00:00
from person import setBio
2020-06-15 12:37:53 +00:00
# from person import generateRSAKey
2019-07-19 10:01:24 +00:00
from skills import setSkillLevel
2019-07-18 15:09:23 +00:00
from roles import setRole
2019-07-18 16:21:26 +00:00
from roles import outboxDelegate
2019-07-03 18:24:44 +00:00
from auth import createBasicAuthHeader
from auth import authorizeBasic
from auth import storeBasicCredentials
2019-07-11 12:29:31 +00:00
from like import likePost
from like import sendLikeViaServer
2019-07-11 17:55:10 +00:00
from announce import announcePublic
2019-07-16 19:07:45 +00:00
from announce import sendAnnounceViaServer
2019-07-12 19:26:54 +00:00
from media import getMediaPath
2019-08-30 15:50:20 +00:00
from media import getAttachmentMediaType
2019-07-17 17:16:48 +00:00
from delete import sendDeleteViaServer
from inbox import validInbox
2019-07-18 11:35:48 +00:00
from inbox import validInboxFilenames
2020-07-10 14:15:01 +00:00
from content import dangerousMarkup
from content import removeHtml
2019-09-01 08:55:05 +00:00
from content import addWebLinks
2019-09-29 18:48:34 +00:00
from content import replaceEmojiFromTags
from content import addHtmlTags
2019-11-04 20:39:14 +00:00
from content import removeLongWords
2020-05-12 09:34:58 +00:00
from content import replaceContentDuplicates
from content import removeTextFormatting
2019-11-23 13:04:11 +00:00
from theme import setCSSparam
2020-06-15 12:37:53 +00:00
from jsonldsig import testSignJsonld
from jsonldsig import jsonldVerify
2019-06-30 20:14:03 +00:00
2020-04-05 13:25:47 +00:00
testServerAliceRunning = False
testServerBobRunning = False
testServerEveRunning = False
thrAlice = None
thrBob = None
thrEve = None
2019-06-30 21:27:25 +00:00
2019-06-30 20:14:03 +00:00
def testHttpsigBase(withDigest):
print('testHttpsig(' + str(withDigest) + ')')
2019-08-09 09:46:33 +00:00
2020-04-05 13:25:47 +00:00
baseDir = os.getcwd()
path = baseDir + '/.testHttpsigBase'
2019-08-09 09:46:33 +00:00
if os.path.isdir(path):
shutil.rmtree(path)
os.mkdir(path)
os.chdir(path)
2020-04-05 13:25:47 +00:00
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, password)
2019-08-09 09:46:33 +00:00
assert privateKeyPem
2020-04-05 13:25:47 +00:00
messageBodyJson = {
2020-03-22 20:36:19 +00:00
"a key": "a value",
"another key": "A string",
"yet another key": "Another string"
}
2020-04-05 13:25:47 +00:00
messageBodyJsonStr = json.dumps(messageBodyJson)
2019-07-01 09:31:02 +00:00
2020-04-05 13:25:47 +00:00
headersDomain = domain
if port:
2020-04-05 13:25:47 +00:00
if port != 80 and port != 443:
if ':' not in domain:
2020-04-05 13:25:47 +00:00
headersDomain = domain + ':' + str(port)
2019-07-01 09:31:02 +00:00
2020-04-05 13:25:47 +00:00
dateStr = strftime("%a, %d %b %Y %H:%M:%S %Z", gmtime())
boxpath = '/inbox'
2019-06-30 20:14:03 +00:00
if not withDigest:
2020-04-05 13:25:47 +00:00
headers = {
2020-03-22 20:36:19 +00:00
'host': headersDomain,
'date': dateStr,
'content-type': 'application/json'
}
2020-04-05 13:25:47 +00:00
signatureHeader = \
signPostHeaders(dateStr, privateKeyPem, nickname,
domain, port,
domain, port,
2019-08-16 13:47:01 +00:00
boxpath, httpPrefix, None)
2019-06-30 20:14:03 +00:00
else:
2020-04-05 13:25:47 +00:00
bodyDigest = messageContentDigest(messageBodyJsonStr)
contentLength = len(messageBodyJsonStr)
headers = {
2020-03-22 20:36:19 +00:00
'host': headersDomain,
'date': dateStr,
'digest': f'SHA-256={bodyDigest}',
'content-type': contentType,
'content-length': str(contentLength)
}
2020-04-05 13:25:47 +00:00
signatureHeader = \
signPostHeaders(dateStr, privateKeyPem, nickname,
domain, port,
domain, port,
boxpath, httpPrefix, messageBodyJsonStr)
2019-07-01 09:31:02 +00:00
2020-04-05 13:25:47 +00:00
headers['signature'] = signatureHeader
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
boxpath, False, None,
messageBodyJsonStr, False)
2019-11-12 18:21:52 +00:00
if withDigest:
# everything correct except for content-length
2020-04-05 13:25:47 +00:00
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
2019-06-30 20:14:03 +00:00
if not withDigest:
# fake domain
2020-04-05 13:25:47 +00:00
headers = {
2020-03-22 20:36:19 +00:00
'host': 'bogon.domain',
'date': dateStr,
'content-type': 'application/json'
}
2019-06-30 20:14:03 +00:00
else:
# correct domain but fake message
2020-04-05 13:25:47 +00:00
messageBodyJsonStr = \
'{"a key": "a value", "another key": "Fake GNUs", ' + \
'"yet another key": "More Fake GNUs"}'
contentLength = len(messageBodyJsonStr)
bodyDigest = messageContentDigest(messageBodyJsonStr)
headers = {
2020-03-22 20:36:19 +00:00
'host': domain,
'date': dateStr,
'digest': f'SHA-256={bodyDigest}',
'content-type': contentType,
'content-length': str(contentLength)
}
2020-04-05 13:25:47 +00:00
headers['signature'] = signatureHeader
assert verifyPostHeaders(httpPrefix, publicKeyPem, headers,
boxpath, True, None,
messageBodyJsonStr, False) is False
2019-11-12 18:21:52 +00:00
2019-08-09 09:46:33 +00:00
os.chdir(baseDir)
shutil.rmtree(path)
2019-06-30 20:14:03 +00:00
2020-04-05 13:25:47 +00:00
2019-06-30 20:14:03 +00:00
def testHttpsig():
testHttpsigBase(True)
2019-08-15 21:34:25 +00:00
testHttpsigBase(False)
2019-06-30 20:14:03 +00:00
2020-04-05 13:25:47 +00:00
2019-06-30 20:14:03 +00:00
def testCache():
print('testCache')
2020-04-05 13:25:47 +00:00
personUrl = "cat@cardboard.box"
personJson = {
"id": 123456,
"test": "This is a test"
}
personCache = {}
storePersonInCache(None, personUrl, personJson, personCache)
result = getPersonFromCache(None, personUrl, personCache)
assert result['id'] == 123456
assert result['test'] == 'This is a test'
2019-06-30 20:14:03 +00:00
def testThreadsFunction(param: str):
for i in range(10000):
time.sleep(2)
2020-04-05 13:25:47 +00:00
2019-06-30 20:14:03 +00:00
def testThreads():
print('testThreads')
2020-04-05 13:25:47 +00:00
thr = \
threadWithTrace(target=testThreadsFunction,
args=('test',),
daemon=True)
2019-06-30 20:14:03 +00:00
thr.start()
2020-04-05 13:25:47 +00:00
assert thr.isAlive() is True
2019-06-30 20:14:03 +00:00
time.sleep(1)
thr.kill()
thr.join()
2020-04-05 13:25:47 +00:00
assert thr.isAlive() is False
2019-06-30 21:20:02 +00:00
2020-04-05 13:25:47 +00:00
def createServerAlice(path: str, domain: str, port: int,
bobAddress: str, federationList: [],
hasFollows: bool, hasPosts: bool,
ocapAlways: bool, sendThreads: []):
print('Creating test server: Alice on port ' + str(port))
2019-06-30 21:20:02 +00:00
if os.path.isdir(path):
shutil.rmtree(path)
os.mkdir(path)
os.chdir(path)
2020-04-05 13:25:47 +00:00
nickname = 'alice'
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2020-04-05 13:25:47 +00:00
password = 'alicepass'
noreply = False
nolike = False
nopics = False
noannounce = False
cw = False
useBlurhash = True
maxReplies = 64
domainMaxPostsPerDay = 1000
accountMaxPostsPerDay = 1000
allowDeletion = True
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(path, nickname, domain, port, httpPrefix, True, password)
deleteAllPosts(path, nickname, domain, 'inbox')
deleteAllPosts(path, nickname, domain, 'outbox')
assert setSkillLevel(path, nickname, domain, 'hacking', 90)
assert setRole(path, nickname, domain, 'someproject', 'guru')
2019-07-06 13:49:25 +00:00
if hasFollows:
2020-04-05 13:25:47 +00:00
followPerson(path, nickname, domain, 'bob', bobAddress,
federationList, False)
followerOfPerson(path, nickname, domain, 'bob', bobAddress,
federationList, False)
2019-07-06 13:49:25 +00:00
if hasPosts:
2020-04-05 13:25:47 +00:00
clientToServer = False
createPublicPost(path, nickname, domain, port, httpPrefix,
"No wise fish would go anywhere without a porpoise",
False, True, clientToServer, None, None, useBlurhash)
createPublicPost(path, nickname, domain, port, httpPrefix,
"Curiouser and curiouser!", False, True,
clientToServer, None, None, useBlurhash)
createPublicPost(path, nickname, domain, port, httpPrefix,
"In the gardens of memory, in the palace " +
"of dreams, that is where you and I shall meet",
False, True, clientToServer, None, None, useBlurhash)
2019-06-30 21:27:25 +00:00
global testServerAliceRunning
2020-04-05 13:25:47 +00:00
testServerAliceRunning = True
maxMentions = 10
maxEmoji = 10
onionDomain = None
i2pDomain = None
2019-06-30 21:20:02 +00:00
print('Server running: Alice')
2020-04-05 13:25:47 +00:00
runDaemon(False, False, 5, True, True, 'en', __version__,
"instanceId", False, path, domain,
onionDomain, i2pDomain, port, port,
2020-04-05 13:25:47 +00:00
httpPrefix, federationList, maxMentions, maxEmoji, False,
noreply, nolike, nopics, noannounce, cw, ocapAlways,
2020-06-09 11:03:59 +00:00
proxyType, maxReplies,
2020-04-05 13:25:47 +00:00
domainMaxPostsPerDay, accountMaxPostsPerDay,
allowDeletion, True, True, False, sendThreads, False)
def createServerBob(path: str, domain: str, port: int,
aliceAddress: str, federationList: [],
hasFollows: bool, hasPosts: bool,
ocapAlways: bool, sendThreads: []):
print('Creating test server: Bob on port ' + str(port))
2019-06-30 21:20:02 +00:00
if os.path.isdir(path):
shutil.rmtree(path)
os.mkdir(path)
os.chdir(path)
2020-04-05 13:25:47 +00:00
nickname = 'bob'
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2020-04-05 13:25:47 +00:00
clientToServer = False
password = 'bobpass'
noreply = False
nolike = False
nopics = False
noannounce = False
cw = False
useBlurhash = False
maxReplies = 64
domainMaxPostsPerDay = 1000
accountMaxPostsPerDay = 1000
allowDeletion = True
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(path, nickname, domain, port, httpPrefix, True, password)
deleteAllPosts(path, nickname, domain, 'inbox')
deleteAllPosts(path, nickname, domain, 'outbox')
assert setRole(path, nickname, domain, 'bandname', 'bass player')
assert setRole(path, nickname, domain, 'bandname', 'publicist')
2019-07-06 13:49:25 +00:00
if hasFollows:
2020-04-05 13:25:47 +00:00
followPerson(path, nickname, domain,
'alice', aliceAddress, federationList, False)
followerOfPerson(path, nickname, domain,
'alice', aliceAddress, federationList, False)
2019-07-06 13:49:25 +00:00
if hasPosts:
2020-04-05 13:25:47 +00:00
createPublicPost(path, nickname, domain, port, httpPrefix,
"It's your life, live it your way.",
False, True, clientToServer, None, None, useBlurhash)
createPublicPost(path, nickname, domain, port, httpPrefix,
"One of the things I've realised is that " +
"I am very simple",
False, True, clientToServer, None, None, useBlurhash)
createPublicPost(path, nickname, domain, port, httpPrefix,
"Quantum physics is a bit of a passion of mine",
False, True, clientToServer, None, None, useBlurhash)
2019-06-30 21:27:25 +00:00
global testServerBobRunning
2020-04-05 13:25:47 +00:00
testServerBobRunning = True
maxMentions = 10
maxEmoji = 10
onionDomain = None
i2pDomain = None
2019-06-30 21:20:02 +00:00
print('Server running: Bob')
2020-04-05 13:25:47 +00:00
runDaemon(False, False, 5, True, True, 'en', __version__,
"instanceId", False, path, domain,
onionDomain, i2pDomain, port, port,
2020-04-05 13:25:47 +00:00
httpPrefix, federationList, maxMentions, maxEmoji, False,
noreply, nolike, nopics, noannounce, cw, ocapAlways,
2020-06-09 11:03:59 +00:00
proxyType, maxReplies,
2020-04-05 13:25:47 +00:00
domainMaxPostsPerDay, accountMaxPostsPerDay,
allowDeletion, True, True, False, sendThreads, False)
def createServerEve(path: str, domain: str, port: int, federationList: [],
hasFollows: bool, hasPosts: bool,
ocapAlways: 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)
2020-04-05 13:25:47 +00:00
nickname = 'eve'
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2020-04-05 13:25:47 +00:00
password = 'evepass'
noreply = False
nolike = False
nopics = False
noannounce = False
cw = False
maxReplies = 64
allowDeletion = True
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(path, nickname, domain, port, httpPrefix, True, password)
deleteAllPosts(path, nickname, domain, 'inbox')
deleteAllPosts(path, nickname, domain, 'outbox')
global testServerEveRunning
2020-04-05 13:25:47 +00:00
testServerEveRunning = True
maxMentions = 10
maxEmoji = 10
onionDomain = None
i2pDomain = None
print('Server running: Eve')
2020-04-05 13:25:47 +00:00
runDaemon(False, False, 5, True, True, 'en', __version__,
"instanceId", False, path, domain,
onionDomain, i2pDomain, port, port,
2020-04-05 13:25:47 +00:00
httpPrefix, federationList, maxMentions, maxEmoji, False,
noreply, nolike, nopics, noannounce, cw, ocapAlways,
2020-06-09 11:03:59 +00:00
proxyType, maxReplies, allowDeletion, True, True, False,
2020-04-05 13:25:47 +00:00
sendThreads, False)
2019-06-30 21:20:02 +00:00
def testPostMessageBetweenServers():
print('Testing sending message from one server to the inbox of another')
2019-06-30 21:27:25 +00:00
global testServerAliceRunning
global testServerBobRunning
2020-04-05 13:25:47 +00:00
testServerAliceRunning = False
testServerBobRunning = False
2019-06-30 21:27:25 +00:00
2020-04-05 13:25:47 +00:00
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2019-06-30 22:56:37 +00:00
2020-04-05 13:25:47 +00:00
baseDir = os.getcwd()
if os.path.isdir(baseDir + '/.tests'):
shutil.rmtree(baseDir + '/.tests')
os.mkdir(baseDir + '/.tests')
2019-06-30 21:20:02 +00:00
2020-04-05 13:25:47 +00:00
ocapAlways = False
2020-03-22 21:16:02 +00:00
2019-06-30 21:20:02 +00:00
# create the servers
2020-04-05 13:25:47 +00:00
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)
2019-07-11 12:29:31 +00:00
2020-02-19 12:27:21 +00:00
global thrAlice
if thrAlice:
while thrAlice.isAlive():
thrAlice.stop()
time.sleep(1)
thrAlice.kill()
2020-04-05 13:25:47 +00:00
thrAlice = \
threadWithTrace(target=createServerAlice,
args=(aliceDir, aliceDomain, alicePort, bobAddress,
federationList, False, False,
ocapAlways, aliceSendThreads),
daemon=True)
2020-02-19 12:27:21 +00:00
global thrBob
if thrBob:
while thrBob.isAlive():
thrBob.stop()
time.sleep(1)
thrBob.kill()
2020-04-05 13:25:47 +00:00
thrBob = \
threadWithTrace(target=createServerBob,
args=(bobDir, bobDomain, bobPort, aliceAddress,
federationList, False, False,
ocapAlways, bobSendThreads),
daemon=True)
2019-07-01 21:01:43 +00:00
thrAlice.start()
2019-06-30 21:20:02 +00:00
thrBob.start()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is True
assert thrBob.isAlive() is True
2019-06-30 21:20:02 +00:00
2019-06-30 21:27:25 +00:00
# wait for both servers to be running
while not (testServerAliceRunning and testServerBobRunning):
time.sleep(1)
2020-03-22 21:16:02 +00:00
2019-07-04 20:25:19 +00:00
time.sleep(1)
2019-06-30 21:20:02 +00:00
2019-07-11 12:29:31 +00:00
print('\n\n*******************************************************')
2019-06-30 22:56:37 +00:00
print('Alice sends to Bob')
os.chdir(aliceDir)
2020-06-09 11:03:59 +00:00
sessionAlice = createSession(proxyType)
2020-04-05 13:25:47 +00:00
inReplyTo = None
inReplyToAtomUri = None
subject = None
alicePostLog = []
followersOnly = False
saveToFile = True
clientToServer = False
ccUrl = None
alicePersonCache = {}
aliceCachedWebfingers = {}
attachedImageFilename = baseDir + '/img/logo.png'
mediaType = getAttachmentMediaType(attachedImageFilename)
attachedImageDescription = 'Logo'
useBlurhash = True
isArticle = False
2019-07-11 12:29:31 +00:00
# nothing in Alice's outbox
2020-04-05 13:25:47 +00:00
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, attachedImageFilename, mediaType,
attachedImageDescription, useBlurhash, federationList,
aliceSendThreads, alicePostLog, aliceCachedWebfingers,
alicePersonCache, isArticle, inReplyTo,
inReplyToAtomUri, subject)
print('sendResult: ' + str(sendResult))
queuePath = bobDir + '/accounts/bob@' + bobDomain + '/queue'
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
mPath = getMediaPath()
mediaPath = aliceDir + '/' + mPath
2019-07-07 22:25:22 +00:00
for i in range(30):
2019-07-04 20:36:12 +00:00
if os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
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:
2019-10-19 17:50:05 +00:00
break
2019-07-01 21:01:43 +00:00
time.sleep(1)
2019-07-09 08:52:53 +00:00
2019-07-12 19:26:54 +00:00
# Image attachment created
2020-04-05 13:25:47 +00:00
assert len([name for name in os.listdir(mediaPath)
if os.path.isfile(os.path.join(mediaPath, name))]) > 0
2019-07-11 12:29:31 +00:00
# inbox item created
2020-04-05 13:25:47 +00:00
assert len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
2019-07-11 12:29:31 +00:00
# queue item removed
2020-04-05 13:25:47 +00:00
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)
2019-11-09 09:50:58 +00:00
print('Check that message received from Alice contains the expected text')
for name in os.listdir(inboxPath):
2020-04-05 13:25:47 +00:00
filename = os.path.join(inboxPath, name)
2019-11-09 09:50:58 +00:00
assert os.path.isfile(filename)
2020-04-05 13:25:47 +00:00
receivedJson = loadJson(filename, 0)
if receivedJson:
2019-11-09 09:50:58 +00:00
pprint(receivedJson['object']['content'])
2019-11-09 09:52:09 +00:00
assert receivedJson
2020-04-05 13:25:47 +00:00
assert 'Why is a mouse when it spins?' in \
receivedJson['object']['content']
2019-11-09 09:52:09 +00:00
assert 'यह एक परीक्षण है' in receivedJson['object']['content']
2019-07-11 12:59:00 +00:00
print('\n\n*******************************************************')
print("Bob likes Alice's post")
2020-04-05 13:25:47 +00:00
followerOfPerson(bobDir, 'bob', bobDomain, 'alice',
aliceDomain + ':' + str(alicePort), federationList, False)
followPerson(aliceDir, 'alice', aliceDomain, 'bob',
bobDomain + ':' + str(bobPort), federationList, False)
2020-06-09 11:03:59 +00:00
sessionBob = createSession(proxyType)
2020-04-05 13:25:47 +00:00
bobPostLog = []
bobPersonCache = {}
bobCachedWebfingers = {}
statusNumber = None
outboxPostFilename = None
outboxPath = aliceDir + '/accounts/alice@' + aliceDomain + '/outbox'
2019-07-11 12:59:00 +00:00
for name in os.listdir(outboxPath):
if '#statuses#' in name:
2020-04-05 13:25:47 +00:00
statusNumber = \
int(name.split('#statuses#')[1].replace('.json', ''))
outboxPostFilename = outboxPath + '/' + name
assert statusNumber > 0
2019-07-11 12:59:00 +00:00
assert outboxPostFilename
2020-04-05 13:25:47 +00:00
assert likePost({}, sessionBob, bobDir, federationList,
'bob', bobDomain, bobPort, httpPrefix,
'alice', aliceDomain, alicePort, [],
statusNumber, False, bobSendThreads, bobPostLog,
bobPersonCache, bobCachedWebfingers,
True, __version__)
2019-07-11 12:59:00 +00:00
for i in range(20):
if 'likes' in open(outboxPostFilename).read():
break
time.sleep(1)
2020-04-05 13:25:47 +00:00
alicePostJson = loadJson(outboxPostFilename, 0)
if alicePostJson:
2019-07-11 12:59:00 +00:00
pprint(alicePostJson)
2020-03-22 21:16:02 +00:00
2019-07-11 12:59:00 +00:00
assert 'likes' in open(outboxPostFilename).read()
2019-07-11 17:55:10 +00:00
print('\n\n*******************************************************')
print("Bob repeats Alice's post")
2020-04-05 13:25:47 +00:00
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
2019-07-11 19:31:02 +00:00
for i in range(10):
time.sleep(1)
2020-03-27 19:54:41 +00:00
if not os.path.isdir(inboxPath):
continue
2020-04-05 13:25:47 +00:00
if len([name for name in os.listdir(outboxPath)
if os.path.isfile(os.path.join(outboxPath, name))]) > 0:
outboxMessageArrived = True
2020-03-27 19:54:41 +00:00
print('Announce created by Bob')
2020-04-05 13:25:47 +00:00
if len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))]) > 0:
announceMessageArrived = True
2020-03-27 19:54:41 +00:00
print('Announce message sent to Alice!')
if announceMessageArrived and outboxMessageArrived:
break
2020-04-05 13:25:47 +00:00
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
2019-06-30 21:20:02 +00:00
# stop the servers
thrAlice.kill()
thrAlice.join()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is False
2019-06-30 21:20:02 +00:00
thrBob.kill()
thrBob.join()
2020-04-05 13:25:47 +00:00
assert thrBob.isAlive() is False
2019-06-30 21:20:02 +00:00
os.chdir(baseDir)
shutil.rmtree(aliceDir)
shutil.rmtree(bobDir)
2020-04-05 13:25:47 +00:00
2019-10-15 09:31:10 +00:00
def testFollowBetweenServers():
print('Testing sending a follow request from one server to another')
global testServerAliceRunning
global testServerBobRunning
2020-04-05 13:25:47 +00:00
testServerAliceRunning = False
testServerBobRunning = False
2019-10-15 09:31:10 +00:00
2020-04-05 13:25:47 +00:00
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2020-04-05 13:25:47 +00:00
federationList = []
2019-10-15 09:31:10 +00:00
2020-04-05 13:25:47 +00:00
baseDir = os.getcwd()
if os.path.isdir(baseDir + '/.tests'):
shutil.rmtree(baseDir + '/.tests')
os.mkdir(baseDir + '/.tests')
2019-10-15 09:31:10 +00:00
2020-04-05 13:25:47 +00:00
ocapAlways = False
2019-10-15 09:31:10 +00:00
# create the servers
2020-04-05 13:25:47 +00:00
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)
2020-02-19 12:27:21 +00:00
global thrAlice
if thrAlice:
while thrAlice.isAlive():
thrAlice.stop()
time.sleep(1)
thrAlice.kill()
2020-04-05 13:25:47 +00:00
thrAlice = \
threadWithTrace(target=createServerAlice,
args=(aliceDir, aliceDomain, alicePort, bobAddress,
federationList, False, False,
ocapAlways, aliceSendThreads),
daemon=True)
2019-10-15 09:31:10 +00:00
2020-02-19 12:27:21 +00:00
global thrBob
if thrBob:
while thrBob.isAlive():
thrBob.stop()
time.sleep(1)
thrBob.kill()
2020-04-05 13:25:47 +00:00
thrBob = \
threadWithTrace(target=createServerBob,
args=(bobDir, bobDomain, bobPort, aliceAddress,
federationList, False, False,
ocapAlways, bobSendThreads),
daemon=True)
2019-10-15 09:31:10 +00:00
thrAlice.start()
thrBob.start()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is True
assert thrBob.isAlive() is True
2019-10-15 09:31:10 +00:00
# wait for all servers to be running
2020-04-05 13:25:47 +00:00
ctr = 0
2019-10-15 09:31:10 +00:00
while not (testServerAliceRunning and testServerBobRunning):
time.sleep(1)
2020-04-05 13:25:47 +00:00
ctr += 1
if ctr > 60:
2019-10-15 09:31:10 +00:00
break
2020-04-05 13:25:47 +00:00
print('Alice online: ' + str(testServerAliceRunning))
print('Bob online: ' + str(testServerBobRunning))
assert ctr <= 60
2019-10-15 09:31:10 +00:00
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)
2020-06-09 11:03:59 +00:00
sessionAlice = createSession(proxyType)
2020-04-05 13:25:47 +00:00
inReplyTo = None
inReplyToAtomUri = None
subject = None
alicePostLog = []
followersOnly = False
saveToFile = True
clientToServer = False
ccUrl = None
alicePersonCache = {}
aliceCachedWebfingers = {}
alicePostLog = []
sendResult = \
sendFollowRequest(sessionAlice, aliceDir,
'alice', aliceDomain, alicePort, httpPrefix,
'bob', bobDomain, bobPort, httpPrefix,
clientToServer, federationList,
aliceSendThreads, alicePostLog,
aliceCachedWebfingers, alicePersonCache,
True, __version__)
print('sendResult: ' + str(sendResult))
2019-10-15 09:31:10 +00:00
for t in range(10):
2020-04-05 13:25:47 +00:00
if os.path.isfile(bobDir + '/accounts/bob@' +
bobDomain + '/followers.txt'):
if os.path.isfile(aliceDir + '/accounts/alice@' +
aliceDomain + '/following.txt'):
2019-10-15 09:31:10 +00:00
break
time.sleep(1)
2020-04-05 13:25:47 +00:00
assert validInbox(bobDir, 'bob', bobDomain)
assert validInboxFilenames(bobDir, 'bob', bobDomain,
aliceDomain, alicePort)
2019-10-15 09:31:10 +00:00
print('\n\n*********************************************************')
print('Alice sends a message to Bob')
2020-04-05 13:25:47 +00:00
alicePostLog = []
alicePersonCache = {}
aliceCachedWebfingers = {}
alicePostLog = []
useBlurhash = False
isArticle = False
sendResult = \
sendPost(__version__,
sessionAlice, aliceDir, 'alice', aliceDomain, alicePort,
'bob', bobDomain, bobPort, ccUrl,
httpPrefix, 'Alice message', followersOnly, saveToFile,
clientToServer, None, None, None, useBlurhash, federationList,
aliceSendThreads, alicePostLog, aliceCachedWebfingers,
alicePersonCache, isArticle, inReplyTo,
inReplyToAtomUri, subject)
print('sendResult: ' + str(sendResult))
queuePath = bobDir + '/accounts/bob@' + bobDomain + '/queue'
inboxPath = bobDir + '/accounts/bob@' + bobDomain + '/inbox'
aliceMessageArrived = False
2019-10-15 09:31:10 +00:00
for i in range(20):
time.sleep(1)
if os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
if len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))]) > 0:
aliceMessageArrived = True
2019-10-15 09:31:10 +00:00
print('Alice message sent to Bob!')
break
2020-04-05 13:25:47 +00:00
assert aliceMessageArrived is True
2019-10-15 09:31:10 +00:00
print('Message from Alice to Bob succeeded')
# stop the servers
thrAlice.kill()
thrAlice.join()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is False
2019-10-15 09:31:10 +00:00
thrBob.kill()
thrBob.join()
2020-04-05 13:25:47 +00:00
assert thrBob.isAlive() is False
2020-03-22 21:16:02 +00:00
2020-04-05 13:25:47 +00:00
assert 'alice@' + aliceDomain in open(bobDir + '/accounts/bob@' +
bobDomain + '/followers.txt').read()
assert 'bob@' + bobDomain in open(aliceDir + '/accounts/alice@' +
aliceDomain + '/following.txt').read()
2019-07-09 08:52:53 +00:00
# queue item removed
2019-11-09 21:39:04 +00:00
time.sleep(4)
2020-04-05 13:25:47 +00:00
assert len([name for name in os.listdir(queuePath)
if os.path.isfile(os.path.join(queuePath, name))]) == 0
2020-03-22 21:16:02 +00:00
2019-07-06 13:49:25 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
shutil.rmtree(baseDir + '/.tests')
2019-07-06 13:49:25 +00:00
def testFollowersOfPerson():
print('testFollowersOfPerson')
2020-04-05 13:25:47 +00:00
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)
2020-03-22 21:16:02 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
createPerson(baseDir, nickname, domain, port,
httpPrefix, True, password)
createPerson(baseDir, 'maxboardroom', domain, port,
httpPrefix, True, password)
createPerson(baseDir, 'ultrapancake', domain, port,
httpPrefix, True, password)
createPerson(baseDir, 'drokk', domain, port,
httpPrefix, True, password)
createPerson(baseDir, 'sausagedog', domain, port,
httpPrefix, True, password)
clearFollows(baseDir, nickname, domain)
followPerson(baseDir, nickname, domain, 'maxboardroom', domain,
federationList, False)
followPerson(baseDir, 'drokk', domain, 'ultrapancake', domain,
federationList, False)
# deliberate duplication
2020-04-05 13:25:47 +00:00
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)
2020-04-05 13:25:47 +00:00
def testNoOfFollowersOnDomain():
print('testNoOfFollowersOnDomain')
2020-04-05 13:25:47 +00:00
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)
2020-03-22 21:16:02 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
createPerson(baseDir, nickname, domain, port, httpPrefix, True, password)
createPerson(baseDir, 'maxboardroom', otherdomain, port,
httpPrefix, True, password)
createPerson(baseDir, 'ultrapancake', otherdomain, port,
httpPrefix, True, password)
createPerson(baseDir, 'drokk', otherdomain, port,
httpPrefix, True, password)
createPerson(baseDir, 'sausagedog', otherdomain, port,
httpPrefix, True, 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
unfollowerOfPerson(baseDir, nickname, domain, 'sausagedog', otherdomain)
followersOnOtherDomain = \
noOfFollowersOnDomain(baseDir, nickname + '@' + domain, otherdomain)
assert followersOnOtherDomain == 2
2020-03-22 21:16:02 +00:00
os.chdir(currDir)
shutil.rmtree(baseDir)
2020-04-05 13:25:47 +00:00
def testGroupFollowers():
print('testGroupFollowers')
2020-04-05 13:25:47 +00:00
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)
2020-03-22 21:16:02 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
createPerson(baseDir, nickname, domain, port, httpPrefix, True, 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')
2020-04-05 13:25:47 +00:00
assert len(grouped['zzz.domain']) == 2
assert len(grouped['wild.domain']) == 3
assert len(grouped['clutterly.domain']) == 1
2020-03-22 21:16:02 +00:00
os.chdir(currDir)
shutil.rmtree(baseDir)
2020-03-22 21:16:02 +00:00
2019-07-03 09:24:55 +00:00
def testFollows():
2019-07-03 10:04:23 +00:00
print('testFollows')
2020-04-05 13:25:47 +00:00
currDir = os.getcwd()
nickname = 'test529'
domain = 'testdomain.com'
password = 'mypass'
port = 80
httpPrefix = 'https'
federationList = ['wild.com', 'mesh.com']
baseDir = currDir + '/.tests_testfollows'
2019-07-03 09:24:55 +00:00
if os.path.isdir(baseDir):
shutil.rmtree(baseDir)
os.mkdir(baseDir)
2020-03-22 21:16:02 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
createPerson(baseDir, nickname, domain, port, httpPrefix, True, 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)
f = open(baseDir + '/accounts/' + nickname + '@' + domain +
'/following.txt', "r")
domainFound = False
2019-07-03 09:24:55 +00:00
for followingDomain in f:
2020-05-22 11:32:38 +00:00
testDomain = followingDomain.split('@')[1]
testDomain = testDomain.replace('\n', '').replace('\r', '')
2020-04-05 13:25:47 +00:00
if testDomain == 'mesh.com':
domainFound = True
2019-07-03 09:24:55 +00:00
if testDomain not in federationList:
print(testDomain)
assert(False)
2019-07-03 09:33:28 +00:00
assert(domainFound)
2020-04-05 13:25:47 +00:00
unfollowPerson(baseDir, nickname, domain, 'batman', 'mesh.com')
2019-07-03 09:33:28 +00:00
2020-04-05 13:25:47 +00:00
domainFound = False
2019-07-03 09:33:28 +00:00
for followingDomain in f:
2020-05-22 11:32:38 +00:00
testDomain = followingDomain.split('@')[1]
testDomain = testDomain.replace('\n', '').replace('\r', '')
2020-04-05 13:25:47 +00:00
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)
f = open(baseDir + '/accounts/' + nickname + '@' + domain +
'/followers.txt', "r")
2019-07-03 09:24:55 +00:00
for followerDomain in f:
2020-05-22 11:32:38 +00:00
testDomain = followerDomain.split('@')[1]
testDomain = testDomain.replace('\n', '').replace('\r', '')
2019-07-03 09:24:55 +00:00
if testDomain not in federationList:
print(testDomain)
assert(False)
os.chdir(currDir)
shutil.rmtree(baseDir)
2019-07-03 10:04:23 +00:00
2020-04-05 13:25:47 +00:00
2019-07-03 10:04:23 +00:00
def testCreatePerson():
print('testCreatePerson')
2020-04-05 13:25:47 +00:00
currDir = os.getcwd()
nickname = 'test382'
domain = 'badgerdomain.com'
password = 'mypass'
port = 80
httpPrefix = 'https'
clientToServer = False
useBlurhash = False
baseDir = currDir + '/.tests_createperson'
2019-07-03 10:04:23 +00:00
if os.path.isdir(baseDir):
shutil.rmtree(baseDir)
os.mkdir(baseDir)
os.chdir(baseDir)
2020-03-22 21:16:02 +00:00
2020-04-05 13:25:47 +00:00
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(baseDir, nickname, domain, port,
httpPrefix, True, 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)
2020-04-05 13:25:47 +00:00
createPublicPost(baseDir, nickname, domain, port, httpPrefix,
"G'day world!", False, True, clientToServer,
None, None, useBlurhash, None, None,
'Not suitable for Vogons')
2019-07-03 10:04:23 +00:00
os.chdir(currDir)
shutil.rmtree(baseDir)
2020-04-05 13:25:47 +00:00
2019-07-18 16:21:26 +00:00
def testDelegateRoles():
print('testDelegateRoles')
2020-04-05 13:25:47 +00:00
currDir = os.getcwd()
nickname = 'test382'
nicknameDelegated = 'test383'
domain = 'badgerdomain.com'
password = 'mypass'
port = 80
httpPrefix = 'https'
baseDir = currDir + '/.tests_delegaterole'
2019-07-18 16:21:26 +00:00
if os.path.isdir(baseDir):
shutil.rmtree(baseDir)
os.mkdir(baseDir)
os.chdir(baseDir)
2020-03-22 21:16:02 +00:00
2020-04-05 13:25:47 +00:00
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(baseDir, nickname, domain, port,
httpPrefix, True, password)
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(baseDir, nicknameDelegated, domain, port,
httpPrefix, True, 'insecure')
httpPrefix = 'http'
project = 'artechoke'
role = 'delegator'
actorDelegated = \
httpPrefix + '://' + domain + '/users/' + nicknameDelegated
newRoleJson = {
2019-07-18 16:21:26 +00:00
'type': 'Delegate',
2020-04-05 13:25:47 +00:00
'actor': httpPrefix + '://' + domain + '/users/' + nickname,
2019-07-18 16:21:26 +00:00
'object': {
'type': 'Role',
2020-04-05 13:25:47 +00:00
'actor': actorDelegated,
'object': project + ';' + role,
2019-07-18 16:21:26 +00:00
'to': [],
2020-03-22 21:16:02 +00:00
'cc': []
2019-07-18 16:21:26 +00:00
},
'to': [],
'cc': []
}
2020-04-05 13:25:47 +00:00
assert outboxDelegate(baseDir, nickname, newRoleJson, False)
2019-07-18 16:21:26 +00:00
# second time delegation has already happened so should return false
2020-04-05 13:25:47 +00:00
assert outboxDelegate(baseDir, nickname, newRoleJson, False) is False
2019-07-18 16:21:26 +00:00
2020-04-05 13:25:47 +00:00
assert '"delegator"' in open(baseDir + '/accounts/' + nickname +
'@' + domain + '.json').read()
assert '"delegator"' in open(baseDir + '/accounts/' + nicknameDelegated +
'@' + domain + '.json').read()
2020-03-22 21:16:02 +00:00
2020-04-05 13:25:47 +00:00
newRoleJson = {
2019-07-18 16:21:26 +00:00
'type': 'Delegate',
2020-04-05 13:25:47 +00:00
'actor': httpPrefix + '://' + domain + '/users/' + nicknameDelegated,
2019-07-18 16:21:26 +00:00
'object': {
'type': 'Role',
2020-04-05 13:25:47 +00:00
'actor': httpPrefix + '://' + domain + '/users/' + nickname,
2019-07-18 16:21:26 +00:00
'object': 'otherproject;otherrole',
'to': [],
2020-03-22 21:16:02 +00:00
'cc': []
2019-07-18 16:21:26 +00:00
},
'to': [],
'cc': []
}
# non-delegators cannot assign roles
2020-04-05 13:25:47 +00:00
assert outboxDelegate(baseDir, nicknameDelegated,
newRoleJson, False) is False
assert '"otherrole"' not in open(baseDir + '/accounts/' +
nickname + '@' + domain + '.json').read()
2019-07-18 16:21:26 +00:00
os.chdir(currDir)
shutil.rmtree(baseDir)
2020-04-05 13:25:47 +00:00
2019-07-03 18:24:44 +00:00
def testAuthentication():
print('testAuthentication')
2020-04-05 13:25:47 +00:00
currDir = os.getcwd()
nickname = 'test8743'
password = 'SuperSecretPassword12345'
2019-07-03 18:24:44 +00:00
2020-04-05 13:25:47 +00:00
baseDir = currDir + '/.tests_authentication'
2019-07-03 18:24:44 +00:00
if os.path.isdir(baseDir):
shutil.rmtree(baseDir)
os.mkdir(baseDir)
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
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)
2019-07-03 18:24:44 +00:00
2020-04-05 13:25:47 +00:00
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
2019-07-03 18:24:44 +00:00
2020-04-05 13:25:47 +00:00
authHeader = createBasicAuthHeader(nickname, password + '1')
assert authorizeBasic(baseDir, '/users/' + nickname + '/inbox',
authHeader, False) is False
2019-07-03 18:24:44 +00:00
2020-04-05 13:25:47 +00:00
password = 'someOtherPassword'
assert storeBasicCredentials(baseDir, nickname, password)
2019-07-03 19:13:23 +00:00
2020-04-05 13:25:47 +00:00
authHeader = createBasicAuthHeader(nickname, password)
assert authorizeBasic(baseDir, '/users/' + nickname + '/inbox',
authHeader, False)
2019-07-03 19:13:23 +00:00
2019-07-03 18:24:44 +00:00
os.chdir(currDir)
shutil.rmtree(baseDir)
2019-07-16 10:19:04 +00:00
2020-04-05 13:25:47 +00:00
2019-07-16 10:19:04 +00:00
def testClientToServer():
print('Testing sending a post via c2s')
global testServerAliceRunning
global testServerBobRunning
2020-04-05 13:25:47 +00:00
testServerAliceRunning = False
testServerBobRunning = False
2019-07-16 10:19:04 +00:00
2020-04-05 13:25:47 +00:00
httpPrefix = 'http'
2020-06-09 11:03:59 +00:00
proxyType = None
2020-04-05 13:25:47 +00:00
federationList = []
2019-07-16 10:19:04 +00:00
2020-04-05 13:25:47 +00:00
baseDir = os.getcwd()
if os.path.isdir(baseDir + '/.tests'):
shutil.rmtree(baseDir + '/.tests')
os.mkdir(baseDir + '/.tests')
2019-07-16 10:19:04 +00:00
2020-04-05 13:25:47 +00:00
ocapAlways = False
2019-07-16 10:19:04 +00:00
# create the servers
2020-04-05 13:25:47 +00:00
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)
2020-02-19 12:27:21 +00:00
global thrAlice
if thrAlice:
while thrAlice.isAlive():
thrAlice.stop()
time.sleep(1)
thrAlice.kill()
2020-04-05 13:25:47 +00:00
thrAlice = \
threadWithTrace(target=createServerAlice,
args=(aliceDir, aliceDomain, alicePort, bobAddress,
federationList, False, False,
ocapAlways, aliceSendThreads),
daemon=True)
2020-03-22 21:16:02 +00:00
2020-02-19 12:27:21 +00:00
global thrBob
if thrBob:
while thrBob.isAlive():
thrBob.stop()
time.sleep(1)
thrBob.kill()
2020-04-05 13:25:47 +00:00
thrBob = \
threadWithTrace(target=createServerBob,
args=(bobDir, bobDomain, bobPort, aliceAddress,
federationList, False, False,
ocapAlways, bobSendThreads),
daemon=True)
2019-07-16 10:19:04 +00:00
thrAlice.start()
thrBob.start()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is True
assert thrBob.isAlive() is True
2019-07-16 10:19:04 +00:00
# wait for both servers to be running
2020-04-05 13:25:47 +00:00
ctr = 0
2019-07-16 10:19:04 +00:00
while not (testServerAliceRunning and testServerBobRunning):
time.sleep(1)
2020-04-05 13:25:47 +00:00
ctr += 1
if ctr > 60:
2019-07-16 10:19:04 +00:00
break
2020-04-05 13:25:47 +00:00
print('Alice online: ' + str(testServerAliceRunning))
print('Bob online: ' + str(testServerBobRunning))
2019-07-16 10:19:04 +00:00
time.sleep(1)
2020-03-22 21:16:02 +00:00
2019-07-16 10:19:04 +00:00
print('\n\n*******************************************************')
print('Alice sends to Bob via c2s')
2020-06-09 11:03:59 +00:00
sessionAlice = createSession(proxyType)
2020-04-05 13:25:47 +00:00
followersOnly = False
attachedImageFilename = baseDir+'/img/logo.png'
mediaType = getAttachmentMediaType(attachedImageFilename)
attachedImageDescription = 'Logo'
useBlurhash = False
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,
attachedImageFilename, mediaType,
attachedImageDescription, useBlurhash,
cachedWebfingers, personCache, isArticle,
True, None, None, None)
print('sendResult: ' + str(sendResult))
2019-07-16 10:19:04 +00:00
for i in range(30):
if os.path.isdir(outboxPath):
2020-04-05 13:25:47 +00:00
if len([name for name in os.listdir(outboxPath)
if os.path.isfile(os.path.join(outboxPath, name))]) == 1:
2019-07-16 10:19:04 +00:00
break
time.sleep(1)
2019-07-16 11:33:40 +00:00
2020-04-05 13:25:47 +00:00
assert len([name for name in os.listdir(outboxPath)
if os.path.isfile(os.path.join(outboxPath, name))]) == 1
2019-07-16 11:33:40 +00:00
print(">>> c2s post arrived in Alice's outbox")
2020-03-22 21:16:02 +00:00
2019-07-16 11:33:40 +00:00
for i in range(30):
if os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
if len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))]) == 1:
2019-07-16 11:33:40 +00:00
break
time.sleep(1)
2020-04-05 13:25:47 +00:00
assert len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))]) == 1
2019-07-16 11:33:40 +00:00
print(">>> s2s post arrived in Bob's inbox")
print("c2s send success")
2019-07-16 19:07:45 +00:00
print('\n\nGetting message id for the post')
2020-04-05 13:25:47 +00:00
statusNumber = 0
outboxPostFilename = None
outboxPostId = None
2019-07-16 19:07:45 +00:00
for name in os.listdir(outboxPath):
if '#statuses#' in name:
2020-04-05 13:25:47 +00:00
statusNumber = name.split('#statuses#')[1].replace('.json', '')
statusNumber = int(statusNumber.replace('#activity', ''))
outboxPostFilename = outboxPath + '/' + name
postJsonObject = loadJson(outboxPostFilename, 0)
if postJsonObject:
2020-04-05 13:25:47 +00:00
outboxPostId = postJsonObject['id'].replace('/activity', '')
2019-07-16 19:07:45 +00:00
assert outboxPostId
2020-04-05 13:25:47 +00:00
print('message id obtained: ' + outboxPostId)
assert validInbox(bobDir, 'bob', bobDomain)
assert validInboxFilenames(bobDir, 'bob', bobDomain,
aliceDomain, alicePort)
2019-07-16 21:38:06 +00:00
print('\n\nAlice follows Bob')
2020-04-05 13:25:47 +00:00
sendFollowRequestViaServer(aliceDir, sessionAlice,
'alice', password,
aliceDomain, alicePort,
'bob', bobDomain, bobPort,
httpPrefix,
cachedWebfingers, personCache,
True, __version__)
aliceFollowingFilename = \
aliceDir + '/accounts/alice@' + aliceDomain + '/following.txt'
bobFollowersFilename = \
bobDir + '/accounts/bob@' + bobDomain + '/followers.txt'
2020-03-02 21:28:22 +00:00
for t in range(10):
if os.path.isfile(bobFollowersFilename):
2020-04-05 13:25:47 +00:00
if 'alice@' + aliceDomain + ':' + str(alicePort) in \
open(bobFollowersFilename).read():
2020-03-02 21:28:22 +00:00
if os.path.isfile(aliceFollowingFilename):
2020-04-05 13:25:47 +00:00
if 'bob@' + bobDomain + ':' + str(bobPort) in \
open(aliceFollowingFilename).read():
2019-07-17 11:24:11 +00:00
break
2019-07-16 21:38:06 +00:00
time.sleep(1)
2020-03-02 21:28:22 +00:00
assert os.path.isfile(bobFollowersFilename)
assert os.path.isfile(aliceFollowingFilename)
2020-04-05 13:25:47 +00:00
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)
2019-07-17 17:16:48 +00:00
print('\n\nBob follows Alice')
2020-04-05 13:25:47 +00:00
sendFollowRequestViaServer(aliceDir, sessionAlice,
'bob', 'bobpass',
bobDomain, bobPort,
'alice', aliceDomain, alicePort,
httpPrefix,
cachedWebfingers, personCache,
True, __version__)
2019-07-17 17:16:48 +00:00
for t in range(10):
2020-04-05 13:25:47 +00:00
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'):
2020-07-03 21:56:38 +00:00
aliceHandleStr = \
'alice@' + aliceDomain + ':' + str(alicePort)
if aliceHandleStr in \
2020-04-05 13:25:47 +00:00
open(bobDir + '/accounts/bob@' + bobDomain +
'/following.txt').read():
2020-07-03 21:56:38 +00:00
if os.path.isfile(bobDir + '/accounts/bob@' +
bobDomain +
'/followingCalendar.txt'):
if aliceHandleStr in \
open(bobDir + '/accounts/bob@' + bobDomain +
'/followingCalendar.txt').read():
break
2019-07-17 17:16:48 +00:00
time.sleep(1)
2020-04-05 13:25:47 +00:00
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')
2020-06-09 11:03:59 +00:00
sessionBob = createSession(proxyType)
2020-04-05 13:25:47 +00:00
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):
2020-03-22 21:16:02 +00:00
if os.path.isdir(outboxPath) and os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
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)
2020-04-05 13:25:47 +00:00
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')
2020-03-22 21:16:02 +00:00
print('\n\nBob repeats the post')
2020-04-05 13:25:47 +00:00
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__)
2019-07-16 22:57:45 +00:00
for i in range(20):
2020-03-22 21:16:02 +00:00
if os.path.isdir(outboxPath) and os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
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:
2019-07-16 22:57:45 +00:00
break
2019-07-16 19:07:45 +00:00
time.sleep(1)
2020-04-05 13:25:47 +00:00
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
2019-07-16 19:07:45 +00:00
print('Post repeated')
2019-07-17 11:54:13 +00:00
2020-04-05 13:25:47 +00:00
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__)
2019-07-17 17:16:48 +00:00
for i in range(30):
if os.path.isdir(inboxPath):
2020-04-05 13:25:47 +00:00
test = len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))])
if test == postsBefore-1:
2019-07-17 17:16:48 +00:00
break
time.sleep(1)
2020-04-05 13:25:47 +00:00
test = len([name for name in os.listdir(inboxPath)
if os.path.isfile(os.path.join(inboxPath, name))])
assert test == postsBefore - 1
2019-07-17 17:16:48 +00:00
print(">>> post deleted from Alice's outbox and Bob's inbox")
2020-04-05 13:25:47 +00:00
assert validInbox(bobDir, 'bob', bobDomain)
assert validInboxFilenames(bobDir, 'bob', bobDomain,
aliceDomain, alicePort)
2020-03-22 21:16:02 +00:00
2019-07-17 11:54:13 +00:00
print('\n\nAlice unfollows Bob')
2020-04-05 13:25:47 +00:00
password = 'alicepass'
sendUnfollowRequestViaServer(baseDir, sessionAlice,
'alice', password,
aliceDomain, alicePort,
'bob', bobDomain, bobPort,
httpPrefix,
cachedWebfingers, personCache,
True, __version__)
2019-07-17 11:54:13 +00:00
for t in range(10):
2020-04-05 13:25:47 +00:00
if 'alice@' + aliceDomain + ':' + str(alicePort) not in \
open(bobFollowersFilename).read():
if 'bob@' + bobDomain + ':' + str(bobPort) not in \
open(aliceFollowingFilename).read():
2019-07-17 11:54:13 +00:00
break
time.sleep(1)
2020-03-02 21:28:22 +00:00
assert os.path.isfile(bobFollowersFilename)
assert os.path.isfile(aliceFollowingFilename)
2020-04-05 13:25:47 +00:00
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)
2019-07-17 11:54:13 +00:00
2019-07-16 11:33:40 +00:00
# stop the servers
thrAlice.kill()
thrAlice.join()
2020-04-05 13:25:47 +00:00
assert thrAlice.isAlive() is False
2019-07-16 11:33:40 +00:00
thrBob.kill()
thrBob.join()
2020-04-05 13:25:47 +00:00
assert thrBob.isAlive() is False
2019-07-16 11:33:40 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
# shutil.rmtree(aliceDir)
# shutil.rmtree(bobDir)
2019-07-16 10:19:04 +00:00
2019-08-21 16:35:46 +00:00
def testActorParsing():
print('testActorParsing')
2020-04-05 13:25:47 +00:00
actor = 'https://mydomain:72/users/mynick'
domain, port = getDomainFromActor(actor)
assert domain == 'mydomain'
assert port == 72
nickname = getNicknameFromActor(actor)
assert nickname == 'mynick'
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'
2019-08-21 16:35:46 +00:00
2019-09-01 08:55:05 +00:00
def testWebLinks():
2019-09-01 08:57:51 +00:00
print('testWebLinks')
2020-01-24 10:52:59 +00:00
2020-04-05 13:25:47 +00:00
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"' + \
' 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)
2019-10-01 10:36:51 +00:00
assert 'ellipsis' in linkedText
2020-01-24 10:52:59 +00:00
2020-04-05 13:25:47 +00:00
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 = \
'<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'
resultText = removeLongWords(addWebLinks(exampleText), 40, [])
assert resultText == exampleText
2020-03-29 10:48:31 +00:00
assert 'ellipsis' not in resultText
2020-03-29 09:59:54 +00:00
2020-05-12 09:34:58 +00:00
exampleText = \
'<p>' + \
'<<\\p>'
2020-05-12 09:34:58 +00:00
resultText = replaceContentDuplicates(exampleText)
assert resultText == \
'<p>' + \
''
2020-01-24 10:52:59 +00:00
2020-02-21 15:09:31 +00:00
def testAddEmoji():
print('testAddEmoji')
2020-04-05 13:25:47 +00:00
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'
2019-09-29 18:48:34 +00:00
if not os.path.isdir(path):
os.mkdir(path)
2020-04-05 13:25:47 +00:00
path = baseDir + '/.tests/emoji'
2019-09-29 18:48:34 +00:00
if os.path.isdir(path):
shutil.rmtree(path)
2020-03-22 21:16:02 +00:00
os.mkdir(path)
2020-04-05 13:25:47 +00:00
baseDir = path
path = baseDir + '/emoji'
2019-09-29 18:48:34 +00:00
if os.path.isdir(path):
shutil.rmtree(path)
2020-03-22 21:16:02 +00:00
os.mkdir(path)
2020-04-05 13:25:47 +00:00
copytree(baseDirOriginal + '/emoji', baseDir + '/emoji')
2019-09-29 18:48:34 +00:00
os.chdir(baseDir)
2020-04-05 13:25:47 +00:00
privateKeyPem, publicKeyPem, person, wfEndpoint = \
createPerson(baseDir, nickname, domain, port,
httpPrefix, True, 'password')
contentModified = \
addHtmlTags(baseDir, httpPrefix,
nickname, domain, content,
recipients, hashtags, True)
2020-02-21 15:09:31 +00:00
assert ':lemon:' in contentModified
2020-02-21 09:53:36 +00:00
assert contentModified.startswith('<p>')
assert contentModified.endswith('</p>')
2020-04-05 13:25:47 +00:00
tags = []
for tagName, tag in hashtags.items():
2019-09-29 18:48:34 +00:00
tags.append(tag)
2020-04-05 13:25:47 +00:00
content = contentModified
contentModified = replaceEmojiFromTags(content, tags, 'content')
# print('contentModified: '+contentModified)
assert contentModified == '<p>Emoji 🍋 🍓 🍌</p>'
2019-09-29 18:48:34 +00:00
os.chdir(baseDirOriginal)
2020-04-05 13:25:47 +00:00
shutil.rmtree(baseDirOriginal + '/.tests')
2019-09-29 18:48:34 +00:00
2019-10-12 12:45:53 +00:00
def testGetStatusNumber():
print('testGetStatusNumber')
2020-04-05 13:25:47 +00:00
prevStatusNumber = None
for i in range(1, 20):
statusNumber, published = getStatusNumber()
2019-10-12 12:45:53 +00:00
if prevStatusNumber:
assert len(statusNumber) == 18
assert int(statusNumber) > prevStatusNumber
2020-04-05 13:25:47 +00:00
prevStatusNumber = int(statusNumber)
2019-10-12 12:45:53 +00:00
2019-11-09 21:39:04 +00:00
def testCommentJson() -> None:
2019-11-09 12:13:39 +00:00
print('testCommentJson')
2020-04-05 13:25:47 +00:00
filename = '/tmp/test.json'
messageStr = "Crème brûlée यह एक परीक्षण ह"
testJson = {
2019-11-09 12:13:39 +00:00
"content": messageStr
}
2020-04-05 13:25:47 +00:00
assert saveJson(testJson, filename)
receivedJson = loadJson(filename, 0)
2019-11-09 12:13:39 +00:00
assert receivedJson
2020-04-05 13:25:47 +00:00
assert receivedJson['content'] == messageStr
encodedStr = json.dumps(testJson, ensure_ascii=False)
assert messageStr in encodedStr
2019-11-09 12:13:39 +00:00
2020-04-05 13:25:47 +00:00
2019-11-23 10:13:57 +00:00
def testSaveLoadJson():
print('testSaveLoadJson')
2020-04-05 13:25:47 +00:00
testJson = {
2019-11-23 10:13:57 +00:00
"param1": 3,
2019-11-23 10:20:30 +00:00
"param2": '"Crème brûlée यह एक परीक्षण ह"'
2019-11-23 10:13:57 +00:00
}
2020-04-05 13:25:47 +00:00
testFilename = '/tmp/.epicyonTestSaveLoadJson.json'
2019-11-23 10:13:57 +00:00
if os.path.isfile(testFilename):
os.remove(testFilename)
2020-04-05 13:25:47 +00:00
assert saveJson(testJson, testFilename)
2019-11-23 10:13:57 +00:00
assert os.path.isfile(testFilename)
2020-04-05 13:25:47 +00:00
testLoadJson = loadJson(testFilename)
2019-11-23 10:13:57 +00:00
assert(testLoadJson)
assert testLoadJson.get('param1')
assert testLoadJson.get('param2')
2020-04-05 13:25:47 +00:00
assert testLoadJson['param1'] == 3
assert testLoadJson['param2'] == '"Crème brûlée यह एक परीक्षण ह"'
2019-11-23 10:13:57 +00:00
os.remove(testFilename)
2019-11-23 13:04:11 +00:00
2020-04-05 13:25:47 +00:00
2019-11-23 13:04:11 +00:00
def testTheme():
print('testTheme')
2020-04-05 13:25:47 +00:00
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;'
2019-11-23 13:04:11 +00:00
def testRecentPostsCache():
print('testRecentPostsCache')
2020-04-05 13:25:47 +00:00
recentPostsCache = {}
maxRecentPosts = 3
htmlStr = '<html></html>'
for i in range(5):
2020-04-05 13:25:47 +00:00
postJsonObject = {
"id": "https://somesite.whatever/users/someuser/statuses/"+str(i)
}
2020-04-05 13:25:47 +00:00
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>')
2020-06-15 12:37:53 +00:00
def testJsonld():
print("testJsonld")
jldDocument = {
"description": "My json document",
"numberField": 83582,
"object": {
"content": "Some 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 = testSignJsonld(jldDocument, privateKeyPem)
assert(signedDocument)
2020-06-15 13:35:33 +00:00
assert(signedDocument.get('signature'))
assert(signedDocument['signature'].get('signatureValue'))
assert(signedDocument['signature'].get('type'))
assert(len(signedDocument['signature']['signatureValue']) > 50)
assert(signedDocument['signature']['type'] == 'RsaSignatureSuite2017')
2020-06-15 12:37:53 +00:00
assert(jsonldVerify(signedDocument, publicKeyPem))
2020-06-22 16:55:19 +00:00
def testSiteIsActive():
print('testSiteIsActive')
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.')
2020-07-10 14:15:01 +00:00
def testDangerousMarkup():
print('testDangerousMarkup')
content = '<p>This is a valid message</p>'
assert(not dangerousMarkup(content))
content = 'This is a valid message without markup'
assert(not dangerousMarkup(content))
content = '<p>This is a valid-looking message. But wait... ' + \
'<script>document.getElementById("concentrated")' + \
'.innerHTML = "evil";</script></p>'
assert(dangerousMarkup(content))
content = '<p>This is a valid-looking message. But wait... ' + \
'<script src="https://evilsite/payload.js" /></p>'
assert(dangerousMarkup(content))
content = '<p>This message embeds an evil frame.' + \
'<iframe src="somesite"></iframe></p>'
assert(dangerousMarkup(content))
content = '<p>This message tries to obfuscate an evil frame.' + \
'< iframe src = "somesite"></ iframe ></p>'
assert(dangerousMarkup(content))
content = '<p>This message is not necessarily evil, but annoying.' + \
'<hr><br><br><br><br><br><br><br><hr><hr></p>'
assert(dangerousMarkup(content))
content = '<p>This message contans a ' + \
'<a href="https://validsite/index.html">valid link.</a></p>'
assert(not dangerousMarkup(content))
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))
2019-06-30 21:20:02 +00:00
def runAllTests():
print('Running tests...')
2020-07-10 14:15:01 +00:00
testDangerousMarkup()
testRemoveHtml()
2020-06-22 16:55:19 +00:00
testSiteIsActive()
2020-06-15 12:37:53 +00:00
testJsonld()
testRemoveTextFormatting()
2020-01-24 11:27:12 +00:00
testWebLinks()
testRecentPostsCache()
2019-11-23 13:04:11 +00:00
testTheme()
2019-11-23 10:13:57 +00:00
testSaveLoadJson()
2019-11-09 12:13:39 +00:00
testCommentJson()
2019-10-12 12:45:53 +00:00
testGetStatusNumber()
2020-02-21 15:09:31 +00:00
testAddEmoji()
2019-08-21 16:35:46 +00:00
testActorParsing()
2019-06-30 21:20:02 +00:00
testHttpsig()
testCache()
testThreads()
2019-07-03 10:04:23 +00:00
testCreatePerson()
2019-07-03 18:24:44 +00:00
testAuthentication()
testFollowersOfPerson()
testNoOfFollowersOnDomain()
testFollows()
testGroupFollowers()
2019-07-18 16:21:26 +00:00
testDelegateRoles()
2020-03-22 21:16:02 +00:00
print('Tests succeeded\n')