Dietcube\Controller::setVars PHP Method

    public function setVars($key, $value = null)
    {
        if (is_array($key)) {
            $this->view_vars = array_merge($this->view_vars, $key);
        } else {
            $this->view_vars[$key] = $value;
        }
        return $this;
    }

 public function testRenderVars()
 {
     $app = $this->getMockBuilder('\\Dietcube\\Application')->disableOriginalConstructor()->getMockForAbstractClass();
     $renderer = $this->getMock('Twig_Environment');
     $renderer->expects($this->any())->method('render')->will($this->returnArgument(1));
     $container = self::getContainerAsFixture(['app' => $app, 'app.renderer' => $renderer]);
     $controller = new Controller($container);
     $controller->setVars('key', 'value');
     $render = $this->getInvokableMethod('render');
     $this->assertEquals(['key' => 'value'], $render->invokeArgs($controller, ['template']));
     $this->assertEquals(['key' => 'value2'], $render->invokeArgs($controller, ['template', ['key' => 'value2']]));
 }