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
114
115
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Framework\MessageQueue\UseCase;
use Magento\Framework\ObjectManagerInterface;
use Magento\TestFramework\Helper\Bootstrap;
use Magento\Framework\MessageQueue\PublisherInterface;
use Magento\TestFramework\MessageQueue\PublisherConsumerController;
use Magento\TestFramework\MessageQueue\EnvironmentPreconditionException;
use Magento\TestFramework\MessageQueue\PreconditionFailedException;
/**
* Base test case for message queue tests.
*/
class QueueTestCaseAbstract extends \PHPUnit\Framework\TestCase
{
/**
* @var string[]
*/
protected $consumers = [];
/**
* @var ObjectManagerInterface
*/
protected $objectManager;
/**
* @var PublisherInterface
*/
protected $publisher;
/**
* @var string
*/
protected $logFilePath;
/**
* @var int|null
*/
protected $maxMessages = null;
/**
* @var PublisherConsumerController
*/
private $publisherConsumerController;
protected function setUp()
{
$this->objectManager = Bootstrap::getObjectManager();
$this->logFilePath = TESTS_TEMP_DIR . "/MessageQueueTestLog.txt";
$this->publisherConsumerController = $this->objectManager->create(PublisherConsumerController::class, [
'consumers' => $this->consumers,
'logFilePath' => $this->logFilePath,
'maxMessages' => $this->maxMessages,
'appInitParams' => \Magento\TestFramework\Helper\Bootstrap::getInstance()->getAppInitParams()
]);
try {
$this->publisherConsumerController->initialize();
} catch (EnvironmentPreconditionException $e) {
$this->markTestSkipped($e->getMessage());
} catch (PreconditionFailedException $e) {
$this->fail(
$e->getMessage()
);
}
$this->publisher = $this->publisherConsumerController->getPublisher();
}
protected function tearDown()
{
$this->publisherConsumerController->stopConsumers();
}
/**
* Wait for asynchronous handlers to log data to file.
*
* @param int $expectedLinesCount
* @param string $logFilePath
*/
protected function waitForAsynchronousResult($expectedLinesCount, $logFilePath)
{
try {
//$expectedLinesCount, $logFilePath
$this->publisherConsumerController->waitForAsynchronousResult([$this, 'checkLogsExists'], [
$expectedLinesCount, $logFilePath
]);
} catch (PreconditionFailedException $e) {
$this->fail($e->getMessage());
}
}
public function checkLogsExists($expectedLinesCount)
{
$actualCount = file_exists($this->logFilePath) ? count(file($this->logFilePath)) : 0;
return $expectedLinesCount === $actualCount;
}
/**
* Workaround for https://bugs.php.net/bug.php?id=72286
*/
public static function tearDownAfterClass()
{
if (version_compare(phpversion(), '7') == -1) {
$closeConnection = new \ReflectionMethod(\Magento\Amqp\Model\Config::class, 'closeConnection');
$closeConnection->setAccessible(true);
$config = Bootstrap::getObjectManager()->get(\Magento\Amqp\Model\Config::class);
$closeConnection->invoke($config);
}
}
}