/
DocumentExists.php
77 lines (71 loc) · 1.73 KB
/
DocumentExists.php
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
<?php
/**
* @author Jonathan Moss
* @copyright 2010 Jonathan Moss <xirisr@gmail.com>
* @package MongoUnit
* @subpackage Constraint
*/
namespace mongoUnit\constraint;
require_once 'PHPUnit/Util/Type.php';
/**
* A PHP Unit constraint that checks to ensure a specific document exists
*
* @package MongoUnit
* @subpackage Constraint
*/
class DocumentExists extends \PHPUnit_Framework_Constraint
{
/**
* @var MongoDB
*/
private $db;
private $collection;
/**
* @param \MongoDB $db
* @param \MongoCollection $collection
*/
public function __construct($db, $collection)
{
$this->db = $db;
$this->collection = $collection;
}
/**
* Evaluates the constraint for parameter $other. Returns TRUE if the
* constraint is met, FALSE otherwise.
*
* @param mixed $other Value or object to evaluate.
* @return bool
*/
public function evaluate($other)
{
$documentExists = false;
$query = array('_id' => $other);
$data = $this->db->selectCollection($this->collection)->findOne($query);
if (!empty($data)) {
$documentExists = true;
}
return $documentExists;
}
/**
* Returns a string representation of the constraint.
*
* @return string
*/
public function toString()
{
return "exists in the collection {$this->collection}";
}
/**
* @param mixed $other
* @param string $description
* @param boolean $not
*/
protected function customFailureDescription($other, $description, $not)
{
return \sprintf(
'Failed asserting that the document %s %s',
$other,
$this->toString()
);
}
}