phpstan-src/tests/PHPStan/Analyser/nsrt/array-chunk.php at 355d8243d3a4e8447c92b6a5faabcb162a7b49b7 · VincentLanglet/phpstan-src · GitHub

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
<?php declare(strict_types = 1);

namespace ArrayChunk;

use function PHPStan\Testing\assertType;

class Foo
{

	public function generalArrays(array $arr): void
	{
		/** @var mixed[] $arr */
		assertType('list<non-empty-list<mixed>>', array_chunk($arr, 2));
		assertType('list<non-empty-array<mixed>>', array_chunk($arr, 2, true));

		/** @var array<string, int> $arr */
		assertType('list<non-empty-list<int>>', array_chunk($arr, 2));
		assertType('list<non-empty-array<string, int>>', array_chunk($arr, 2, true));

		/** @var non-empty-array<int|string, bool> $arr */
		assertType('non-empty-list<non-empty-list<bool>>', array_chunk($arr, 1));
		assertType('non-empty-list<non-empty-array<int|string, bool>>', array_chunk($arr, 1, true));
	}


	public function constantArrays(array $arr): void
	{
		/** @var array{a: 0, 17: 1, b: 2} $arr */
		assertType('array{array{0, 1}, array{2}}', array_chunk($arr, 2));
		assertType('array{array{a: 0, 17: 1}, array{b: 2}}', array_chunk($arr, 2, true));
		assertType('array{array{0}, array{1}, array{2}}', array_chunk($arr, 1));
		assertType('array{array{a: 0}, array{17: 1}, array{b: 2}}', array_chunk($arr, 1, true));
	}

	public function constantArraysWithOptionalKeys(array $arr): void
	{
		/** @var array{a: 0, b?: 1, c: 2} $arr */
		assertType('array{array{a: 0, b?: 1, c?: 2}, array{c?: 2}}', array_chunk($arr, 2, true));
		assertType('array{array{a: 0, b?: 1, c: 2}}', array_chunk($arr, 3, true));
		assertType('array{array{a: 0}, array{b?: 1, c?: 2}, array{c?: 2}}', array_chunk($arr, 1, true));

		/** @var array{a?: 0, b?: 1, c?: 2} $arr */
		assertType('array{array{a?: 0, b?: 1, c?: 2}, array{c?: 2}}', array_chunk($arr, 2, true));
	}

	/**
	 * @param int<2, 3> $positiveRange
	 * @param 2|3 $positiveUnion
	 */
	public function chunkUnionTypeLength(array $arr, $positiveRange, $positiveUnion) {
		/** @var array{a: 0, b?: 1, c: 2} $arr */
		assertType('array{0: list{0: 0, 1?: 1|2, 2?: 2}, 1?: array{0?: 2}}', array_chunk($arr, $positiveRange));
		assertType('array{0: array{a: 0, b?: 1, c?: 2}, 1?: array{c?: 2}}', array_chunk($arr, $positiveRange, true));
		assertType('array{0: list{0: 0, 1?: 1|2, 2?: 2}, 1?: array{0?: 2}}', array_chunk($arr, $positiveUnion));
		assertType('array{0: array{a: 0, b?: 1, c?: 2}, 1?: array{c?: 2}}', array_chunk($arr, $positiveUnion, true));
	}

	/**
	 * @param positive-int $positiveInt
	 * @param int<50, max> $bigger50
	 */
	public function lengthIntRanges(array $arr, int $positiveInt, int $bigger50) {
		assertType('list<non-empty-list>', array_chunk($arr, $positiveInt));
		assertType('list<non-empty-list>', array_chunk($arr, $bigger50));
	}

	/**
	 * @param int<1, 4> $oneToFour
	 * @param int<1, 5> $tooBig
	 */
	function testLimits(array $arr, int $oneToFour, int $tooBig) {
		/** @var array{a: 0, b?: 1, c: 2, d: 3} $arr */
		assertType('array{0: list{0: 0, 1?: 1|2, 2?: 2|3, 3?: 3}, 1?: list{0?: 2|3, 1?: 3}}|array{array{0}, list{0?: 1|2, 1?: 2}, list{0?: 2|3, 1?: 3}, array{0?: 3}}', array_chunk($arr, $oneToFour));
		assertType('non-empty-list<non-empty-list<0|1|2|3>>', array_chunk($arr, $tooBig));
	}

	/** @param array<string, string> $map */
	public function offsets(array $arr, array $map): void
	{
		if (array_key_exists('foo', $arr)) {
			assertType('non-empty-list<non-empty-list>', array_chunk($arr, 2));
			assertType('non-empty-list<non-empty-array>', array_chunk($arr, 2, true));
		}
		if (array_key_exists('foo', $arr) && $arr['foo'] === 'bar') {
			assertType('non-empty-list<non-empty-list>', array_chunk($arr, 2));
			assertType('non-empty-list<non-empty-array>', array_chunk($arr, 2, true));
		}

		if (array_key_exists('foo', $map)) {
			assertType('non-empty-list<non-empty-list<string>>', array_chunk($map, 2));
			assertType('non-empty-list<non-empty-array<string, string>>', array_chunk($map, 2, true));
		}
		if (array_key_exists('foo', $map) && $map['foo'] === 'bar') {
			assertType('non-empty-list<non-empty-list<string>>', array_chunk($map, 2));
			assertType('non-empty-list<non-empty-array<string, string>>', array_chunk($map, 2, true));
		}
	}

}