<?php
Library::import('recess.lang.ObjectRegistry');

Library::import('recess.database.Databases');
Library::import('recess.database.orm.Model');
Library::import('recess.database.orm.ModelDataSource');

require_once('PHPUnit/Extensions/Database/TestCase.php');

/**
 * !BelongsTo owner, Class: Person, Key: personId, OnDelete: Cascade
 * !Table cars
 */
class Car extends Model {
	
	/** !Column PrimaryKey, Integer, AutoIncrement */
	public $pk;
	
	/** !Column Boolean */
	public $isDriveable;
	
}

/**
 * !HasMany persons
 * !Table political_partys
 */
class PoliticalParty extends Model {}

/**
 * !HasMany persons, Through: Groupship, OnDelete: Delete
 * !Table groups
 */
class Group extends Model {}

/**
 * !BelongsTo group
 * !BelongsTo person
 * !Table groupships
 */
class Groupship extends Model {}

/**
 * !HasMany books, Key: authorId, OnDelete: Cascade
 * !HasMany novels, Key: authorId, Class: Book
 * !HasMany cars, OnDelete: Nullify
 * !HasMany groups, Through: Groupship, OnDelete: Nullify
 * !BelongsTo politicalParty
 * !Table persons
 */
class Person extends Model {}

/**
 * !BelongsTo author, Class: Person, OnDelete: Delete
 * !HasMany chapters, OnDelete: Delete
 * !HasMany generas, Through: BooksGenerasJoin, OnDelete: Delete
 * !Table books
 */
class Book extends Model { }

/**
 * !BelongsTo book, OnDelete: Cascade
 * !BelongsTo genera
 * !Table books_generas_joins
 */
class BooksGenerasJoin extends Model { }

/**
 * !BelongsTo book, OnDelete: Nullify
 * !Table chapters
 */
class Chapter extends Model { }

/**
 * !HasMany books, Through: BooksGenerasJoin, OnDelete: Cascade
 * !HasMany movies, Through: MoviesGenerasJoin
 * !Table generas
 */
class Genera extends Model { }

/**
 * !BelongsTo movie
 * !BelongsTo genera
 * !Table movies_generas_joins
 */
class MoviesGenerasJoin extends Model { }

/**
 * !HasMany generas, Through: MoviesGenerasJoin
 * !Table movies
 */
class Movie extends Model { }

/**
 * !HasMany children, Class: Page, Key: parentId
 * !BelongsTo parent, Class: Page, Key: parentId
 * !Table pages
 */
class Page extends Model {}


abstract class ModelTest extends PHPUnit_Extensions_Database_TestCase {
	protected $source;
	
	function getDataSet() {
		Databases::setDefaultSource($this->source);
		Object::clearDescriptors();
		return $this->createXMLDataSet('recess/database/orm/sample-data.xml');
	}
	
	function tearDown() {
		unset($this->source);
	}

	function testHierarchy() {
		$page = new Page();
		$page = $page->equal('id',1)->first();
		$this->assertEquals($page->children()->count(), 1);
		$this->assertEquals($page->parent(), false);
		$this->assertEquals($page->children()->equal('title','Child 1')->count(), 1);
	}
	
	function testAll() {
		$person = new Person();
		$people = $person->all();
		$this->assertEquals(count($people),6);	

		$genera = Make::a('Genera')->equal('title','Social Healing')->first();
		$this->assertEquals($genera->title, 'Social Healing');
		$this->assertEquals($genera->id, 3);
	}

	function testFindCriteria() {
		$person = new Person();
		$person->age = 23;
		$people = $person->find()->orderBy('lastName DESC');
		$this->assertEquals(count($people),2);
		$this->assertEquals($people[0]->lastName, 'Sutherland');
		$this->assertEquals(get_class($people[0]), 'Person');
	}
	
	function testBooleanType() {
		$car = new Car();
		$car->isDriveable = true;
		$car->insert();
		$carId = $car->pk;
		
		$car = new Car();
		$car->pk = $carId;
		$driveable =  $car->find()->first()->isDriveable;
		
		$this->assertEquals($driveable === true, true);
	}
	
	function testFindTailCriteria() {
		$person = new Person();
		$people = $person->find()->greaterThan('age',22);
		$this->assertEquals(count($people),4);
	}
	
	function testBetween() {
		$people = Make::a('Person')->all()->between('age',21,25);
		$this->assertEquals(count($people), 4);
	}
	
	function testMultipleConditionsOnSingleProperty() {
		$people = Make::a('Person')->greaterThan('age',21)->lessThan('age',25);
		$this->assertEquals(count($people), 4);
	}
	
	function testMultipleConditionsOnStrings() {
		$people = Make::a('Person')->like('firstName','%K%')->like('firstName','%s%');
		$this->assertEquals(count($people), 1);
	}
	
	function testHasManyRelationship() {
		$person = new Person();
		$person->id = 4;
		$books = $person->books()->orderBy('title');
		$this->assertEquals(count($books), 3);
	}
	
