Added consul configuration with tests.

Gendoria · Feb 19, 2016 · 9b1dc45 · 9b1dc45
1 parent b6560cb
commit 9b1dc45
Show file tree

Hide file tree

Showing 4 changed files with 470 additions and 1 deletion.
diff --git a/README.markdown b/README.markdown
@@ -11,7 +11,7 @@ but in PHP with a simple ZeroMQ interface (rather than Thrift).
 
 This is a serous rewrite of [https://github.com/dvomedia/cruftflake](https://github.com/dvomedia/cruftflake).
 It organizes the code in modules and adds several interfaces, allowing easier extension
-of other server mechanisms.
+of other server and configuration mechanisms.
 
 ## Implementation
 

diff --git a/Tests/ConsulConfigTest.php b/Tests/ConsulConfigTest.php
@@ -0,0 +1,210 @@
+<?php
+
+use Gendoria\CruftFlake\Config\ConsulConfig;
+
+/**
+ * To change this license header, choose License Headers in Project Properties.
+ * To change this template file, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+/**
+ * Description of ConsulConfigTest
+ *
+ * @author Tomasz Struczyński <tomasz.struczynski@isobar.com>
+ */
+class ConsulConfigTest extends PHPUnit_Framework_TestCase
+{
+    public function testGetMachineIdOnEmptyCurrent()
+    {
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '600s',
+            "Behavior" => "delete",
+            'LockDelay' => '300s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+                array('/kv/'.$kvPrefix.$sessionId.'?acquire='.$sessionId, 0, true),
+            )));
+        $config = new ConsulConfig($curl, 600, $kvPrefix);
+        $machine = $config->getMachine();
+        $this->assertEquals(0, $machine);
+    }
+
+    public function testGetMachineIdOnExistingCurrent()
+    {
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '600s',
+            "Behavior" => "delete",
+            'LockDelay' => '300s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+                array('/kv/'.$kvPrefix.$sessionId.'?acquire='.$sessionId, 2, true),
+            )));
+        $curl->expects($this->any())
+            ->method('performGetRequest')
+            ->will($this->returnValueMap(array(
+                array('/kv/'.$kvPrefix.'?recurse', array(
+                    array(
+                        'Key' => 'testold',
+                        'Value' => base64_encode(0),
+                    ),
+                    array(
+                        'Key' => 'testold',
+                        'Value' => base64_encode(1),
+                    ),
+                    array(
+                        'Key' => 'testold',
+                        'Value' => base64_encode(3),
+                    ),
+                )),
+            )));
+        $config = new ConsulConfig($curl, 600, $kvPrefix);
+        $machine = $config->getMachine();
+        $this->assertEquals(2, $machine);
+    }
+
+    public function testGetMachineIdImpossible()
+    {
+        $this->setExpectedException('RuntimeException', 'Cannot acquire machine ID');
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '600s',
+            "Behavior" => "delete",
+            'LockDelay' => '300s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+            )));
+
+        $filledValues = array_map(function($val) {
+            return array(
+                'Key' => 'test'.$val,
+                'Value' => base64_encode($val),
+            );
+        }, range(0, 1023));
+
+        $curl->expects($this->any())
+            ->method('performGetRequest')
+            ->will($this->returnValueMap(array(
+                array('/kv/'.$kvPrefix.'?recurse', $filledValues),
+            )));
+        $config = new ConsulConfig($curl, 600, $kvPrefix);
+        $config->getMachine();
+    }
+
+    public function testHeartbeatNothingToDo()
+    {
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '200s',
+            "Behavior" => "delete",
+            'LockDelay' => '100s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+            )));
+
+        $config = new ConsulConfig($curl, 200, $kvPrefix);
+        $this->assertFalse($config->heartbeat());
+    }
+
+    public function testHeartbeatSessionRenevalSuccessfull()
+    {
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '0s',
+            "Behavior" => "delete",
+            'LockDelay' => '0s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+                array("/session/renew/".$sessionId, null, true)
+            )));
+
+        $config = new ConsulConfig($curl, 0, $kvPrefix);
+        $this->assertFalse($config->heartbeat());
+    }
+
+    public function testHeartbeatSessionRenevalUnsuccessfull()
+    {
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $sessionCreatePayload = $payload = array(
+            'TTL' => '0s',
+            "Behavior" => "delete",
+            'LockDelay' => '0s',
+        );
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnValueMap(array(
+                array('/session/create', json_encode($sessionCreatePayload), array('ID' => $sessionId)),
+                array('/kv/'.$kvPrefix.'?acquire='.$sessionId, $sessionId, true),
+                array("/session/renew/".$sessionId, null, false)
+            )));
+
+        $config = new ConsulConfig($curl, 0, $kvPrefix);
+        $this->assertTrue($config->heartbeat());
+    }
+
+    public function testHeartbeatSessionRenevalAndCreationUnsuccessfull()
+    {
+        $this->setExpectedException('RuntimeException', 'Cannot create session');
+        $kvPrefix = 'test/';
+        $sessionId = 'test';
+        $curl = $this->getMock('\Gendoria\CruftFlake\Config\ConsulCurl', array(), array(''));
+        $curl->expects($this->any())
+            ->method('performPutRequest')
+            ->will($this->returnCallback(function($url) use ($sessionId, $kvPrefix) {
+                static $invCount = 0;
+                if ($url == '/kv/'.$kvPrefix.'?acquire='.$sessionId) {
+                    return true;
+                } elseif($url == '/session/renew') {
+                    return false;
+                } elseif ($url == '/session/create') {
+                    if ($invCount == 0) {
+                        $invCount++;
+                        return array('ID' => $sessionId);
+                    } else {
+                        return null;
+                    }
+                }
+            }));
+
+        try {
+            $config = new ConsulConfig($curl, 0, $kvPrefix);
+        } catch (RuntimeException $e) {
+            $this->fail('Failed too quickly on '.$e->getMessage());
+        }
+        $config->heartbeat();
+    }    
+}