<?php namespace yiiunit\framework\db; use yii\db\ActiveQuery; use yiiunit\data\ar\ActiveRecord; use yiiunit\data\ar\Customer; use yiiunit\data\ar\NullValues; use yiiunit\data\ar\OrderItem; use yiiunit\data\ar\Order; use yiiunit\data\ar\Item; /** * @group db * @group mysql */ class ActiveRecordTest extends DatabaseTestCase { protected function setUp() { parent::setUp(); ActiveRecord::$db = $this->getConnection(); } public function testFind() { // find one $result = Customer::find(); $this->assertTrue($result instanceof ActiveQuery); $customer = $result->one(); $this->assertTrue($customer instanceof Customer); // find all $customers = Customer::find()->all(); $this->assertEquals(3, count($customers)); $this->assertTrue($customers[0] instanceof Customer); $this->assertTrue($customers[1] instanceof Customer); $this->assertTrue($customers[2] instanceof Customer); // find by a single primary key $customer = Customer::find(2); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user2', $customer->name); // find by column values $customer = Customer::find(['id' => 2, 'name' => 'user2']); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user2', $customer->name); $customer = Customer::find(['id' => 2, 'name' => 'user1']); $this->assertNull($customer); // find by attributes $customer = Customer::find()->where(['name' => 'user2'])->one(); $this->assertTrue($customer instanceof Customer); $this->assertEquals(2, $customer->id); // find custom column $customer = Customer::find()->select(['*', '(status*2) AS status2']) ->where(['name' => 'user3'])->one(); $this->assertEquals(3, $customer->id); $this->assertEquals(4, $customer->status2); // find count, sum, average, min, max, scalar $this->assertEquals(3, Customer::find()->count()); $this->assertEquals(2, Customer::find()->where('id=1 OR id=2')->count()); $this->assertEquals(6, Customer::find()->sum('id')); $this->assertEquals(2, Customer::find()->average('id')); $this->assertEquals(1, Customer::find()->min('id')); $this->assertEquals(3, Customer::find()->max('id')); $this->assertEquals(3, Customer::find()->select('COUNT(*)')->scalar()); // scope $this->assertEquals(2, Customer::find()->active()->count()); // asArray $customer = Customer::find()->where('id=2')->asArray()->one(); $this->assertEquals([ 'id' => '2', 'email' => 'user2@example.com', 'name' => 'user2', 'address' => 'address2', 'status' => '1', ], $customer); // indexBy $customers = Customer::find()->indexBy('name')->orderBy('id')->all(); $this->assertEquals(3, count($customers)); $this->assertTrue($customers['user1'] instanceof Customer); $this->assertTrue($customers['user2'] instanceof Customer); $this->assertTrue($customers['user3'] instanceof Customer); // indexBy callable $customers = Customer::find()->indexBy(function ($customer) { return $customer->id . '-' . $customer->name; })->orderBy('id')->all(); $this->assertEquals(3, count($customers)); $this->assertTrue($customers['1-user1'] instanceof Customer); $this->assertTrue($customers['2-user2'] instanceof Customer); $this->assertTrue($customers['3-user3'] instanceof Customer); } public function testFindBySql() { // find one $customer = Customer::findBySql('SELECT * FROM tbl_customer ORDER BY id DESC')->one(); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user3', $customer->name); // find all $customers = Customer::findBySql('SELECT * FROM tbl_customer')->all(); $this->assertEquals(3, count($customers)); // find with parameter binding $customer = Customer::findBySql('SELECT * FROM tbl_customer WHERE id=:id', [':id' => 2])->one(); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user2', $customer->name); } public function testFindLazy() { /** @var Customer $customer */ $customer = Customer::find(2); $orders = $customer->orders; $this->assertEquals(2, count($orders)); $orders = $customer->getOrders()->where('id=3')->all(); $this->assertEquals(1, count($orders)); $this->assertEquals(3, $orders[0]->id); } public function testFindEager() { $customers = Customer::find()->with('orders')->all(); $this->assertEquals(3, count($customers)); $this->assertEquals(1, count($customers[0]->orders)); $this->assertEquals(2, count($customers[1]->orders)); } public function testFindLazyVia() { /** @var Order $order */ $order = Order::find(1); $this->assertEquals(1, $order->id); $this->assertEquals(2, count($order->items)); $this->assertEquals(1, $order->items[0]->id); $this->assertEquals(2, $order->items[1]->id); $order = Order::find(1); $order->id = 100; $this->assertEquals([], $order->items); } public function testFindEagerViaRelation() { $orders = Order::find()->with('items')->orderBy('id')->all(); $this->assertEquals(3, count($orders)); $order = $orders[0]; $this->assertEquals(1, $order->id); $this->assertEquals(2, count($order->items)); $this->assertEquals(1, $order->items[0]->id); $this->assertEquals(2, $order->items[1]->id); } public function testFindLazyViaTable() { /** @var Order $order */ $order = Order::find(1); $this->assertEquals(1, $order->id); $this->assertEquals(2, count($order->books)); $this->assertEquals(1, $order->items[0]->id); $this->assertEquals(2, $order->items[1]->id); $order = Order::find(2); $this->assertEquals(2, $order->id); $this->assertEquals(0, count($order->books)); } public function testFindEagerViaTable() { $orders = Order::find()->with('books')->orderBy('id')->all(); $this->assertEquals(3, count($orders)); $order = $orders[0]; $this->assertEquals(1, $order->id); $this->assertEquals(2, count($order->books)); $this->assertEquals(1, $order->books[0]->id); $this->assertEquals(2, $order->books[1]->id); $order = $orders[1]; $this->assertEquals(2, $order->id); $this->assertEquals(0, count($order->books)); $order = $orders[2]; $this->assertEquals(3, $order->id); $this->assertEquals(1, count($order->books)); $this->assertEquals(2, $order->books[0]->id); } public function testFindNestedRelation() { $customers = Customer::find()->with('orders', 'orders.items')->all(); $this->assertEquals(3, count($customers)); $this->assertEquals(1, count($customers[0]->orders)); $this->assertEquals(2, count($customers[1]->orders)); $this->assertEquals(0, count($customers[2]->orders)); $this->assertEquals(2, count($customers[0]->orders[0]->items)); $this->assertEquals(3, count($customers[1]->orders[0]->items)); $this->assertEquals(1, count($customers[1]->orders[1]->items)); } public function testLink() { $customer = Customer::find(2); $this->assertEquals(2, count($customer->orders)); // has many $order = new Order; $order->total = 100; $this->assertTrue($order->isNewRecord); $customer->link('orders', $order); $this->assertEquals(3, count($customer->orders)); $this->assertFalse($order->isNewRecord); $this->assertEquals(3, count($customer->getOrders()->all())); $this->assertEquals(2, $order->customer_id); // belongs to $order = new Order; $order->total = 100; $this->assertTrue($order->isNewRecord); $customer = Customer::find(1); $this->assertNull($order->customer); $order->link('customer', $customer); $this->assertFalse($order->isNewRecord); $this->assertEquals(1, $order->customer_id); $this->assertEquals(1, $order->customer->id); // via table $order = Order::find(2); $this->assertEquals(0, count($order->books)); $orderItem = OrderItem::find(['order_id' => 2, 'item_id' => 1]); $this->assertNull($orderItem); $item = Item::find(1); $order->link('books', $item, ['quantity' => 10, 'subtotal' => 100]); $this->assertEquals(1, count($order->books)); $orderItem = OrderItem::find(['order_id' => 2, 'item_id' => 1]); $this->assertTrue($orderItem instanceof OrderItem); $this->assertEquals(10, $orderItem->quantity); $this->assertEquals(100, $orderItem->subtotal); // via model $order = Order::find(1); $this->assertEquals(2, count($order->items)); $this->assertEquals(2, count($order->orderItems)); $orderItem = OrderItem::find(['order_id' => 1, 'item_id' => 3]); $this->assertNull($orderItem); $item = Item::find(3); $order->link('items', $item, ['quantity' => 10, 'subtotal' => 100]); $this->assertEquals(3, count($order->items)); $this->assertEquals(3, count($order->orderItems)); $orderItem = OrderItem::find(['order_id' => 1, 'item_id' => 3]); $this->assertTrue($orderItem instanceof OrderItem); $this->assertEquals(10, $orderItem->quantity); $this->assertEquals(100, $orderItem->subtotal); } public function testUnlink() { // has many $customer = Customer::find(2); $this->assertEquals(2, count($customer->orders)); $customer->unlink('orders', $customer->orders[1], true); $this->assertEquals(1, count($customer->orders)); $this->assertNull(Order::find(3)); // via model $order = Order::find(2); $this->assertEquals(3, count($order->items)); $this->assertEquals(3, count($order->orderItems)); $order->unlink('items', $order->items[2], true); $this->assertEquals(2, count($order->items)); $this->assertEquals(2, count($order->orderItems)); // via table $order = Order::find(1); $this->assertEquals(2, count($order->books)); $order->unlink('books', $order->books[1], true); $this->assertEquals(1, count($order->books)); $this->assertEquals(1, count($order->orderItems)); } public function testInsert() { $customer = new Customer; $customer->email = 'user4@example.com'; $customer->name = 'user4'; $customer->address = 'address4'; $this->assertNull($customer->id); $this->assertTrue($customer->isNewRecord); Customer::$afterSaveNewRecord = null; Customer::$afterSaveInsert = null; $customer->save(); $this->assertEquals(4, $customer->id); $this->assertFalse(Customer::$afterSaveNewRecord); $this->assertTrue(Customer::$afterSaveInsert); $this->assertFalse($customer->isNewRecord); } public function testUpdate() { // save $customer = Customer::find(2); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user2', $customer->name); $this->assertFalse($customer->isNewRecord); Customer::$afterSaveNewRecord = null; Customer::$afterSaveInsert = null; $customer->name = 'user2x'; $customer->save(); $this->assertEquals('user2x', $customer->name); $this->assertFalse($customer->isNewRecord); $this->assertFalse(Customer::$afterSaveNewRecord); $this->assertFalse(Customer::$afterSaveInsert); $customer2 = Customer::find(2); $this->assertEquals('user2x', $customer2->name); // updateCounters $pk = ['order_id' => 2, 'item_id' => 4]; $orderItem = OrderItem::find($pk); $this->assertEquals(1, $orderItem->quantity); $ret = $orderItem->updateCounters(['quantity' => -1]); $this->assertTrue($ret); $this->assertEquals(0, $orderItem->quantity); $orderItem = OrderItem::find($pk); $this->assertEquals(0, $orderItem->quantity); // updateAll $customer = Customer::find(3); $this->assertEquals('user3', $customer->name); $ret = Customer::updateAll(['name' => 'temp'], ['id' => 3]); $this->assertEquals(1, $ret); $customer = Customer::find(3); $this->assertEquals('temp', $customer->name); // updateCounters $pk = ['order_id' => 1, 'item_id' => 2]; $orderItem = OrderItem::find($pk); $this->assertEquals(2, $orderItem->quantity); $ret = OrderItem::updateAllCounters([ 'quantity' => 3, 'subtotal' => -10, ], $pk); $this->assertEquals(1, $ret); $orderItem = OrderItem::find($pk); $this->assertEquals(5, $orderItem->quantity); $this->assertEquals(30, $orderItem->subtotal); } public function testDelete() { // delete $customer = Customer::find(2); $this->assertTrue($customer instanceof Customer); $this->assertEquals('user2', $customer->name); $customer->delete(); $customer = Customer::find(2); $this->assertNull($customer); // deleteAll $customers = Customer::find()->all(); $this->assertEquals(2, count($customers)); $ret = Customer::deleteAll(); $this->assertEquals(2, $ret); $customers = Customer::find()->all(); $this->assertEquals(0, count($customers)); } public function testStoreNull() { $record = new NullValues(); $this->assertNull($record->var1); $this->assertNull($record->var2); $this->assertNull($record->var3); $this->assertNull($record->stringcol); $record->id = 1; $record->var1 = 123; $record->var2 = 456; $record->var3 = 789; $record->stringcol = 'hello!'; $record->save(false); $this->assertTrue($record->refresh()); $this->assertEquals(123, $record->var1); $this->assertEquals(456, $record->var2); $this->assertEquals(789, $record->var3); $this->assertEquals('hello!', $record->stringcol); $record->var1 = null; $record->var2 = null; $record->var3 = null; $record->stringcol = null; $record->save(false); $this->assertTrue($record->refresh()); $this->assertNull($record->var1); $this->assertNull($record->var2); $this->assertNull($record->var3); $this->assertNull($record->stringcol); $record->var1 = 0; $record->var2 = 0; $record->var3 = 0; $record->stringcol = ''; $record->save(false); $this->assertTrue($record->refresh()); $this->assertEquals(0, $record->var1); $this->assertEquals(0, $record->var2); $this->assertEquals(0, $record->var3); $this->assertEquals('', $record->stringcol); } public function testStoreEmpty() { $record = new NullValues(); $record->id = 1; // this is to simulate empty html form submission $record->var1 = ''; $record->var2 = ''; $record->var3 = ''; $record->stringcol = ''; $record->save(false); $this->assertTrue($record->refresh()); // https://github.com/yiisoft/yii2/commit/34945b0b69011bc7cab684c7f7095d837892a0d4#commitcomment-4458225 $this->assertTrue($record->var1 === $record->var2); $this->assertTrue($record->var2 === $record->var3); } /** * Some PDO implementations(e.g. cubrid) do not support boolean values. * Make sure this does not affect AR layer. */ public function testBooleanAttribute() { $customer = new Customer(); $customer->name = 'boolean customer'; $customer->email = 'mail@example.com'; $customer->status = true; $customer->save(false); $customer->refresh(); $this->assertEquals(1, $customer->status); $customer->status = false; $customer->save(false); $customer->refresh(); $this->assertEquals(0, $customer->status); $customers = Customer::find()->where(['status' => true])->all(); $this->assertEquals(2, count($customers)); $customers = Customer::find()->where(['status' => false])->all(); $this->assertEquals(1, count($customers)); } }