Add unit test for limiting message size

dailycoding · dailycoding · commit c19e61bdcc51 · 2018-02-22T19:24:16.000-05:00
diff --git a/test/ConsumerLibCurlTest.php b/test/ConsumerLibCurlTest.php
@@ -68,6 +68,34 @@ function testAlias() {
       "userId" => "user-id"
     )));
   }
+
+  /**
+   * @expectedException \RuntimeException
+   */
+  function testLargeMessageSizeError () {
+    $options = array(
+      "debug"    => true,
+      "consumer" => "lib_curl"
+    );
+
+    $client = new Segment_Client("testlargesize", $options);
+
+    $big_property = "";
+
+    for ($i = 0; $i < 32 * 1024; $i++) {
+      $big_property .= "a";
+    }
+
+    $this->assertTrue($client->track(array(
+      "userId" => "some-user",
+      "event" => "Super Large PHP Event",
+      "properties" => array("big_property" => $big_property)
+    )));
+
+    $client->__destruct();
+  }
+
+
 }
 
 ?>
diff --git a/test/ConsumerSocketTest.php b/test/ConsumerSocketTest.php
@@ -135,6 +135,32 @@ function testLargeMessage () {
     $client->__destruct();
   }
 
+  /**
+   * @expectedException \RuntimeException
+   */
+  function testLargeMessageSizeError () {
+    $options = array(
+      "debug"    => true,
+      "consumer" => "socket"
+    );
+
+    $client = new Segment_Client("testlargesize", $options);
+
+    $big_property = "";
+
+    for ($i = 0; $i < 32 * 1024; $i++) {
+      $big_property .= "a";
+    }
+
+    $this->assertTrue($client->track(array(
+      "userId" => "some-user",
+      "event" => "Super Large PHP Event",
+      "properties" => array("big_property" => $big_property)
+    )));
+
+    $client->__destruct();
+  }
+
   /**
    * @expectedException \RuntimeException
    */
