mirror of
https://github.com/Icinga/icingaweb2-module-director.git
synced 2025-07-24 22:34:25 +02:00
Add SyncTest abstract TestCase
So we have a common class to test various syncs
This commit is contained in:
parent
e9546f68bf
commit
0d00d5d97b
52
library/Director/Test/ImportSourceDummy.php
Normal file
52
library/Director/Test/ImportSourceDummy.php
Normal file
@ -0,0 +1,52 @@
|
||||
<?php
|
||||
|
||||
namespace Icinga\Module\Director\Test;
|
||||
|
||||
use Icinga\Module\Director\Hook\ImportSourceHook;
|
||||
|
||||
class ImportSourceDummy extends ImportSourceHook
|
||||
{
|
||||
protected static $rows = array();
|
||||
|
||||
/**
|
||||
* Returns an array containing importable objects
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function fetchData()
|
||||
{
|
||||
return self::$rows;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a list of all available columns
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function listColumns()
|
||||
{
|
||||
$keys = array();
|
||||
foreach (self::$rows as $row) {
|
||||
$keys = array_merge($keys, array_keys($row));
|
||||
}
|
||||
return $keys;
|
||||
}
|
||||
|
||||
public static function clearRows()
|
||||
{
|
||||
self::$rows = array();
|
||||
}
|
||||
|
||||
public static function setRows($rows)
|
||||
{
|
||||
static::clearRows();
|
||||
foreach ($rows as $row) {
|
||||
static::addRow($row);
|
||||
}
|
||||
}
|
||||
|
||||
public static function addRow($row)
|
||||
{
|
||||
self::$rows[] = (object) $row;
|
||||
}
|
||||
}
|
105
library/Director/Test/SyncTest.php
Normal file
105
library/Director/Test/SyncTest.php
Normal file
@ -0,0 +1,105 @@
|
||||
<?php
|
||||
|
||||
namespace Icinga\Module\Director\Test;
|
||||
|
||||
use Icinga\Exception\IcingaException;
|
||||
use Icinga\Module\Director\Data\Db\DbObject;
|
||||
use Icinga\Module\Director\Db\Cache\PrefetchCache;
|
||||
use Icinga\Module\Director\Import\Sync;
|
||||
use Icinga\Module\Director\Objects\IcingaObject;
|
||||
use Icinga\Module\Director\Objects\ImportSource;
|
||||
use Icinga\Module\Director\Objects\SyncProperty;
|
||||
use Icinga\Module\Director\Objects\SyncRule;
|
||||
|
||||
abstract class SyncTest extends BaseTestCase
|
||||
{
|
||||
protected $objectType;
|
||||
|
||||
protected $keyColumn;
|
||||
|
||||
/** @var ImportSource */
|
||||
protected $source;
|
||||
|
||||
/** @var SyncRule */
|
||||
protected $rule;
|
||||
|
||||
/** @var SyncProperty[] */
|
||||
protected $properties = array();
|
||||
|
||||
/** @var Sync */
|
||||
protected $sync;
|
||||
|
||||
public function setUp()
|
||||
{
|
||||
$this->source = ImportSource::create(array(
|
||||
'source_name' => 'testimport',
|
||||
'provider_class' => 'Icinga\\Module\\Director\\Test\\ImportSourceDummy',
|
||||
'key_column' => $this->keyColumn,
|
||||
));
|
||||
$this->source->store($this->getDb());
|
||||
|
||||
$this->rule = SyncRule::create(array(
|
||||
'rule_name' => 'testrule',
|
||||
'object_type' => $this->objectType,
|
||||
'update_policy' => 'merge',
|
||||
'purge_existing' => 'n'
|
||||
));
|
||||
$this->rule->store($this->getDb());
|
||||
|
||||
$this->sync = new Sync($this->rule);
|
||||
}
|
||||
|
||||
public function tearDown()
|
||||
{
|
||||
// properties should be deleted automatically
|
||||
if ($this->rule !== null && $this->rule->hasBeenLoadedFromDb()) {
|
||||
$this->rule->delete();
|
||||
}
|
||||
|
||||
if ($this->source !== null && $this->source->hasBeenLoadedFromDb()) {
|
||||
$this->source->delete();
|
||||
}
|
||||
|
||||
// find objects created by this class and delete them
|
||||
$db = $this->getDb();
|
||||
$dummy = IcingaObject::createByType($this->objectType, array(), $db);
|
||||
$query = $db->getDbAdapter()->select()
|
||||
->from($dummy->getTableName())
|
||||
->where('object_name LIKE ?', 'SYNCTEST_%');
|
||||
|
||||
/** @var IcingaObject $object */
|
||||
foreach (IcingaObject::loadAllByType($this->objectType, $db, $query) as $object) {
|
||||
$object->delete();
|
||||
}
|
||||
|
||||
// make sure cache is clean for other tests
|
||||
PrefetchCache::forget();
|
||||
DbObject::clearAllPrefetchCaches();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array $rows
|
||||
*
|
||||
* @throws IcingaException
|
||||
*/
|
||||
protected function runImport($rows)
|
||||
{
|
||||
ImportSourceDummy::setRows($rows);
|
||||
$this->source->runImport();
|
||||
if ($this->source->get('import_state') !== 'in-sync') {
|
||||
throw new IcingaException('Import failed: %s', $this->source->get('last_error_message'));
|
||||
}
|
||||
}
|
||||
|
||||
protected function setUpProperty($properties = array())
|
||||
{
|
||||
$properties = array_merge(array(
|
||||
'rule_id' => $this->rule->id,
|
||||
'source_id' => $this->source->id,
|
||||
'merge_policy' => 'override',
|
||||
), $properties);
|
||||
|
||||
$this->properties[] = $property = SyncProperty::create($properties);
|
||||
$property->store($this->getDb());
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user