public function testGetOrder() { //order and order_id was not set $this->assertNull($this->model->getOrder()); //set order_id and get order by id $orderId = 123; $order = $this->getMock('Magento\\Sales\\Model\\Order', [], [], '', false); $order->expects($this->once())->method('load')->with($orderId)->willReturnSelf(); $this->orderFactory->expects($this->once())->method('create')->willReturn($order); $this->model->setOrderId($orderId); $this->assertEquals($order, $this->model->getOrder()); //get existed order $this->assertEquals($order, $this->model->getOrder()); }