	function testHasManyRelationshipCriteria() {
		$person = new Person();
		$person->firstName = 'Barack';
		$person->lastName = 'Obama';
		$books = $person->books()->like('title','%Dream%')->orderBy('title ASC');
		$this->assertEquals(count($books), 2);	
		$this->assertEquals($books[0]->title, 'Dreams from My Father: A Story of Race and Inheritance');
	}
	
	function testAliasHasManyRelationship() {
		$person = new Person();
		$person->id = 4;
		$novels = $person->novels();
		$this->assertEquals(count($novels),3);
	}
	
	function testChainMultipleRelationships() {
		$person = new Person();
		$person->id = 3;
		$generas = $person->books()->generas();
		$this->assertEquals(count($generas), 3);
		
		$person = new Person();
		$generas = $person->equal('age',22)->books()->generas();
		$this->assertEquals(count($generas), 3);
		
		$person = new Person();
		$generas = $person->equal('age',22)->books()->like('title','%Dream%')->generas();
		$this->assertEquals(count($generas), 1);
	}
	
	function testMultipleJoinTables() {		
		$movies = Make::a('Person')
						->equal('age',22)
						->books()
						->like('title','%Dream%') 
						->generas()
						->movies();
		
		$this->assertEquals(count($movies),2);
	}
	
	function testBelongsTo() {
		$book = new Book();
		$book->id = 1;
		$barack = $book->author();
		$this->assertEquals(get_class($barack), 'Person');
		$this->assertEquals($barack->firstName, 'Barack');
		$baracksBooks = $barack->books();
		$this->assertEquals(count($baracksBooks), 3);
	}
	
	function testHasAndBelongsToMany() {
		$book = new Book();
		$book->id = 2;
		$generas = $book->generas();
		$this->assertEquals(count($generas), 2);
		
		$books = $generas[0]->books();
		$this->assertEquals(count($books),4);
	}
	
	function testHasAndBelongsToManyWithCriteria() {
		$book = new Book();
		$generas = 
			$book
				->generas()
				->like('books.title', '%Dream%');
		$this->assertEquals(count($generas),2);
	}
	
	function testUpdate() {
		$people = Make::a('Person')->all();
		$person = $people[0];
		$name = $person->firstName;
		
		$person->firstName = 'UPDATE!';
		$person->update();
		
		$people = Make::a('Person')->all();
		
		$this->assertEquals($person->firstName, $people[0]->firstName);
	}
	
	function testUpdateOnCollection() {
		$person = new Person();
		$person->age = 23;
		$person->lessThan('age',23)->update();
		
		$people = $person->find();
		$this->assertEquals(count($people), 4);
	
		$person = new Person();
		$person->age = 100;
		$person->all()->update();
		
		$people = $person->find();
		$this->assertEquals(count($people),6);
	}
	
	function testInsert() {
		$people = Make::a('Person')->all();
		$people_count = count($people);
		
		$person = new Person();
		$person->firstName = 'Joe';
		$person->lastName = 'Biden';
		$person->age = 65;
		$person->insert();
		
		$people = Make::a('Person')->all();
		$this->assertEquals($people_count + 1, count($people));
	}
	
	function testDelete() {		
		$people = Make::a('Person')->all();
		$people_count = count($people);
		$this->assertTrue($people[0]->delete());
		
		$people = Make::a('Person')->all();
		$this->assertEquals($people_count - 1, count($people));
	}
	
	function testDeleteOnCollection() {
		Make::a('Person')->greaterThanOrEqualTo('age',23)->delete();
		$people = Make::a('Person')->all();
		$this->assertEquals(count($people), 2);
		
		Make::a('Person')->all()->delete();
		$people = Make::a('Person')->all();
		$this->assertEquals(count($people), 0);
	}
	
	function testAddToHasManyRelationship() {
		$person = new Person;
		$person->firstName = 'John';
		$person->age = 22; 
		
		$book = new Book();
		$book->title = 'Obama Wins!';
		
		$person->addToBooks($book);
		
		$this->assertEquals($book->author()->id, $person->id);
	}
	
	function testRemoveFromHasManyRelationship() {
		$barack = Make::a('Person')->equal('firstName', 'Barack')->first();
		
		$barackBooksCount = count($barack->books());
		
		$book = Make::a('Book')->like('title', '%Audacity%')->first();
		$barack->removeFromBooks($book);
		
		$this->assertEquals(count($barack->books()), $barackBooksCount - 1);
	}
	
	function testSetOnBelongsToRelationship() {
		$person = new Person;
		$person->firstName = 'John';
		$person->age = 22; 
		
		$book = new Book();
		$book->title = 'Obama Wins!';
		$book->setAuthor($person);
		
		$this->assertEquals($book->author()->id, $person->id);
	}
	
	function testUnsetOnBelongsToRelationship() {
		$barack = Make::a('Person')->equal('firstName', 'Barack')->first();
		
		$barackBooksCount = count($barack->books());
		
		$book = Make::a('Book')->like('title', '%Audacity%')->first();
		$book->unsetAuthor();
		
		$this->assertEquals(count($barack->books()), $barackBooksCount - 1);
	}
	
