save me

vendor/psy/psysh/test/Psy/Test/Input/CodeArgumentTest.php

@@ -0,0 +1,52 @@
+<?php
+
+/*
+ * This file is part of Psy Shell.
+ *
+ * (c) 2012-2017 Justin Hileman
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Psy\Tests\Input;
+
+use Psy\Input\CodeArgument;
+use Symfony\Component\Console\Input\InputArgument;
+
+class CodeArgumentTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @dataProvider getInvalidModes
+     * @expectedException \InvalidArgumentException
+     */
+    public function testInvalidModes($mode)
+    {
+        new CodeArgument('wat', $mode);
+    }
+
+    public function getInvalidModes()
+    {
+        return array(
+            array(InputArgument::IS_ARRAY),
+            array(InputArgument::IS_ARRAY | InputArgument::REQUIRED),
+            array(InputArgument::IS_ARRAY | InputArgument::OPTIONAL),
+        );
+    }
+
+    /**
+     * @dataProvider getValidModes
+     */
+    public function testValidModes($mode)
+    {
+        $this->assertInstanceOf('Psy\Input\CodeArgument', new CodeArgument('yeah', $mode));
+    }
+
+    public function getValidModes()
+    {
+        return array(
+            array(InputArgument::REQUIRED),
+            array(InputArgument::OPTIONAL),
+        );
+    }
+}