徐总多门店
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

343 lines
14 KiB

2 months ago
<?php
namespace Tests;
use stdClass;
use think\Collection;
use think\helper\Arr;
class ArrTest extends TestCase
{
public function testAdd()
{
$array = Arr::add(['name' => 'ThinkPHP'], 'price', 100);
$this->assertSame(['name' => 'ThinkPHP', 'price' => 100], $array);
}
public function testCrossJoin()
{
// Single dimension
$this->assertSame(
[[1, 'a'], [1, 'b'], [1, 'c']],
Arr::crossJoin([1], ['a', 'b', 'c'])
);
// Square matrix
$this->assertSame(
[[1, 'a'], [1, 'b'], [2, 'a'], [2, 'b']],
Arr::crossJoin([1, 2], ['a', 'b'])
);
// Rectangular matrix
$this->assertSame(
[[1, 'a'], [1, 'b'], [1, 'c'], [2, 'a'], [2, 'b'], [2, 'c']],
Arr::crossJoin([1, 2], ['a', 'b', 'c'])
);
// 3D matrix
$this->assertSame(
[
[1, 'a', 'I'], [1, 'a', 'II'], [1, 'a', 'III'],
[1, 'b', 'I'], [1, 'b', 'II'], [1, 'b', 'III'],
[2, 'a', 'I'], [2, 'a', 'II'], [2, 'a', 'III'],
[2, 'b', 'I'], [2, 'b', 'II'], [2, 'b', 'III'],
],
Arr::crossJoin([1, 2], ['a', 'b'], ['I', 'II', 'III'])
);
// With 1 empty dimension
$this->assertSame([], Arr::crossJoin([], ['a', 'b'], ['I', 'II', 'III']));
$this->assertSame([], Arr::crossJoin([1, 2], [], ['I', 'II', 'III']));
$this->assertSame([], Arr::crossJoin([1, 2], ['a', 'b'], []));
// With empty arrays
$this->assertSame([], Arr::crossJoin([], [], []));
$this->assertSame([], Arr::crossJoin([], []));
$this->assertSame([], Arr::crossJoin([]));
// Not really a proper usage, still, test for preserving BC
$this->assertSame([[]], Arr::crossJoin());
}
public function testDivide()
{
list($keys, $values) = Arr::divide(['name' => 'ThinkPHP']);
$this->assertSame(['name'], $keys);
$this->assertSame(['ThinkPHP'], $values);
}
public function testDot()
{
$array = Arr::dot(['foo' => ['bar' => 'baz']]);
$this->assertSame(['foo.bar' => 'baz'], $array);
$array = Arr::dot([]);
$this->assertSame([], $array);
$array = Arr::dot(['foo' => []]);
$this->assertSame(['foo' => []], $array);
$array = Arr::dot(['foo' => ['bar' => []]]);
$this->assertSame(['foo.bar' => []], $array);
}
public function testExcept()
{
$array = ['name' => 'ThinkPHP', 'price' => 100];
$array = Arr::except($array, ['price']);
$this->assertSame(['name' => 'ThinkPHP'], $array);
}
public function testExists()
{
$this->assertTrue(Arr::exists([1], 0));
$this->assertTrue(Arr::exists([null], 0));
$this->assertTrue(Arr::exists(['a' => 1], 'a'));
$this->assertTrue(Arr::exists(['a' => null], 'a'));
$this->assertFalse(Arr::exists([1], 1));
$this->assertFalse(Arr::exists([null], 1));
$this->assertFalse(Arr::exists(['a' => 1], 0));
}
public function testFirst()
{
$array = [100, 200, 300];
$value = Arr::first($array, function ($value) {
return $value >= 150;
});
$this->assertSame(200, $value);
$this->assertSame(100, Arr::first($array));
$this->assertSame('default', Arr::first([], null, 'default'));
$this->assertSame('default', Arr::first([], function () {
return false;
}, 'default'));
}
public function testLast()
{
$array = [100, 200, 300];
$last = Arr::last($array, function ($value) {
return $value < 250;
});
$this->assertSame(200, $last);
$last = Arr::last($array, function ($value, $key) {
return $key < 2;
});
$this->assertSame(200, $last);
$this->assertSame(300, Arr::last($array));
}
public function testFlatten()
{
// Flat arrays are unaffected
$array = ['#foo', '#bar', '#baz'];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten(['#foo', '#bar', '#baz']));
// Nested arrays are flattened with existing flat items
$array = [['#foo', '#bar'], '#baz'];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Flattened array includes "null" items
$array = [['#foo', null], '#baz', null];
$this->assertSame(['#foo', null, '#baz', null], Arr::flatten($array));
// Sets of nested arrays are flattened
$array = [['#foo', '#bar'], ['#baz']];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Deeply nested arrays are flattened
$array = [['#foo', ['#bar']], ['#baz']];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Nested arrays are flattened alongside arrays
$array = [new Collection(['#foo', '#bar']), ['#baz']];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Nested arrays containing plain arrays are flattened
$array = [new Collection(['#foo', ['#bar']]), ['#baz']];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Nested arrays containing arrays are flattened
$array = [['#foo', new Collection(['#bar'])], ['#baz']];
$this->assertSame(['#foo', '#bar', '#baz'], Arr::flatten($array));
// Nested arrays containing arrays containing arrays are flattened
$array = [['#foo', new Collection(['#bar', ['#zap']])], ['#baz']];
$this->assertSame(['#foo', '#bar', '#zap', '#baz'], Arr::flatten($array));
}
public function testFlattenWithDepth()
{
// No depth flattens recursively
$array = [['#foo', ['#bar', ['#baz']]], '#zap'];
$this->assertSame(['#foo', '#bar', '#baz', '#zap'], Arr::flatten($array));
// Specifying a depth only flattens to that depth
$array = [['#foo', ['#bar', ['#baz']]], '#zap'];
$this->assertSame(['#foo', ['#bar', ['#baz']], '#zap'], Arr::flatten($array, 1));
$array = [['#foo', ['#bar', ['#baz']]], '#zap'];
$this->assertSame(['#foo', '#bar', ['#baz'], '#zap'], Arr::flatten($array, 2));
}
public function testGet()
{
$array = ['products.item' => ['price' => 100]];
$this->assertSame(['price' => 100], Arr::get($array, 'products.item'));
$array = ['products' => ['item' => ['price' => 100]]];
$value = Arr::get($array, 'products.item');
$this->assertSame(['price' => 100], $value);
// Test null array values
$array = ['foo' => null, 'bar' => ['baz' => null]];
$this->assertNull(Arr::get($array, 'foo', 'default'));
$this->assertNull(Arr::get($array, 'bar.baz', 'default'));
// Test null key returns the whole array
$array = ['foo', 'bar'];
$this->assertSame($array, Arr::get($array, null));
// Test $array is empty and key is null
$this->assertSame([], Arr::get([], null));
$this->assertSame([], Arr::get([], null, 'default'));
}
public function testHas()
{
$array = ['products.item' => ['price' => 100]];
$this->assertTrue(Arr::has($array, 'products.item'));
$array = ['products' => ['item' => ['price' => 100]]];
$this->assertTrue(Arr::has($array, 'products.item'));
$this->assertTrue(Arr::has($array, 'products.item.price'));
$this->assertFalse(Arr::has($array, 'products.foo'));
$this->assertFalse(Arr::has($array, 'products.item.foo'));
$array = ['foo' => null, 'bar' => ['baz' => null]];
$this->assertTrue(Arr::has($array, 'foo'));
$this->assertTrue(Arr::has($array, 'bar.baz'));
$array = ['foo', 'bar'];
$this->assertFalse(Arr::has($array, null));
$this->assertFalse(Arr::has([], null));
$array = ['products' => ['item' => ['price' => 100]]];
$this->assertTrue(Arr::has($array, ['products.item']));
$this->assertTrue(Arr::has($array, ['products.item', 'products.item.price']));
$this->assertTrue(Arr::has($array, ['products', 'products']));
$this->assertFalse(Arr::has($array, ['foo']));
$this->assertFalse(Arr::has($array, []));
$this->assertFalse(Arr::has($array, ['products.item', 'products.price']));
$this->assertFalse(Arr::has([], [null]));
}
public function testIsAssoc()
{
$this->assertTrue(Arr::isAssoc(['a' => 'a', 0 => 'b']));
$this->assertTrue(Arr::isAssoc([1 => 'a', 0 => 'b']));
$this->assertTrue(Arr::isAssoc([1 => 'a', 2 => 'b']));
$this->assertFalse(Arr::isAssoc([0 => 'a', 1 => 'b']));
$this->assertFalse(Arr::isAssoc(['a', 'b']));
}
public function testOnly()
{
$array = ['name' => 'ThinkPHP', 'price' => 100, 'orders' => 10];
$array = Arr::only($array, ['name', 'price']);
$this->assertSame(['name' => 'ThinkPHP', 'price' => 100], $array);
}
public function testPrepend()
{
$array = Arr::prepend(['one', 'two', 'three', 'four'], 'zero');
$this->assertSame(['zero', 'one', 'two', 'three', 'four'], $array);
$array = Arr::prepend(['one' => 1, 'two' => 2], 0, 'zero');
$this->assertSame(['zero' => 0, 'one' => 1, 'two' => 2], $array);
}
public function testPull()
{
$array = ['name' => 'ThinkPHP', 'price' => 100];
$name = Arr::pull($array, 'name');
$this->assertSame('ThinkPHP', $name);
$this->assertSame(['price' => 100], $array);
// Only works on first level keys
$array = ['i@example.com' => 'Joe', 'jack@localhost' => 'Jane'];
$name = Arr::pull($array, 'i@example.com');
$this->assertSame('Joe', $name);
$this->assertSame(['jack@localhost' => 'Jane'], $array);
// Does not work for nested keys
$array = ['emails' => ['i@example.com' => 'Joe', 'jack@localhost' => 'Jane']];
$name = Arr::pull($array, 'emails.i@example.com');
$this->assertNull($name);
$this->assertSame(['emails' => ['i@example.com' => 'Joe', 'jack@localhost' => 'Jane']], $array);
}
public function testRandom()
{
$randomValue = Arr::random(['foo', 'bar', 'baz']);
$this->assertContains($randomValue, ['foo', 'bar', 'baz']);
$randomValues = Arr::random(['foo', 'bar', 'baz'], 1);
$this->assertIsArray($randomValues);
$this->assertCount(1, $randomValues);
$this->assertContains($randomValues[0], ['foo', 'bar', 'baz']);
$randomValues = Arr::random(['foo', 'bar', 'baz'], 2);
$this->assertIsArray($randomValues);
$this->assertCount(2, $randomValues);
$this->assertContains($randomValues[0], ['foo', 'bar', 'baz']);
$this->assertContains($randomValues[1], ['foo', 'bar', 'baz']);
}
public function testSet()
{
$array = ['products' => ['item' => ['price' => 100]]];
Arr::set($array, 'products.item.price', 200);
Arr::set($array, 'goods.item.price', 200);
$this->assertSame(['products' => ['item' => ['price' => 200]], 'goods' => ['item' => ['price' => 200]]], $array);
}
public function testWhere()
{
$array = [100, '200', 300, '400', 500];
$array = Arr::where($array, function ($value, $key) {
return is_string($value);
});
$this->assertSame([1 => '200', 3 => '400'], $array);
}
public function testWhereKey()
{
$array = ['10' => 1, 'foo' => 3, 20 => 2];
$array = Arr::where($array, function ($value, $key) {
return is_numeric($key);
});
$this->assertSame(['10' => 1, 20 => 2], $array);
}
public function testForget()
{
$array = ['products' => ['item' => ['price' => 100]]];
Arr::forget($array, null);
$this->assertSame(['products' => ['item' => ['price' => 100]]], $array);
$array = ['products' => ['item' => ['price' => 100]]];
Arr::forget($array, []);
$this->assertSame(['products' => ['item' => ['price' => 100]]], $array);
$array = ['products' => ['item' => ['price' => 100]]];
Arr::forget($array, 'products.item');
$this->assertSame(['products' => []], $array);
$array = ['products' => ['item' => ['price' => 100]]];
Arr::forget($array, 'products.item.price');
$this->assertSame(['products' => ['item' => []]], $array);
$array = ['products' => ['item' => ['price' => 100]]];
Arr::forget($array, 'products.final.price');
$this->assertSame(['products' => ['item' => ['price' => 100]]], $array);
$array = ['shop' => ['cart' => [150 => 0]]];
Arr::forget($array, 'shop.final.cart');
$this->assertSame(['shop' => ['cart' => [150 => 0]]], $array);
$array = ['products' => ['item' => ['price' => ['original' => 50, 'taxes' => 60]]]];
Arr::forget($array, 'products.item.price.taxes');
$this->assertSame(['products' => ['item' => ['price' => ['original' => 50]]]], $array);
$array = ['products' => ['item' => ['price' => ['original' => 50, 'taxes' => 60]]]];
Arr::forget($array, 'products.item.final.taxes');
$this->assertSame(['products' => ['item' => ['price' => ['original' => 50, 'taxes' => 60]]]], $array);
$array = ['products' => ['item' => ['price' => 50], null => 'something']];
Arr::forget($array, ['products.amount.all', 'products.item.price']);
$this->assertSame(['products' => ['item' => [], null => 'something']], $array);
// Only works on first level keys
$array = ['i@example.com' => 'Joe', 'i@thinkphp.com' => 'Jane'];
Arr::forget($array, 'i@example.com');
$this->assertSame(['i@thinkphp.com' => 'Jane'], $array);
// Does not work for nested keys
$array = ['emails' => ['i@example.com' => ['name' => 'Joe'], 'jack@localhost' => ['name' => 'Jane']]];
Arr::forget($array, ['emails.i@example.com', 'emails.jack@localhost']);
$this->assertSame(['emails' => ['i@example.com' => ['name' => 'Joe']]], $array);
}
public function testWrap()
{
$string = 'a';
$array = ['a'];
$object = new stdClass();
$object->value = 'a';
$this->assertSame(['a'], Arr::wrap($string));
$this->assertSame($array, Arr::wrap($array));
$this->assertSame([$object], Arr::wrap($object));
}
}