EntityDataObjectBuilder.php

Go to the documentation of this file.

<?php
namespace tests\unit\Util;

use Magento\FunctionalTestingFramework\DataGenerator\Objects\EntityDataObject;

class EntityDataObjectBuilder
{
    private $data = [
        "name" => "Hopper",
        "gpa" => "3.5678",
        "phone" => "5555555",
        "isprimary" => "true"
    ];

    private $name = "testDataObject";

    private $type = "testType";

    private $linkedEntities = [];

    private $vars = [];

    public function build()
    {
        return new EntityDataObject(
            $this->name,
            $this->type,
            $this->data,
            $this->linkedEntities,
            null,
            $this->vars
        );
    }

    public function withName($name)
    {
        $this->name = $name;
        return $this;
    }

    public function withType($type)
    {
        $this->type = $type;
        return $this;
    }

    public function withDataFields($fields)
    {
        $this->data = $fields;
        return $this;
    }

    public function withLinkedEntities($linkedEntities)
    {
        $this->linkedEntities = $linkedEntities;
        return $this;
    }
}