--- /dev/null
+<?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),
+ );
+ }
+}