blob: a37fcedd29296b40f1dd37565cfee8067bec4d08 (
plain)
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
|
<?php
namespace Icinga\Module\Director\Test;
use Icinga\Module\Director\Objects\IcingaObject;
/**
* Icinga Object test helper class
*/
abstract class IcingaObjectTestCase extends BaseTestCase
{
protected $table;
protected $testObjectName = '___TEST___';
/** @var IcingaObject */
protected $subject = null;
protected $createdObjects = array();
/**
* Creates a fresh object to play with and prepares for tearDown()
*
* @param string $type table to load from
* @param string $object_name of the object
* @param array $properties
* @param bool $storeIt
*
* @return IcingaObject
*/
protected function createObject($object_name, $type = null, $properties = array(), $storeIt = true)
{
if ($type === null) {
$type = $this->table;
}
$properties['object_name'] = '___TEST___' . $type . '_' . $object_name;
$obj = IcingaObject::createByType($type, $properties, $this->getDb());
if ($storeIt === true) {
$obj->store();
$this->prepareObjectTearDown($obj);
}
return $obj;
}
/**
* Helper method for loading an object
*
* @param string $name
* @param null $type
* @return IcingaObject
*/
protected function loadObject($name, $type = null)
{
if ($type === null) {
$type = $this->table;
}
$realName = '___TEST___' . $type . '_' . $name;
return IcingaObject::loadByType($type, $realName, $this->getDb());
}
/**
* Store the object in a list for deletion on tearDown()
*
* @param IcingaObject $object
*
* @return $this
*/
protected function prepareObjectTearDown(IcingaObject $object)
{
$this->assertTrue($object->hasBeenLoadedFromDb());
$this->createdObjects[] = $object;
return $this;
}
/**
* @inheritdoc
*/
public function tearDown()
{
if ($this->hasDb()) {
/** @var IcingaObject $object */
foreach (array_reverse($this->createdObjects) as $object) {
$object->delete();
}
if ($this->subject !== null) {
$this->subject->delete();
}
}
}
}
|
