server/tests/lib/AvatarManagerTest.php

<?php
/**
 * @author Roeland Jago Douma <rullzer@owncloud.com>
 *
 * @copyright Copyright (c) 2015, ownCloud, Inc.
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */

namespace Test;

use OC\AvatarManager;
use Test\Traits\UserTrait;
use Test\Traits\MountProviderTrait;

/**
 * Class AvatarManagerTest
 * @group DB
 */
class AvatarManagerTest extends \Test\TestCase {
	use UserTrait;
	use MountProviderTrait;

	/** @var AvatarManager */
	private $avatarManager;

	/** @var \OC\Files\Storage\Temporary */
	private $storage;

	public function setUp() {
		parent::setUp();

		$this->createUser('valid-user', 'valid-user');

		$this->storage = new \OC\Files\Storage\Temporary();
		$this->registerMount('valid-user', $this->storage, '/valid-user/');

		$this->avatarManager = \OC::$server->getAvatarManager();
	}

	/**
	 * @expectedException \Exception
	 * @expectedExceptionMessage user does not exist
	 */
	public function testGetAvatarInvalidUser() {
		$this->avatarManager->getAvatar('invalidUser');
	}

	public function testGetAvatarValidUser() {
		$avatar = $this->avatarManager->getAvatar('valid-user');

		$this->assertInstanceOf('\OCP\IAvatar', $avatar);
		$this->assertFalse($this->storage->file_exists('files'));
	}

}