Fixed the tests for the get_last param.

pull/316/head
Ruslan Baidan 2020-08-05 17:00:38 +02:00
parent 48ef453db2
commit a93aac2baa
No known key found for this signature in database
GPG Key ID: 4B7724C136BF1D89
2 changed files with 86 additions and 26 deletions

View File

@ -232,6 +232,7 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
array_merge([
'anrs' => [$anr1->getUuid(), $anr2->getUuid(), $anr3->getUuid()],
'type' => StatsDataObject::TYPE_COMPLIANCE,
'get_last' => 0,
], $defaultDates),
$queryParams
);
@ -241,23 +242,8 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
'type' => StatsDataObject::TYPE_RISK,
'anr' => '1232-31abcd-213efgh-123klmp',
'data' => [
'total' => [
[
'label' => 'Low risks',
'value' => 50,
],
[
'label' => 'Medium risks',
'value' => 30,
],
[
'label' => 'High risks',
'value' => 10,
],
],
'risks' => [],
'total' => [],
],
])
]);
@ -265,13 +251,14 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
$stats = $this->statsAnrService->getStats(['type' => StatsDataObject::TYPE_RISK, 'anrs' => [1, 3, 7]]);
$this->assertEquals($this->getStatsResponse($stats), $statsResponse);
$this->assertEquals($this->getStatsResponse($stats), $this->getStatsResponse());
parse_str($this->statsApiMockHandler->getLastRequest()->getUri()->getQuery(), $queryParams);
$this->assertEquals(
array_merge([
'anrs' => [$anr1->getUuid(), $anr3->getUuid()],
'type' => StatsDataObject::TYPE_RISK,
'get_last' => 0,
], $defaultDates),
$queryParams
);
@ -307,21 +294,25 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
parse_str($this->statsApiMockHandler->getLastRequest()->getUri()->getQuery(), $queryParams);
$this->assertEquals(array_merge([
'anrs' => $anrUuids,
'type' => StatsDataObject::TYPE_RISK
'type' => StatsDataObject::TYPE_RISK,
'get_last' => 0,
], $defaultDates), $queryParams);
$this->statsAnrService->getStats(['type' => StatsDataObject::TYPE_VULNERABILITY]);
parse_str($this->statsApiMockHandler->getLastRequest()->getUri()->getQuery(), $queryParams);
$this->assertEquals(array_merge(['type' => StatsDataObject::TYPE_VULNERABILITY], $defaultDates), $queryParams);
$this->assertEquals(
array_merge(['type' => StatsDataObject::TYPE_VULNERABILITY], $defaultDates + ['get_last' => 0]),
$queryParams
);
}
public function testItCanSendDifferentParamsToGetTheStats()
{
$user = $this->createMock(User::class);
$user->expects($this->exactly(2))->method('hasRole')->with(UserRole::USER_ROLE_CEO)->willReturn(true);
$user->expects($this->exactly(3))->method('hasRole')->with(UserRole::USER_ROLE_CEO)->willReturn(true);
$user->expects($this->never())->method('getUserAnrs');
$this->connectedUserService->expects($this->exactly(2))->method('getConnectedUser')->willReturn($user);
$this->connectedUserService->expects($this->exactly(3))->method('getConnectedUser')->willReturn($user);
/** @var AnrTable $anrTable */
$anrTable = $this->getApplicationServiceLocator()->get(AnrTable::class);
@ -335,6 +326,7 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
'dateTo' => (new DateTime())->format('Y-m-d'),
];
$this->statsApiMockHandler->append(new Response(200, [], $this->getStatsResponse()));
$this->statsApiMockHandler->append(new Response(200, [], $this->getStatsResponse()));
$this->statsApiMockHandler->append(new Response(200, [], $this->getStatsResponse()));
@ -348,7 +340,8 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
$this->assertEquals(array_merge([
'anrs' => $anrUuids,
'aggregation_period' => 'month',
'type' => StatsDataObject::TYPE_COMPLIANCE
'type' => StatsDataObject::TYPE_COMPLIANCE,
'get_last' => 0,
], [
'date_from' => $datesRange['dateFrom'],
'date_to' => $datesRange['dateTo'],
@ -362,11 +355,23 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
parse_str($this->statsApiMockHandler->getLastRequest()->getUri()->getQuery(), $queryParams);
$this->assertEquals(array_merge([
'aggregation_period' => 'week',
'get_last' => 0,
'type' => StatsDataObject::TYPE_VULNERABILITY
], [
'date_from' => $datesRange['dateFrom'],
'date_to' => $datesRange['dateTo'],
]), $queryParams);
$this->statsAnrService->getStats(array_merge([
'getLast' => true,
'type' => StatsDataObject::TYPE_CARTOGRAPHY,
], $datesRange));
parse_str($this->statsApiMockHandler->getLastRequest()->getUri()->getQuery(), $queryParams);
$this->assertEquals(array_merge([
'get_last' => 1,
'type' => StatsDataObject::TYPE_CARTOGRAPHY
]), $queryParams);
}
public function testItFetchesStatsForDefaultPeriodIfFromAndToDatesAreNotPassed()
@ -392,6 +397,7 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
'date_from' => $defaultDates['dateFrom'],
'date_to' => $defaultDates['dateTo'],
'type' => StatsDataObject::TYPE_THREAT,
'get_last' => 0,
], $queryParams);
$this->statsAnrService->getStats([
@ -404,6 +410,7 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
'date_from' => (new DateTime())->modify('-6 months')->format('Y-m-d'),
'date_to' => $defaultDates['dateTo'],
'type' => StatsDataObject::TYPE_THREAT,
'get_last' => 0,
], $queryParams);
$dateTo = (new DateTimeImmutable())->modify('-6 months');
@ -413,7 +420,8 @@ class StatsAnrServiceTest extends AbstractIntegrationTestCase
$this->assertEquals([
'type' => StatsDataObject::TYPE_THREAT,
'date_from' => $dateTo->modify('-' . StatsAnrService::DEFAULT_STATS_DATES_RANGE)->format('Y-m-d'),
'date_to' => $dateTo->format('Y-m-d')
'date_to' => $dateTo->format('Y-m-d'),
'get_last' => 0,
], $queryParams);
}

