szurubooru/tests/Dao/FavoritesDaoTest.php

<?php
namespace Szurubooru\Tests\Dao;
use Szurubooru\Dao\FavoritesDao;
use Szurubooru\Entities\Favorite;
use Szurubooru\Entities\Post;
use Szurubooru\Entities\User;
use Szurubooru\Services\TimeService;
use Szurubooru\Tests\AbstractDatabaseTestCase;

final class FavoritesDaoTest extends AbstractDatabaseTestCase
{
	private $timeServiceMock;

	public function setUp()
	{
		parent::setUp();
		$this->timeServiceMock = $this->mock(TimeService::class);
	}

	public function testSaving()
	{
		$user = new User(1);
		$user->setName('olivia');

		$post = new Post(2);
		$post->setName('sword');

		$favorite = new Favorite();
		$favorite->setUserId($user->getId());
		$favorite->setPostId($post->getId());
		$favorite->setTime(date('c'));
		$favoritesDao = $this->getFavoritesDao();
		$favoritesDao->save($favorite);

		$savedFavorite = $favoritesDao->findById($favorite->getId());
		$this->assertEquals(1, $savedFavorite->getUserId());
		$this->assertEquals(2, $savedFavorite->getPostId());
		$this->assertEquals($favorite->getTime(), $savedFavorite->getTime());
		$this->assertEquals($user->getId(), $savedFavorite->getUserId());
		$this->assertEquals($post->getId(), $savedFavorite->getPostId());
	}

	private function getFavoritesDao()
	{
		return new FavoritesDao(
			$this->databaseConnection,
			$this->timeServiceMock);
	}
}