	function testNonDefaultPrimaryKey() {
		$person = new Person;
		$person->firstName = 'Katie';
		$person->age = 22;
		
		$car = new Car();
		$car->make = 'Honda';
		$car->insert();
		
		$person->addToCars($car);
		
		$this->assertEquals($car->pk, 3);
		$this->assertEquals($car->owner()->id, $person->id);
	}
	
	function testAddToHasAndBelongsToManyRelationship() {
		$book = new Book();
		$book->title = 'Hello world.';
		
		$genera = new Genera();
		$genera->title = 'Scary';
		
		$book->addToGeneras($genera);
		
		$resultBook = $genera->books()->first();
		
		$this->assertEquals($book->id, $resultBook->id);
	}
	
	function testRemoveFromHasAndBelongsToManyRelationship() {
		$book = Make::a('Book')->like('title','%Sketch%')->first();
		$generas = $book->generas();
		$generasCount = count($generas);
		
		$genera = $generas->first();
		$book->removeFromGeneras($genera);
		
		$this->assertEquals($generasCount - 1, count($book->generas()));
	}
	
	function testHasManyThrough() {
		$kris = Make::a('Person')->equal('firstName','Kris')->first();
		$groups = $kris->groups();
		$this->assertEquals(count($groups), 2);
	}
	
	function testHasManyOnDeleteDelete() {
		$audacity = Make::a('Book')->like('title', '%Audacity%')->first();
		
		$allChaptersCount = count(Make::a('Chapter')->all());
		$audacityChaptersCount = count($audacity->chapters());

		$audacity->delete();
		
		$this->assertEquals(count(Make::a('Chapter')->all()), $allChaptersCount - $audacityChaptersCount);
	}
	
	function testHasManyOnDeleteCascade() {
		$barack = Make::a('Person')->like('firstName','%Barack%')->first();
		$barack->delete();
		$this->assertEquals(count(Make::a('Chapter')->all()), 0);
	}
	
	function testHasManyOnDeleteNullify() {
		$kris = Make::a('Person')->equal('firstName', 'Kris')->first();
		
		$car = Make::a('Car');
		
		$all = $car->all();
		
		$allCarsCount = count($all);
		
		$kris->delete();
		
		$this->assertEquals($allCarsCount, count(Make::a('Car')->all()));
	}
	
	function testBelongsToDeleteNullify() {
		$audacity = Make::a('Book')->like('title','%Audacity%')->first();
		$audacity->chapters()->first()->delete();
		
		$audacity = Make::a('Book')->like('title','%Audacity%');
		$this->assertEquals(1, count($audacity));
	}
	
	function testBelongsToDeleteDelete() {
		$audacity = Make::a('Book')->like('title','%Audacity%')->first();
		$audacity->delete();
		$barack = Make::a('Person')->equal('firstName','Barack');
		$this->assertEquals(0, count($barack));
	}
	
	function testBelongsToDeleteCascade() {
		$car = Make::a('Car')->equal('make','VW')->first();
		$car->delete();
		
		$book = Make::a('Book')->like('title','%Michael Scott%');
		$this->assertEquals(0, count($book));
	}
	
	function testHasManyThroughDeleteNullify() {
		$groupshipsCount = count(Make::a('Groupship')->all());
		
		$kris = Make::a('Person')->equal('firstName','Kris')->first();
		$krisGroupsCount = count($kris->groups());
		
		$kris = Make::a('Person')->equal('firstName','Kris')->first();
		$kris->delete();
		
		$this->assertEquals($groupshipsCount - $krisGroupsCount, count(Make::a('Groupship')->all()));
	}
	
	function testHasManyThroughDeleteDelete() {
		$groupshipsCount = count(Make::a('Groupship')->all());
		
		$members = Make::a('Group')->equal('name','NRA')->persons();
		$nraPeopleCount = count($members);
			
		$nra = Make::a('Group')->equal('name','NRA')->first();
		$nra->delete();
		
		$this->assertEquals($groupshipsCount - $nraPeopleCount, count(Make::a('Groupship')->all()));
	}
	
	function testHasManyThroughOnFind() {
		$nra = Make::a('Group')->equal('name','NRA');
		$nraMembers = $nra->persons();
		$nraMembers2 = $nra->first()->persons();
		$this->assertEquals(count($nraMembers), count($nraMembers2));
	}
	
	function testHasManyThroughDeleteCascade() {
		$booksCount = count(Make::a('Book')->all());
		$sports = Make::a('Genera')->like('title','%Sports%');
		$sportsCount = count($sports->books());
		$sports->delete();
		$this->assertEquals($booksCount - $sportsCount, count(Make::a('Book')->all()));
	}
	
	function testNoLongerByReference() {
		$people = Make::a('Person')->all();
		$kris = $people->equal('firstName','Kris');
		$this->assertNotEquals(count($people),count($kris));
	}
	
	function testTicket68FirstAfterEqual() {
		$person = Make::a('Person')->equal('phone', '321-456-7890')->first();
		$this->assertEquals($person->id, 1);
		$this->assertEquals($person->phone, '321-456-7890');
	}
	

}

?>