Google\Cloud\Tests\Datastore\DatastoreSessionHandlerTest::testWrite PHP Method

    public function testWrite()
    {
        $data = 'sessiondata';
        $key = new Key('projectid');
        $key->pathElement(self::KIND, 'sessionid');
        $entity = new Entity($key, ['data' => $data]);
        $this->transaction->upsert($entity)->shouldBeCalledTimes(1);
        $this->transaction->commit()->shouldBeCalledTimes(1);
        $this->datastore->transaction()->shouldBeCalledTimes(1)->willReturn($this->transaction->reveal());
        $this->datastore->key(self::KIND, 'sessionid', ['namespaceId' => self::NAMESPACE_ID])->shouldBeCalledTimes(1)->willReturn($key);
        $that = $this;
        $this->datastore->entity($key, Argument::type('array'))->will(function ($args) use($that, $key, $entity) {
            $that->assertEquals($key, $args[0]);
            $that->assertEquals('sessiondata', $args[1]['data']);
            $that->assertInternalType('int', $args[1]['t']);
            $that->assertTrue(time() >= $args[1]['t']);
            // 2 seconds grace period should be enough
            $that->assertTrue(time() - $args[1]['t'] <= 2);
            return $entity;
        });
        $datastoreSessionHandler = new DatastoreSessionHandler($this->datastore->reveal());
        $datastoreSessionHandler->open(self::NAMESPACE_ID, self::KIND);
        $ret = $datastoreSessionHandler->write('sessionid', $data);
        $this->assertEquals(true, $ret);
    }