piotrbelina/ImageFetcher.php

## ImageFetcher.php
<?php
class ImageFetcher
{
    protected $url;

    public function __construct()
    {
        libxml_use_internal_errors(true);
    }

    public function getUrl() {
        return $this->url;
    }

    public function setUrl($url) {
        $this->url = $url;
    }

    public function fetch()
    {
        if (!$this->url) {
            throw new InvalidArgumentException('Not set url');
        }
        return $this->extractImages(file_get_contents($this->url));
    }

    /**
     *
     * @param string $html
     * @return array
     */
    public function extractImages($html)
    {
        $images = array();

        $dom = new DOMDocument();
        $dom->loadHTML($html);

        $imgs = $dom->getElementsByTagName('img');

        foreach ($imgs as $img) {
            $images[] = $img->getAttribute('src');
        }

        return $images;
    }

    public function getImageName($url)
    {
        $urlParts = explode('/', $url);
        return array_pop($urlParts);
    }

    public function downloadImages($path)
    {
        $images = $this->fetch();

        foreach ($images as $image) {
            file_put_contents($path . '/'. $this->getImageName($image), file_get_contents($image));
        }
    }
}

## ImageFetcherTest.php
<?php
require_once ('../ImageFetcher.php');
/**
 * Generated by PHPUnit_SkeletonGenerator on 2012-08-17 at 14:53:21.
 */
class ImageFetcherTest extends PHPUnit_Framework_TestCase {

    /**
     * @var ImageFetcher
     */
    protected $object;

    /**
     * Sets up the fixture, for example, opens a network connection.
     * This method is called before a test is executed.
     */
    protected function setUp() {
        $this->object = new ImageFetcher;
    }

    /**
     * Tears down the fixture, for example, closes a network connection.
     * This method is called after a test is executed.
     */
    protected function tearDown() {

    }

    public function testExceptionWithoutUrl() {
        $if = new ImageFetcher;
        $this->setExpectedException('InvalidArgumentException');
        $if->fetch();
    }

    public function testFetching() {
        $this->object->setUrl('http://php.net/');
        $results = $this->object->fetch();

        $this->assertInternalType('array', $results);
    }

    public function testExtractImages()
    {
        $html = '<html><body><img src="test1" alt="" /><p><img src="test2" alt="" />'
            . '</p></body></html>';

        $images = $this->object->extractImages($html);
        $this->assertEquals('test1', $images[0]);
        $this->assertEquals('test2', $images[1]);
    }

    public function testGettingName()
    {
        $url = 'http://static.php.net/www.php.net/images/php.gif';

        $this->assertEquals('php.gif', $this->object->getImageName($url));
    }

    public function testDownloadImage()
    {
        $path = 'tmp';
        if (!file_exists($path)) {
            mkdir($path);
        }
        $this->object->setUrl('http://php.net/');
        $this->object->downloadImages($path);

        $this->assertTrue(file_exists('tmp/php.gif'));
    }

}
	<?php
	class ImageFetcher
	{
	protected $url;

	public function __construct()
	{
	libxml_use_internal_errors(true);
	}

	public function getUrl() {
	return $this->url;
	}

	public function setUrl($url) {
	$this->url = $url;
	}

	public function fetch()
	{
	if (!$this->url) {
	throw new InvalidArgumentException('Not set url');
	}
	return $this->extractImages(file_get_contents($this->url));
	}

	/**
	*
	* @param string $html
	* @return array
	*/
	public function extractImages($html)
	{
	$images = array();

	$dom = new DOMDocument();
	$dom->loadHTML($html);

	$imgs = $dom->getElementsByTagName('img');

	foreach ($imgs as $img) {
	$images[] = $img->getAttribute('src');
	}

	return $images;
	}

	public function getImageName($url)
	{
	$urlParts = explode('/', $url);
	return array_pop($urlParts);
	}

	public function downloadImages($path)
	{
	$images = $this->fetch();

	foreach ($images as $image) {
	file_put_contents($path . '/'. $this->getImageName($image), file_get_contents($image));
	}
	}
	}
	<?php
	require_once ('../ImageFetcher.php');
	/**
	* Generated by PHPUnit_SkeletonGenerator on 2012-08-17 at 14:53:21.
	*/
	class ImageFetcherTest extends PHPUnit_Framework_TestCase {

	/**
	* @var ImageFetcher
	*/
	protected $object;

	/**
	* Sets up the fixture, for example, opens a network connection.
	* This method is called before a test is executed.
	*/
	protected function setUp() {
	$this->object = new ImageFetcher;
	}

	/**
	* Tears down the fixture, for example, closes a network connection.
	* This method is called after a test is executed.
	*/
	protected function tearDown() {

	}

	public function testExceptionWithoutUrl() {
	$if = new ImageFetcher;
	$this->setExpectedException('InvalidArgumentException');
	$if->fetch();
	}

	public function testFetching() {
	$this->object->setUrl('http://php.net/');
	$results = $this->object->fetch();

	$this->assertInternalType('array', $results);
	}

	public function testExtractImages()
	{
	$html = '<html><body><img src="test1" alt="" /><p><img src="test2" alt="" />'
	. '</p></body></html>';

	$images = $this->object->extractImages($html);
	$this->assertEquals('test1', $images[0]);
	$this->assertEquals('test2', $images[1]);
	}

	public function testGettingName()
	{
	$url = 'http://static.php.net/www.php.net/images/php.gif';

	$this->assertEquals('php.gif', $this->object->getImageName($url));
	}

	public function testDownloadImage()
	{
	$path = 'tmp';
	if (!file_exists($path)) {
	mkdir($path);
	}
	$this->object->setUrl('http://php.net/');
	$this->object->downloadImages($path);

	$this->assertTrue(file_exists('tmp/php.gif'));
	}

	}