1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
<?php
namespace Basho\Tests;
use Basho\Riak;
use Basho\Riak\Command;
/**
* Scenario tests for when storing / retrieving binary data
*
* @author Christopher Mancini <cmancini at basho d0t com>
*/
class EncodedDataTest extends TestCase
{
const BUCKET = "encodeddata";
const BASE64_KEY = "base64";
const BINARY_KEY = "binary.png";
const TEST_CONTENT_TYPE = "image/png";
/**
* Test storing and fetching an image represented in base64 encoding
*/
public function testBase64EncodedData() {
$image = file_get_contents(__DIR__ . DIRECTORY_SEPARATOR . ".." . DIRECTORY_SEPARATOR . static::TEST_IMG);
$object = (new Riak\DataObject($image))->setContentEncoding("base64")->setContentType(static::TEST_CONTENT_TYPE);
$storeCommand = (new Command\Builder\StoreObject(static::$riak))
->withObject($object)
->buildLocation(static::BASE64_KEY, static::BUCKET)
->build();
$response = $storeCommand->execute();
$this->assertEquals('204', $response->getCode());
$fetchCommand = (new Command\Builder\FetchObject(static::$riak))
->buildLocation(static::BASE64_KEY, static::BUCKET)
->build();
$response = $fetchCommand->execute();
$this->assertEquals('200', $response->getCode());
$this->assertInstanceOf('Basho\Riak\DataObject', $response->getObject());
$this->assertEquals(static::TEST_CONTENT_TYPE, $response->getObject()->getContentType());
$this->assertEquals(base64_encode($image), $storeCommand->getEncodedData());
$this->assertEquals($storeCommand->getEncodedData(), $response->getObject()->getData());
$this->assertEquals(md5($storeCommand->getEncodedData()), md5($response->getObject()->getData()));
}
/**
* Test storing and fetching an image as a base64 embedded HTML img tag so it can be accessed via the browser
*/
public function testNoEncoding() {
$image = file_get_contents(__DIR__ . DIRECTORY_SEPARATOR . ".." . DIRECTORY_SEPARATOR . static::TEST_IMG);
$object = (new Riak\DataObject('<img src="data:image/png;base64,' . base64_encode($image) . '"/>'))->setContentEncoding("none")->setContentType("text/html");
$storeCommand = (new Command\Builder\StoreObject(static::$riak))
->withObject($object)
->buildLocation(static::BASE64_KEY . '.png', static::BUCKET)
->build();
$response = $storeCommand->execute();
$this->assertEquals('204', $response->getCode());
$this->assertEquals(md5($object->getData()), md5($storeCommand->getEncodedData()));
$fetchCommand = (new Command\Builder\FetchObject(static::$riak))
->buildLocation(static::BASE64_KEY . '.png', static::BUCKET)
->build();
$response = $fetchCommand->execute();
$this->assertEquals('200', $response->getCode());
$this->assertInstanceOf('Basho\Riak\DataObject', $response->getObject());
$this->assertEquals($object->getData(), $response->getObject()->getData());
}
/**
* Test storing and fetching a Binary image
*/
public function testBinaryData() {
$image = file_get_contents(__DIR__ . DIRECTORY_SEPARATOR . ".." . DIRECTORY_SEPARATOR . static::TEST_IMG);
$md5 = md5($image);
$object = (new Riak\DataObject($image))->setContentEncoding("binary")->setContentType(static::TEST_CONTENT_TYPE);
$storeCommand = (new Command\Builder\StoreObject(static::$riak))
->withObject($object)
->buildLocation(static::BINARY_KEY, static::BUCKET)
->build();
$response = $storeCommand->execute();
$this->assertEquals('204', $response->getCode());
$this->assertEquals($md5, md5($storeCommand->getEncodedData()));
$fetchCommand = (new Command\Builder\FetchObject(static::$riak))
->buildLocation(static::BINARY_KEY, static::BUCKET)
->build();
$response = $fetchCommand->execute();
$this->assertEquals('200', $response->getCode());
$this->assertInstanceOf('Basho\Riak\DataObject', $response->getObject());
$this->assertEquals(static::TEST_CONTENT_TYPE, $response->getObject()->getContentType());
// Since Riak doesn't return ContentEncoding used to store the object, we have to access
// the raw_data retrieved in the response to avoid data corruption from rawurldecode
$this->assertEquals($md5, md5($response->getObject()->getRawData()));
}
}
|
