from nose.tools import raises
from model.User import User
from model.Notebook import Notebook


class Test_user( object ):
  def setUp( self ):
    self.object_id = u"17"
    self.username = u"bob"
    self.password = u"foobar"
    self.email_address = u"bob@example.com"

    self.user = User( self.object_id, self.username, self.password, self.email_address )

  def test_create( self ):
    assert self.user.username == self.username
    assert self.user.email_address == self.email_address
    assert self.user.notebooks == []
    assert self.user.storage_bytes == 0
    assert self.user.rate_plan == 0

  def test_check_correct_password( self ):
    assert self.user.check_password( self.password ) == True

  def test_check_incorrect_password( self ):
    assert self.user.check_password( u"wrong" ) == False

  def test_set_password( self ):
    previous_revision = self.user.revision
    new_password = u"newpass"
    self.user.password = new_password

    assert self.user.check_password( self.password ) == False
    assert self.user.check_password( new_password ) == True
    assert self.user.revision > previous_revision

  def test_set_none_password( self ):
    previous_revision = self.user.revision
    new_password = None
    self.user.password = new_password

    assert self.user.check_password( self.password ) == False
    assert self.user.check_password( new_password ) == False
    assert self.user.revision > previous_revision

  def test_set_notebooks( self ):
    previous_revision = self.user.revision
    notebook_id = u"33"
    notebook = Notebook( notebook_id, u"my notebook" )
    self.user.notebooks = [ notebook ]
    
    assert len( self.user.notebooks ) == 1
    assert self.user.notebooks[ 0 ].object_id == notebook_id
    assert self.user.revision > previous_revision

  def test_set_storage_bytes( self ):
    previous_revision = self.user.revision
    storage_bytes = 44
    self.user.storage_bytes = storage_bytes
    
    assert self.user.storage_bytes == storage_bytes
    assert self.user.revision > previous_revision

  def test_set_rate_plan( self ):
    previous_revision = self.user.revision
    rate_plan = 2
    self.user.rate_plan = rate_plan
    
    assert self.user.rate_plan == rate_plan
    assert self.user.revision > previous_revision


class Test_user_with_notebooks( object ):
  def setUp( self ):
    self.object_id = u"17"
    self.username = u"bob"
    self.password = u"foobar"
    self.email_address = u"bob@example.com"
    trash = Notebook( u"32", u"trash" )

    self.notebooks = [
      Notebook( u"33", u"my notebook", trash ),
      Notebook( u"34", u"my other notebook" ),
    ]

    self.user = User( self.object_id, self.username, self.password, self.email_address, self.notebooks )

  def test_create( self ):
    assert self.user.username == self.username
    assert self.user.email_address == self.email_address
    assert self.user.notebooks == self.notebooks

  def test_set_existing_notebooks( self ):
    previous_revision = self.user.revision
    self.user.notebooks = [ self.notebooks[ 1 ] ]
    
    assert len( self.user.notebooks ) == 1
    assert self.user.notebooks[ 0 ].object_id == self.notebooks[ 1 ].object_id
    assert self.user.revision > previous_revision

  def test_set_new_notebooks( self ):
    previous_revision = self.user.revision
    notebook_id = u"35"
    notebook = Notebook( notebook_id, u"my new notebook" )
    self.user.notebooks = [ notebook ]
    
    assert len( self.user.notebooks ) == 1
    assert self.user.notebooks[ 0 ].object_id == notebook_id
    assert self.user.revision > previous_revision

  def test_has_access_true( self ):
    assert self.user.has_access( self.notebooks[ 0 ].object_id ) == True

  def test_has_access_false( self ):
    notebook_id = u"35"
    notebook = Notebook( notebook_id, u"my new notebook" )
    assert self.user.has_access( notebook.object_id ) == False

  def test_has_access_to_trash_true( self ):
    assert self.user.has_access( self.notebooks[ 0 ].trash.object_id ) == True

  def test_has_access_to_trash_false( self ):
    notebook_id = u"35"
    trash_id = u"36"
    trash = Notebook( trash_id, u"trash" )
    notebook = Notebook( notebook_id, u"my new notebook", trash )
    assert self.user.has_access( notebook.object_id ) == False

  def test_to_dict( self ):
    d = self.user.to_dict()

    assert d.get( "username" ) == self.username
    assert d.get( "storage_bytes" ) == self.user.storage_bytes
    assert d.get( "rate_plan" ) == self.user.rate_plan