From 53a8bc4e30b41c982babf1573b3474dece4f2831 Mon Sep 17 00:00:00 2001
From: siad007 <siad.ardroumli@gmail.com>
Date: Mon, 18 May 2015 00:45:27 +0200
Subject: [PATCH] Added socket assumption.

---
 src/Assume.php             |  1 +
 src/Assumptions.php        | 15 +++++++++++++++
 src/Extensions/Network.php | 39 ++++++++++++++++++++++++++++++++++++++
 src/Extensions/System.php  |  4 ++++
 tests/AssumeTest.php       |  4 ++--
 5 files changed, 61 insertions(+), 2 deletions(-)
 create mode 100644 src/Extensions/Network.php

diff --git a/src/Assume.php b/src/Assume.php
index 1d43d3a..27bd33b 100644
--- a/src/Assume.php
+++ b/src/Assume.php
@@ -25,6 +25,7 @@
  */
 class Assume
 {
+    use Extensions\Network;
     use Extensions\System;
 
     /**
diff --git a/src/Assumptions.php b/src/Assumptions.php
index dd0f01f..527f16f 100644
--- a/src/Assumptions.php
+++ b/src/Assumptions.php
@@ -104,3 +104,18 @@ function assumeExtensionLoaded()
         );
     }
 }
+
+if (!function_exists('assumeSocket')) {
+    /**
+     * Make an assumption and throw
+     * {@link MehrAlsNix\Assumptions\AssumptionViolatedException} if it fails.
+     */
+    function assumeSocket()
+    {
+        $args = func_get_args();
+        call_user_func_array(
+            array('MehrAlsNix\Assumptions\Assume', 'assumeSocket'),
+            $args
+        );
+    }
+}
diff --git a/src/Extensions/Network.php b/src/Extensions/Network.php
new file mode 100644
index 0000000..5d26182
--- /dev/null
+++ b/src/Extensions/Network.php
@@ -0,0 +1,39 @@
+<?php
+/**
+ * Assumptions
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ * @copyright 2015 MehrAlsNix (http://www.mehralsnix.de)
+ * @license   http://www.opensource.org/licenses/mit-license.php MIT
+ * @link      http://github.com/MehrAlsNix/Assumptions
+ */
+
+namespace MehrAlsNix\Assumptions\Extensions;
+
+/**
+ * Trait Network
+ * @package MehrAlsNix\Assumptions\Extensions
+ */
+trait Network
+{
+    private static $SOCK_ERR_MSG = 'Unable to create socket: ';
+
+    /**
+     * @param string $address
+     * @param int    $port
+     */
+    public static function assumeSocket($address, $port = 0)
+    {
+        $socket = @socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
+
+        assumeThat($socket, not(is(false)), self::$SOCK_ERR_MSG . socket_last_error($socket));
+        assumeThat(@socket_connect($socket, $address, $port), not(is(false)), 'Unable to connect: ' . $address . ':' . $port);
+    }
+}
diff --git a/src/Extensions/System.php b/src/Extensions/System.php
index 5b47b15..89cc94a 100644
--- a/src/Extensions/System.php
+++ b/src/Extensions/System.php
@@ -32,6 +32,10 @@ public static function assumePhpVersion($atLeast, $message = '')
         assumeTrue(version_compare(phpversion(), $atLeast, '>='), $message);
     }
 
+    /**
+     * @param string $extension
+     * @param string $message
+     */
     public static function assumeExtensionLoaded($extension, $message = '')
     {
         assumeTrue(extension_loaded($extension), $message);
diff --git a/tests/AssumeTest.php b/tests/AssumeTest.php
index 054975d..9e6c9ca 100644
--- a/tests/AssumeTest.php
+++ b/tests/AssumeTest.php
@@ -29,7 +29,7 @@ class AssumeTest extends TestCase
     /**
      * @test
      * @expectedException \MehrAlsNix\Assumptions\AssumptionViolatedException
-     * @expectedExceptionMessage Should stop here...
+     * @expectedExceptionMessage Unable to connect
      */
     public function throwExceptionWithWrongAssumption()
     {
@@ -40,6 +40,6 @@ public function throwExceptionWithWrongAssumption()
         assumeNotNull(1, 2, 3);
         assumeTrue(true, 'Assume "true" failed.');
         assumeFalse(false, 'Assume "false" failed.');
-        assumeFalse(true, 'Should stop here...');
+        assumeSocket('127.0.0.1', 123);
     }
 }