438 lines
14 KiB
Python
438 lines
14 KiB
Python
import cherrypy
|
|
from model.Note import Note
|
|
from model.Notebook import Notebook
|
|
from model.User import User
|
|
from Test_controller import Test_controller
|
|
|
|
|
|
class Test_root( Test_controller ):
|
|
def setUp( self ):
|
|
Test_controller.setUp( self )
|
|
|
|
self.notebook = Notebook.create( self.database.next_id( Notebook ), u"my notebook", trash_id = u"foo" )
|
|
self.database.save( self.notebook )
|
|
|
|
self.anon_notebook = Notebook.create( self.database.next_id( Notebook ), u"Luminotes" )
|
|
self.database.save( self.anon_notebook )
|
|
self.anon_note = Note.create(
|
|
self.database.next_id( Note ), u"<h3>my note</h3>",
|
|
notebook_id = self.anon_notebook.object_id,
|
|
)
|
|
self.database.save( self.anon_note )
|
|
|
|
self.login_note = Note.create(
|
|
self.database.next_id( Note ), u"<h3>login</h3>",
|
|
notebook_id = self.anon_notebook.object_id,
|
|
)
|
|
self.database.save( self.login_note )
|
|
|
|
self.blog_notebook = Notebook.create( self.database.next_id( Notebook ), u"Luminotes blog" )
|
|
self.database.save( self.blog_notebook )
|
|
self.blog_note = Note.create(
|
|
self.database.next_id( Note ), u"<h3>my blog entry</h3>",
|
|
notebook_id = self.blog_notebook.object_id,
|
|
)
|
|
self.database.save( self.blog_note )
|
|
|
|
self.guide_notebook = Notebook.create( self.database.next_id( Notebook ), u"Luminotes user guide" )
|
|
self.database.save( self.guide_notebook )
|
|
self.guide_note = Note.create(
|
|
self.database.next_id( Note ), u"<h3>it's all self-explanatory</h3>",
|
|
notebook_id = self.guide_notebook.object_id,
|
|
)
|
|
self.database.save( self.guide_note )
|
|
|
|
self.privacy_notebook = Notebook.create( self.database.next_id( Notebook ), u"Luminotes privacy policy" )
|
|
self.database.save( self.privacy_notebook )
|
|
self.privacy_note = Note.create(
|
|
self.database.next_id( Note ), u"<h3>yay privacy</h3>",
|
|
notebook_id = self.privacy_notebook.object_id,
|
|
)
|
|
self.database.save( self.privacy_note )
|
|
|
|
self.username = u"mulder"
|
|
self.password = u"trustno1"
|
|
self.email_address = u"outthere@example.com"
|
|
self.user = None
|
|
self.session_id = None
|
|
|
|
self.user = User.create( self.database.next_id( User ), self.username, self.password, self.email_address )
|
|
self.database.save( self.user )
|
|
self.database.execute( self.user.sql_save_notebook( self.notebook.object_id ) )
|
|
|
|
self.anonymous = User.create( self.database.next_id( User ), u"anonymous" )
|
|
self.database.save( self.anonymous )
|
|
self.database.execute( self.anonymous.sql_save_notebook( self.anon_notebook.object_id ) )
|
|
self.database.execute( self.anonymous.sql_save_notebook( self.blog_notebook.object_id ) )
|
|
self.database.execute( self.anonymous.sql_save_notebook( self.guide_notebook.object_id ) )
|
|
self.database.execute( self.anonymous.sql_save_notebook( self.privacy_notebook.object_id ) )
|
|
|
|
def test_index( self ):
|
|
result = self.http_get( "/" )
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) is None
|
|
assert result[ u"user" ].username == u"anonymous"
|
|
assert len( result[ u"notebooks" ] ) == 4
|
|
assert result[ u"first_notebook" ] == None
|
|
assert result[ u"login_url" ] == u"https://luminotes.com/notebooks/%s?note_id=%s" % (
|
|
self.anon_notebook.object_id, self.login_note.object_id,
|
|
)
|
|
assert result[ u"logout_url" ] == u"https://luminotes.com/users/logout"
|
|
assert result[ u"rate_plan" ]
|
|
|
|
def test_index_after_login_without_referer( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
"/",
|
|
session_id = self.session_id,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) == u"https://luminotes.com/notebooks/%s" % self.notebook.object_id
|
|
|
|
def test_index_after_login_with_referer( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
"/",
|
|
headers = [ ( u"Referer", "http://whee" ) ],
|
|
session_id = self.session_id,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) == u"https://luminotes.com/"
|
|
|
|
def test_index_with_https_after_login_without_referer( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
"/",
|
|
session_id = self.session_id,
|
|
pretend_https = True,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) == u"https://luminotes.com/notebooks/%s" % self.notebook.object_id
|
|
|
|
def test_index_with_https_after_login_with_referer( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
"/",
|
|
session_id = self.session_id,
|
|
headers = [ ( u"Referer", "http://whee" ) ],
|
|
pretend_https = True,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) is None
|
|
assert result[ u"user" ].username == self.user.username
|
|
assert len( result[ u"notebooks" ] ) == 5
|
|
assert result[ u"first_notebook" ].object_id == self.notebook.object_id
|
|
assert result[ u"login_url" ] == None
|
|
assert result[ u"logout_url" ] == u"https://luminotes.com/users/logout"
|
|
assert result[ u"rate_plan" ]
|
|
|
|
def test_tour( self ):
|
|
result = self.http_get( u"/tour" )
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) is None
|
|
assert result[ u"user" ].username == u"anonymous"
|
|
assert len( result[ u"notebooks" ] ) == 4
|
|
assert result[ u"first_notebook" ] == None
|
|
assert result[ u"login_url" ] == u"https://luminotes.com/notebooks/%s?note_id=%s" % (
|
|
self.anon_notebook.object_id, self.login_note.object_id,
|
|
)
|
|
assert result[ u"logout_url" ] == u"https://luminotes.com/users/logout"
|
|
assert result[ u"rate_plan" ]
|
|
|
|
def test_take_a_tour( self ):
|
|
result = self.http_get( u"/take_a_tour" )
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) == u"/tour"
|
|
|
|
def test_tour_after_login( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
u"/tour",
|
|
session_id = self.session_id,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) is None
|
|
assert result[ u"user" ].username == self.user.username
|
|
assert len( result[ u"notebooks" ] ) == 5
|
|
assert result[ u"first_notebook" ].object_id == self.notebook.object_id
|
|
assert result[ u"login_url" ] == None
|
|
assert result[ u"logout_url" ] == u"https://luminotes.com/users/logout"
|
|
assert result[ u"rate_plan" ]
|
|
|
|
def test_take_a_tour_after_login( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
u"/take_a_tour",
|
|
session_id = self.session_id,
|
|
)
|
|
|
|
assert result
|
|
assert result.get( u"redirect" ) == u"/tour"
|
|
|
|
def test_default( self ):
|
|
result = self.http_get(
|
|
"/my_note",
|
|
)
|
|
|
|
assert result
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].object_id == self.anon_note.object_id
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert result[ u"user" ].object_id == self.anonymous.object_id
|
|
|
|
def test_default_with_invite_id( self ):
|
|
result = self.http_get(
|
|
"/my_note?invite_id=whee",
|
|
)
|
|
|
|
assert result
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].object_id == self.anon_note.object_id
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert result[ u"invite_id" ] == u"whee"
|
|
assert result[ u"user" ].object_id == self.anonymous.object_id
|
|
|
|
def test_default_with_after_login( self ):
|
|
after_login = "/foo/bar"
|
|
|
|
result = self.http_get(
|
|
"/my_note?after_login=%s" % after_login,
|
|
)
|
|
|
|
assert result
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].object_id == self.anon_note.object_id
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert result[ u"after_login" ] == after_login
|
|
assert result[ u"user" ].object_id == self.anonymous.object_id
|
|
|
|
def test_default_with_after_login_with_full_url( self ):
|
|
after_login = "http://example.com/foo/bar"
|
|
|
|
result = self.http_get(
|
|
"/my_note?after_login=%s" % after_login,
|
|
)
|
|
|
|
assert result
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].object_id == self.anon_note.object_id
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert result.get( u"after_login" ) is None
|
|
assert result[ u"user" ].object_id == self.anonymous.object_id
|
|
|
|
def test_default_after_login( self ):
|
|
self.login()
|
|
|
|
result = self.http_get(
|
|
"/my_note",
|
|
session_id = self.session_id,
|
|
)
|
|
|
|
assert result
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].object_id == self.anon_note.object_id
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert result[ u"user" ].object_id == self.user.object_id
|
|
|
|
def test_default_with_unknown_note( self ):
|
|
result = self.http_get(
|
|
"/unknown_note",
|
|
)
|
|
|
|
body = result.get( u"body" )
|
|
assert body
|
|
assert len( body ) > 0
|
|
assert u"404" in body[ 0 ]
|
|
|
|
def test_default_with_login_note( self ):
|
|
result = self.http_get(
|
|
"/login",
|
|
)
|
|
|
|
assert result
|
|
assert result.get( "redirect" )
|
|
assert result.get( "redirect" ).startswith( "https://" )
|
|
|
|
def test_default_with_sign_up_note( self ):
|
|
result = self.http_get(
|
|
"/sign_up",
|
|
)
|
|
|
|
assert result
|
|
assert result.get( "redirect" )
|
|
assert result.get( "redirect" ).startswith( "https://" )
|
|
|
|
def test_blog( self ):
|
|
result = self.http_get(
|
|
"/blog",
|
|
)
|
|
|
|
assert result
|
|
assert u"error" not in result
|
|
assert result[ u"notebook" ].object_id == self.blog_notebook.object_id
|
|
|
|
def test_blog_with_note_id( self ):
|
|
result = self.http_get(
|
|
"/blog?note_id=%s" % self.blog_note.object_id,
|
|
)
|
|
|
|
assert result
|
|
assert u"error" not in result
|
|
assert result[ u"notebook" ].object_id == self.blog_notebook.object_id
|
|
|
|
def test_blog_rss( self ):
|
|
result = self.http_get(
|
|
"/blog?rss",
|
|
)
|
|
|
|
assert result
|
|
assert u"error" not in result
|
|
assert result[ u"notebook" ].object_id == self.blog_notebook.object_id
|
|
|
|
def test_guide( self ):
|
|
result = self.http_get(
|
|
"/guide",
|
|
)
|
|
|
|
assert result
|
|
assert u"error" not in result
|
|
assert result[ u"notebook" ].object_id == self.guide_notebook.object_id
|
|
|
|
def test_privacy( self ):
|
|
result = self.http_get(
|
|
"/privacy",
|
|
)
|
|
|
|
assert result
|
|
assert u"error" not in result
|
|
assert result[ u"notebook" ].object_id == self.privacy_notebook.object_id
|
|
|
|
def test_upgrade( self ):
|
|
result = self.http_get( "/upgrade" )
|
|
|
|
assert result[ u"user" ].username == u"anonymous"
|
|
assert len( result[ u"notebooks" ] ) == 4
|
|
assert result[ u"notebooks" ][ 0 ].object_id == self.anon_notebook.object_id
|
|
assert result[ u"notebooks" ][ 0 ].name == self.anon_notebook.name
|
|
assert result[ u"notebooks" ][ 0 ].read_write == False
|
|
assert result[ u"notebooks" ][ 0 ].owner == False
|
|
|
|
rate_plan = result[ u"rate_plan" ]
|
|
assert rate_plan
|
|
assert rate_plan[ u"name" ] == u"super"
|
|
assert rate_plan[ u"storage_quota_bytes" ] == 1337 * 10
|
|
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert len( result[ u"startup_notes" ] ) == 0
|
|
assert result[ u"note_read_write" ] is False
|
|
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].title == u"upgrade your wiki"
|
|
assert result[ u"notes" ][ 0 ].notebook_id == self.anon_notebook.object_id
|
|
|
|
contents = result[ u"notes" ][ 0 ].contents
|
|
assert u"upgrade" in contents
|
|
assert u"Super" in contents
|
|
assert u"Extra super" in contents
|
|
|
|
# since the user is not logged in, no subscription buttons should be shown
|
|
assert u"button" not in contents
|
|
|
|
def test_upgrade_after_login( self ):
|
|
self.login()
|
|
|
|
result = self.http_get( "/upgrade", session_id = self.session_id )
|
|
|
|
assert result[ u"user" ].username == self.username
|
|
assert len( result[ u"notebooks" ] ) == 5
|
|
assert result[ u"notebooks" ][ 0 ].object_id == self.notebook.object_id
|
|
assert result[ u"notebooks" ][ 0 ].name == self.notebook.name
|
|
assert result[ u"notebooks" ][ 0 ].read_write == False
|
|
assert result[ u"notebooks" ][ 0 ].owner == False
|
|
|
|
rate_plan = result[ u"rate_plan" ]
|
|
assert rate_plan
|
|
assert rate_plan[ u"name" ] == u"super"
|
|
assert rate_plan[ u"storage_quota_bytes" ] == 1337 * 10
|
|
|
|
assert result[ u"notebook" ].object_id == self.anon_notebook.object_id
|
|
assert len( result[ u"startup_notes" ] ) == 0
|
|
assert result[ u"note_read_write" ] is False
|
|
|
|
assert result[ u"notes" ]
|
|
assert len( result[ u"notes" ] ) == 1
|
|
assert result[ u"notes" ][ 0 ].title == u"upgrade your wiki"
|
|
assert result[ u"notes" ][ 0 ].notebook_id == self.anon_notebook.object_id
|
|
|
|
contents = result[ u"notes" ][ 0 ].contents
|
|
assert u"upgrade" in contents
|
|
assert u"Super" in contents
|
|
assert u"Extra super" in contents
|
|
|
|
# since the user is logged in, subscription buttons should be shown
|
|
assert u"button" in contents
|
|
|
|
def test_next_id( self ):
|
|
result = self.http_get( "/next_id" )
|
|
|
|
assert result.get( "next_id" )
|
|
|
|
result = self.http_get( "/next_id" )
|
|
|
|
assert result.get( "next_id" )
|
|
|
|
def test_404( self ):
|
|
result = self.http_get( "/four_oh_four" )
|
|
|
|
body = result.get( u"body" )
|
|
assert body
|
|
assert len( body ) > 0
|
|
assert u"404" in body[ 0 ]
|
|
|
|
status = result.get( u"status" )
|
|
assert u"404" in status
|
|
|
|
headers = result.get( u"headers" )
|
|
status = headers.get( u"status" )
|
|
assert u"404" in status
|
|
|
|
def login( self ):
|
|
result = self.http_post( "/users/login", dict(
|
|
username = self.username,
|
|
password = self.password,
|
|
login_button = u"login",
|
|
) )
|
|
self.session_id = result[ u"session_id" ]
|
|
|
|
def test_redeem_reset( self ):
|
|
redeem_reset_id = u"foobarbaz"
|
|
result = self.http_get( "/r/%s" % redeem_reset_id )
|
|
|
|
assert result[ u"redirect" ] == u"/users/redeem_reset/%s" % redeem_reset_id
|
|
|
|
def test_redeem_invite( self ):
|
|
invite_id = u"foobarbaz"
|
|
result = self.http_get( "/i/%s" % invite_id )
|
|
|
|
assert result[ u"redirect" ] == u"/users/redeem_invite/%s" % invite_id
|