159 lines
7.5 KiB
Python
159 lines
7.5 KiB
Python
import ClientConstants as CC
|
|
import collections
|
|
import HydrusConstants as HC
|
|
import os
|
|
import TestConstants
|
|
import unittest
|
|
|
|
class TestFunctions( unittest.TestCase ):
|
|
|
|
def test_multipart( self ):
|
|
|
|
hash = '5a1ba880a043e6207dca5f5407089fb0a0b0a588c8a6cd6a2807d173f59223d9'.decode( 'hex' )
|
|
|
|
fields = []
|
|
|
|
fields.append( ( 'sub', CC.FIELD_TEXT, 'subject' ) )
|
|
fields.append( ( 'com', CC.FIELD_COMMENT, u'\u269COP is a fag' ) )
|
|
fields.append( ( 'pass', CC.FIELD_PASSWORD, '123456' ) )
|
|
fields.append( ( 'recaptcha', CC.FIELD_VERIFICATION_RECAPTCHA, 'reticulating splines' ) )
|
|
fields.append( ( 'resto', CC.FIELD_THREAD_ID, '1000000' ) )
|
|
fields.append( ( 'spoiler/on', CC.FIELD_CHECKBOX, True ) )
|
|
fields.append( ( 'upfile', CC.FIELD_FILE, ( hash, HC.IMAGE_GIF, TestConstants.tinest_gif ) ) )
|
|
|
|
result = CC.GenerateDumpMultipartFormDataCTAndBody( fields )
|
|
|
|
expected_result = ('multipart/form-data; boundary=----------AaB03x', '------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="sub"\r\n\r\nsubject\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="com"\r\n\r\n\xe2\x9a\x9cOP is a fag\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="pass"\r\n\r\n123456\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="recaptcha"\r\n\r\nreticulating splines\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="resto"\r\n\r\n1000000\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="spoiler"\r\n\r\non\r\n------------AaB03x\r\nContent-Type: image/gif\r\nContent-Disposition: form-data; name="upfile"; filename="5a1ba880a043e6207dca5f5407089fb0a0b0a588c8a6cd6a2807d173f59223d9.gif"\r\n\r\nGIF89a\x01\x00\x01\x00\x00\xff\x00,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x00;\r\n------------AaB03x--\r\n')
|
|
|
|
self.assertEqual( result, expected_result )
|
|
|
|
#
|
|
|
|
fields = {}
|
|
|
|
fields[ 'act' ] = 'do_login'
|
|
fields[ 'id' ] = 'token'
|
|
fields[ 'pin' ] = 'pin'
|
|
fields[ 'long_login' ] = 'yes'
|
|
fields[ 'random_unicode' ] = u'\u269C'
|
|
|
|
result = CC.GenerateMultipartFormDataCTAndBodyFromDict( fields )
|
|
|
|
expected_result = ('multipart/form-data; boundary=----------AaB03x', '------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="long_login"\r\n\r\nyes\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="random_unicode"\r\n\r\n\xe2\x9a\x9c\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="id"\r\n\r\ntoken\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="pin"\r\n\r\npin\r\n------------AaB03x\r\nContent-Type: application/octet-stream\r\nContent-Disposition: form-data; name="act"\r\n\r\ndo_login\r\n------------AaB03x--\r\n')
|
|
|
|
self.assertEqual( result, expected_result )
|
|
|
|
|
|
class TestManagers( unittest.TestCase ):
|
|
|
|
def test_services( self ):
|
|
|
|
def test_service( service, key, service_type, name, info ):
|
|
|
|
self.assertEqual( service.GetServiceKey(), key )
|
|
self.assertEqual( service.GetServiceType(), service_type )
|
|
self.assertEqual( service.GetName(), name )
|
|
self.assertEqual( service.GetInfo(), info )
|
|
|
|
|
|
repo_key = os.urandom( 32 )
|
|
repo_type = HC.TAG_REPOSITORY
|
|
repo_name = 'test tag repo'
|
|
repo_info = { 'blah' : 5 }
|
|
|
|
repo = CC.Service( repo_key, repo_type, repo_name, repo_info )
|
|
|
|
other_key = os.urandom( 32 )
|
|
|
|
other = CC.Service( other_key, HC.LOCAL_BOORU, 'booru', {} )
|
|
|
|
services = []
|
|
|
|
services.append( repo )
|
|
services.append( other )
|
|
|
|
HC.app.SetRead( 'services', services )
|
|
|
|
services_manager = CC.ServicesManager()
|
|
|
|
#
|
|
|
|
service = services_manager.GetService( repo_key )
|
|
|
|
test_service( service, repo_key, repo_type, repo_name, repo_info )
|
|
|
|
service = services_manager.GetService( other_key )
|
|
|
|
#
|
|
|
|
services = services_manager.GetServices( ( HC.TAG_REPOSITORY, ) )
|
|
|
|
self.assertEqual( len( services ), 1 )
|
|
|
|
self.assertEqual( services[0].GetServiceKey(), repo_key )
|
|
|
|
#
|
|
|
|
services = []
|
|
|
|
services.append( repo )
|
|
|
|
HC.app.SetRead( 'services', services )
|
|
|
|
services_manager.RefreshServices()
|
|
|
|
self.assertRaises( Exception, services_manager.GetService, other_key )
|
|
|
|
|
|
def test_undo( self ):
|
|
|
|
hash_1 = os.urandom( 32 )
|
|
hash_2 = os.urandom( 32 )
|
|
hash_3 = os.urandom( 32 )
|
|
|
|
command_1 = { HC.LOCAL_FILE_SERVICE_KEY : [ HC.ContentUpdate( HC.CONTENT_DATA_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, { hash_1 } ) ] }
|
|
command_2 = { HC.LOCAL_FILE_SERVICE_KEY : [ HC.ContentUpdate( HC.CONTENT_DATA_TYPE_FILES, HC.CONTENT_UPDATE_INBOX, { hash_2 } ) ] }
|
|
command_3 = { HC.LOCAL_FILE_SERVICE_KEY : [ HC.ContentUpdate( HC.CONTENT_DATA_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, { hash_1, hash_3 } ) ] }
|
|
|
|
command_1_inverted = { HC.LOCAL_FILE_SERVICE_KEY : [ HC.ContentUpdate( HC.CONTENT_DATA_TYPE_FILES, HC.CONTENT_UPDATE_INBOX, { hash_1 } ) ] }
|
|
command_2_inverted = { HC.LOCAL_FILE_SERVICE_KEY : [ HC.ContentUpdate( HC.CONTENT_DATA_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, { hash_2 } ) ] }
|
|
|
|
undo_manager = CC.UndoManager()
|
|
|
|
#
|
|
|
|
undo_manager.AddCommand( 'content_updates', command_1 )
|
|
|
|
self.assertEqual( ( u'undo local files->archive 1 files', None ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
undo_manager.AddCommand( 'content_updates', command_2 )
|
|
|
|
self.assertEqual( ( u'undo local files->inbox 1 files', None ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
undo_manager.Undo()
|
|
|
|
self.assertEqual( ( u'undo local files->archive 1 files', u'redo local files->inbox 1 files' ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
self.assertEqual( HC.app.GetWrite( 'content_updates' ), [ ( ( command_2_inverted, ), {} ) ] )
|
|
|
|
undo_manager.Redo()
|
|
|
|
self.assertEqual( HC.app.GetWrite( 'content_updates' ), [ ( ( command_2, ), {} ) ] )
|
|
|
|
self.assertEqual( ( u'undo local files->inbox 1 files', None ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
undo_manager.Undo()
|
|
|
|
self.assertEqual( HC.app.GetWrite( 'content_updates' ), [ ( ( command_2_inverted, ), {} ) ] )
|
|
|
|
undo_manager.Undo()
|
|
|
|
self.assertEqual( HC.app.GetWrite( 'content_updates' ), [ ( ( command_1_inverted, ), {} ) ] )
|
|
|
|
self.assertEqual( ( None, u'redo local files->archive 1 files' ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
undo_manager.AddCommand( 'content_updates', command_3 )
|
|
|
|
self.assertEqual( ( u'undo local files->archive 2 files', None ), undo_manager.GetUndoRedoStrings() )
|
|
|
|
|