def _test_server_admin(self, host, port):
# set up init connection
service_identifier = HC.ClientServiceIdentifier(
os.urandom(32), HC.SERVER_ADMIN, 'server admin')
credentials = CC.Credentials(host, port)
connection = CC.ConnectionToService(service_identifier, credentials)
# init
access_key = os.urandom(32)
HC.app.SetRead('init', access_key)
response = connection.Get('init')
self.assertEqual(response['access_key'], access_key)
# set up connection
credentials = CC.Credentials(host, port, self._access_key)
connection = CC.ConnectionToService(service_identifier, credentials)
# backup
response = connection.Post('backup')
# services
services = {'message': 'hello'}
HC.app.SetRead('services', services)
response = connection.Get('services')
self.assertEqual(response['services_info'], services)
edit_log = 'blah'
connection.Post('services', edit_log=edit_log)
written = HC.app.GetWrite('services')
[(args, kwargs)] = written
(written_service_identifier, written_edit_log) = args
self.assertEqual(edit_log, written_edit_log)
def _test_repo(self, host, port, service_type):
# set up connection
service_identifier = HC.ClientServiceIdentifier(
os.urandom(32), service_type, 'service')
credentials = CC.Credentials(host, port, self._access_key)
connection = CC.ConnectionToService(service_identifier, credentials)
# news
news = 'this is the news'
connection.Post('news', news=news)
written = HC.app.GetWrite('news')
[(args, kwargs)] = written
(written_service_identifier, written_news) = args
self.assertEqual(news, written_news)
# num_petitions
num_petitions = 23
HC.app.SetRead('num_petitions', num_petitions)
response = connection.Get('num_petitions')
self.assertEqual(response['num_petitions'], num_petitions)
# petition
petition = 'petition'
HC.app.SetRead('petition', petition)
response = connection.Get('petition')
self.assertEqual(response['petition'], petition)
# update
update = 'update'
update_key = os.urandom(32)
path = SC.GetExpectedPath('update', update_key)
with open(path, 'wb') as f:
f.write(update)
HC.app.SetRead('update_key', update_key)
response = connection.Get('update', begin=100)
self.assertEqual(response, update)
try:
os.remove(path)
except:
pass
connection.Post('update', update=update)
written = HC.app.GetWrite('update')
[(args, kwargs)] = written
(written_service_identifier, written_account, written_update) = args
self.assertEqual(update, written_update)
def _test_restricted(self, host, port, service_type):
# access_key
registration_key = os.urandom(32)
HC.app.SetRead('access_key', self._access_key)
connection = HC.get_connection(host=host, port=port)
headers = {}
headers['Hydrus-Key'] = registration_key.encode('hex')
response = connection.request('GET', '/access_key', headers=headers)
self.assertEqual(response['access_key'], self._access_key)
# set up connection
service_identifier = HC.ClientServiceIdentifier(
os.urandom(32), service_type, 'service')
credentials = CC.Credentials(host, port, self._access_key)
connection = CC.ConnectionToService(service_identifier, credentials)
# set up session
last_error = 0
account = self._account
service_for_session_manager = CC.ServiceRemoteRestricted(
service_identifier, credentials, last_error, account)
HC.app.SetRead('service', service_for_session_manager)
HC.app.SetRead('account', self._account)
# account
response = connection.Get('account')
self.assertEqual(repr(response['account']), repr(self._account))
# account_info
account_info = {'message': 'hello'}
HC.app.SetRead('account_info', account_info)
response = connection.Get('account_info', subject_account_id=1)
self.assertEqual(response['account_info'], account_info)
response = connection.Get(
'account_info', subject_access_key=os.urandom(32).encode('hex'))
self.assertEqual(response['account_info'], account_info)
response = connection.Get('account_info',
subject_hash=os.urandom(32).encode('hex'))
self.assertEqual(response['account_info'], account_info)
response = connection.Get('account_info',
subject_hash=os.urandom(32).encode('hex'),
subject_tag='hello'.encode('hex'))
self.assertEqual(response['account_info'], account_info)
# account_types
account_types = {'message': 'hello'}
HC.app.SetRead('account_types', account_types)
response = connection.Get('account_types')
self.assertEqual(response['account_types'], account_types)
edit_log = 'blah'
connection.Post('account_types', edit_log=edit_log)
written = HC.app.GetWrite('account_types')
[(args, kwargs)] = written
(written_service_identifier, written_edit_log) = args
self.assertEqual(edit_log, written_edit_log)
# registration_keys
registration_key = os.urandom(32)
HC.app.SetRead('registration_keys', [registration_key])
response = connection.Get('registration_keys', num=1, title='blah')
self.assertEqual(response['registration_keys'], [registration_key])
response = connection.Get('registration_keys',
num=1,
title='blah',
expiration=100)
self.assertEqual(response['registration_keys'], [registration_key])
# stats
stats = {'message': 'hello'}
HC.app.SetRead('stats', stats)
response = connection.Get('stats')
self.assertEqual(response['stats'], stats)
def _test_file_repo(self, host, port):
# set up connection
service_identifier = HC.ClientServiceIdentifier(
os.urandom(32), HC.FILE_REPOSITORY, 'service')
credentials = CC.Credentials(host, port, self._access_key)
connection = CC.ConnectionToService(service_identifier, credentials)
file_connection = httplib.HTTPConnection(host, port, timeout=10)
# file
path = SC.GetExpectedPath('file', self._file_hash)
with open(path, 'wb') as f:
f.write('file')
response = connection.Get('file', hash=self._file_hash.encode('hex'))
self.assertEqual(response, 'file')
try:
os.remove(path)
except:
pass
path = HC.STATIC_DIR + os.path.sep + 'hydrus.png'
with open(path, 'rb') as f:
file = f.read()
connection.Post('file', file=file)
written = HC.app.GetWrite('file')
[(args, kwargs)] = written
(written_service_identifier, written_account, written_file_dict) = args
self.assertEqual(
written_file_dict['hash'],
'\xadm5\x99\xa6\xc4\x89\xa5u\xeb\x19\xc0&\xfa\xce\x97\xa9\xcdey\xe7G(\xb0\xce\x94\xa6\x01\xd22\xf3\xc3'
)
self.assertEqual(written_file_dict['ip'], '127.0.0.1')
self.assertEqual(written_file_dict['height'], 200)
self.assertEqual(written_file_dict['width'], 200)
self.assertEqual(written_file_dict['mime'], 2)
self.assertEqual(written_file_dict['size'], 5270)
# ip
(ip, timestamp) = ('94.45.87.123', HC.GetNow() - 100000)
HC.app.SetRead('ip', (ip, timestamp))
response = connection.Get('ip', hash=self._file_hash.encode('hex'))
self.assertEqual(response['ip'], ip)
self.assertEqual(response['timestamp'], timestamp)
# thumbnail
path = SC.GetExpectedPath('thumbnail', self._file_hash)
with open(path, 'wb') as f:
f.write('thumb')
response = connection.Get('thumbnail',
hash=self._file_hash.encode('hex'))
self.assertEqual(response, 'thumb')
try:
os.remove(path)
except:
pass