summaryrefslogtreecommitdiffstats
path: root/library/Director/Test/IcingaObjectTestCase.php
blob: 5d0dde3e5ab6f6044738fa60228928d9c6130c0f (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
93
94
<?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(): void
    {
        if ($this->hasDb()) {
            /** @var IcingaObject $object */
            foreach (array_reverse($this->createdObjects) as $object) {
                $object->delete();
            }

            if ($this->subject !== null) {
                $this->subject->delete();
            }
        }

        parent::tearDown();
    }
}