public function testFetchColumns() { $data = ['test1', 'test2']; $stmt = $this->createMock(\PDOStatement::class); $stmt->method('fetchAll')->willReturn($data); $this->driver->method('query')->willReturn($stmt); static::assertEquals($data, $this->database->fetchColumn('SELECT column1, column2 FROM `table`', 1)); }