View File

@ -78,7 +78,8 @@ class GetStatsQueryParamsValidatorTest extends AbstractUnitTestCase
'dateTo' => '2020-06-01',
'anrs' => [],
'type' => StatsDataObject::TYPE_RISK,
'aggregationPeriod' => null
'aggregationPeriod' => null,
'getLast' => false,
],
$this->getStatsQueryParamsValidator->getValidData()
);
@ -96,7 +97,8 @@ class GetStatsQueryParamsValidatorTest extends AbstractUnitTestCase
'dateTo' => '2019-12-01',
'anrs' => [],
'type' => StatsDataObject::TYPE_RISK,
'aggregationPeriod' => null
'aggregationPeriod' => null,
'getLast' => false,
],
$this->getStatsQueryParamsValidator->getValidData()
);
@ -169,4 +171,54 @@ class GetStatsQueryParamsValidatorTest extends AbstractUnitTestCase
'aggregationType' => 'week',
]));
}
public function testGetLastIsSetToFalseWhenNotPassed()
{
static::assertTrue($this->getStatsQueryParamsValidator->isValid([
'type' => StatsDataObject::TYPE_RISK,
]));
static::assertEquals(
[
'type' => StatsDataObject::TYPE_RISK,
'getLast' => false,
'anrs' => [],
'dateFrom' => null,
'dateTo' => null,
'aggregationPeriod' => null,
],
$this->getStatsQueryParamsValidator->getValidData()
);
static::assertTrue($this->getStatsQueryParamsValidator->isValid([
'type' => StatsDataObject::TYPE_RISK,
'getLast' => true,
]));
static::assertEquals(
[
'type' => StatsDataObject::TYPE_RISK,
'getLast' => true,
'anrs' => [],
'dateFrom' => null,
'dateTo' => null,
'aggregationPeriod' => null,
],
$this->getStatsQueryParamsValidator->getValidData()
);
static::assertTrue($this->getStatsQueryParamsValidator->isValid([
'type' => StatsDataObject::TYPE_RISK,
'getLast' => 'the value is converted to boolean.',
]));
static::assertEquals(
[
'type' => StatsDataObject::TYPE_RISK,
'getLast' => true,
'anrs' => [],
'dateFrom' => null,
'dateTo' => null,
'aggregationPeriod' => null,
],
$this->getStatsQueryParamsValidator->getValidData()
);
}
}