ScanJobProxyTest.php

<?php
/*
 SPDX-FileCopyrightText: © 2015 Siemens AG
 
 SPDX-License-Identifier: GPL-2.0-only
*/
 
namespace Fossology\Lib\Proxy;
 
use Fossology\Lib\Dao\AgentDao;
use Fossology\Lib\Data\AgentRef;
use Mockery as M;
 
class ScanJobProxyTest extends \PHPUnit\Framework\TestCase
{
  private $agentDaoMock;
  private $uploadId = 23;
  private $agentId = 601;
  private $agentName = 'scanMe';
  private $scanJobProxy;
 
  protected function setUp() : void
  {
    $this->agentDaoMock = M::mock(AgentDao::class);
    $this->scanJobProxy = new ScanJobProxy($this->agentDaoMock,$this->uploadId);
    $this->assertCountBefore = \Hamcrest\MatcherAssert::getCount();
  }
 
  protected function tearDown() : void
  {
    $this->addToAssertionCount(\Hamcrest\MatcherAssert::getCount()-$this->assertCountBefore);
    M::close();
  }
 
  private function prepareScanAgentStatus()
  {
    $reflection = new \ReflectionClass(get_class($this->scanJobProxy));
    $method = $reflection->getMethod('scanAgentStatus');
    $method->setAccessible(true);
 
    return $method;
  }
 
  public function testScanAgentStatus()
  {
    $method = $this->prepareScanAgentStatus();
    $successfulAgents = array(array('agent_id'=>$this->agentId,'agent_rev'=>'a0815','agent_name'=>$this->agentName));
    $this->agentDaoMock->shouldReceive('getSuccessfulAgentEntries')->with($this->agentName, $this->uploadId)
            ->andReturn($successfulAgents);
    $this->agentDaoMock->shouldReceive('getRunningAgentIds')->never();
    $this->agentDaoMock->shouldReceive('getCurrentAgentRef')->with($this->agentName)
            ->andReturn(new AgentRef($this->agentId, $this->agentName, 'a0815'));
 
    $vars = $method->invoke($this->scanJobProxy,$this->agentName);
    assertThat($vars, is(array(
      'successfulAgents'=>$successfulAgents,
      'uploadId'=>$this->uploadId,
      'agentName'=>$this->agentName,
      'currentAgentId'=>$this->agentId,
      'currentAgentRev'=>'a0815'
    )));
  }
 
 
  public function testScanAgentStatusLatestStillRuns()
  {
    $method = $this->prepareScanAgentStatus();
    $successfulAgents = array(array('agent_id'=>$this->agentId,'agent_rev'=>'a0815','agent_name'=>$this->agentName));
    $this->agentDaoMock->shouldReceive('getSuccessfulAgentEntries')->with($this->agentName, $this->uploadId)
            ->andReturn($successfulAgents);
    $runningAgentId = $this->agentId+1;
    $this->agentDaoMock->shouldReceive('getRunningAgentIds')->with($this->uploadId, $this->agentName)
            ->andReturn(array($runningAgentId))->once();
    $this->agentDaoMock->shouldReceive('getCurrentAgentRef')->with($this->agentName)
            ->andReturn(new AgentRef($runningAgentId, $this->agentName, 'b1234'));
 
    $vars = $method->invoke($this->scanJobProxy,$this->agentName);
    assertThat($vars, is(array(
      'successfulAgents'=>$successfulAgents,
      'uploadId'=>$this->uploadId,
      'agentName'=>$this->agentName,
      'isAgentRunning'=>true,
      'currentAgentId'=>$runningAgentId,
      'currentAgentRev'=>'b1234'
    )));
  }
 
 
  public function testScanAgentStatusLatestNotRuns()
  {
    $method = $this->prepareScanAgentStatus();
    $successfulAgents = array(array('agent_id'=>$this->agentId,'agent_rev'=>'a0815','agent_name'=>$this->agentName));
    $this->agentDaoMock->shouldReceive('getSuccessfulAgentEntries')->with($this->agentName, $this->uploadId)
            ->andReturn($successfulAgents);
    $runningAgentId = $this->agentId+1;
    $this->agentDaoMock->shouldReceive('getRunningAgentIds')->with($this->uploadId, $this->agentName)
            ->andReturn(array())->once();
    $this->agentDaoMock->shouldReceive('getCurrentAgentRef')->with($this->agentName)
            ->andReturn(new AgentRef($runningAgentId, $this->agentName, 'b1234'));
 
    $vars = $method->invoke($this->scanJobProxy,$this->agentName);
    assertThat($vars, is(array(
      'successfulAgents'=>$successfulAgents,
      'uploadId'=>$this->uploadId,
      'agentName'=>$this->agentName,
      'isAgentRunning'=>false,
      'currentAgentId'=>$runningAgentId,
      'currentAgentRev'=>'b1234'
    )));
  }
 
 
  public function testScanAgentStatusWithoutSuccess()
  {
    $method = $this->prepareScanAgentStatus();
    $successfulAgents = array();
    $this->agentDaoMock->shouldReceive('getSuccessfulAgentEntries')->with($this->agentName, $this->uploadId)
            ->andReturn($successfulAgents);
    $runningAgentId = $this->agentId+1;
    $this->agentDaoMock->shouldReceive('getRunningAgentIds')->with($this->uploadId, $this->agentName)
            ->andReturn(array($runningAgentId))->once();
    $this->agentDaoMock->shouldReceive('getCurrentAgentRef')->with($this->agentName)
            ->andReturn(new AgentRef($runningAgentId, $this->agentName, 'b1234'));
 
    $vars = $method->invoke($this->scanJobProxy,$this->agentName);
    assertThat($vars, is(array(
      'successfulAgents'=>$successfulAgents,
      'uploadId'=>$this->uploadId,
      'agentName'=>$this->agentName,
      'isAgentRunning'=>true
    )));
  }
 
 
  public function testCreateAgentStatus()
  {
    $successfulAgents = array(array('agent_id'=>$this->agentId,'agent_rev'=>'a0815','agent_name'=>$this->agentName));
    $this->agentDaoMock->shouldReceive('getSuccessfulAgentEntries')->with($this->agentName, $this->uploadId)
            ->andReturn($successfulAgents);
    $this->agentDaoMock->shouldReceive('getRunningAgentIds')->never();
    $this->agentDaoMock->shouldReceive('getCurrentAgentRef')->with($this->agentName)
            ->andReturn(new AgentRef($this->agentId, $this->agentName, 'a0815'));
    $fakedAgentName = 'ghost';
    $this->agentDaoMock->shouldReceive('arsTableExists')->with(M::anyOf($this->agentName,$fakedAgentName))->andReturn(true,false)->twice();
 
    $vars = $this->scanJobProxy->createAgentStatus(array($this->agentName,$fakedAgentName));
    assertThat($vars, is(array(array(
      'successfulAgents'=>$successfulAgents,
      'uploadId'=>$this->uploadId,
      'agentName'=>$this->agentName,
      'currentAgentId'=>$this->agentId,
      'currentAgentRev'=>'a0815'
    ))));
  }
 
 
  private function pretendScanAgentStatus($successfulAgents)
  {
    $reflection = new \ReflectionObject($this->scanJobProxy);
    $prop = $reflection->getProperty('successfulScanners');
    $prop->setAccessible(true);
    $prop->setValue($this->scanJobProxy, $successfulAgents);
  }
 
  public function testGetAgentMap()
  {
    $successfulAgents = array($this->agentName=>array(new AgentRef($this->agentId, $this->agentName, 'a0815')));
    $this->pretendScanAgentStatus($successfulAgents);
 
    $expected = array($this->agentId=>"$this->agentName a0815");
    $map = $this->scanJobProxy->getAgentMap();
    assertThat($map,is(equalTo($expected)));
  }
 
 
  public function testGetLatestSuccessfulAgentIds()
  {
    $otherAgentName = 'drinkMe';
    $otherAgentId = 603;
    $successfulAgents = array($this->agentName=>array(new AgentRef($this->agentId, $this->agentName, 'a0815'),
            new AgentRef($this->agentId-1, $this->agentName, 'beforeA0815')),
        $otherAgentName=>array(new AgentRef($otherAgentId, $otherAgentName, 'coffee')));
    $this->pretendScanAgentStatus($successfulAgents);
 
    $expected = array($this->agentName=>$this->agentId,$otherAgentName=>$otherAgentId);
    $ids = $this->scanJobProxy->getLatestSuccessfulAgentIds();
    assertThat($ids,is(equalTo($expected)));
  }
 
  public function testGetSuccessfulAgents()
  {
    $otherAgentName = 'drinkMe';
    $otherAgentId = 603;
    $successfulAgents = array($this->agentName=>array(new AgentRef($this->agentId, $this->agentName, 'a0815'),
            new AgentRef($this->agentId-1, $this->agentName, 'beforeA0815')),
        $otherAgentName=>array(new AgentRef($otherAgentId, $otherAgentName, 'coffee')));
    $this->pretendScanAgentStatus($successfulAgents);
 
    $expected = array_merge($successfulAgents[$this->agentName],$successfulAgents[$otherAgentName]);
    $ids = $this->scanJobProxy->getSuccessfulAgents();
    assertThat($ids,is(equalTo($expected)));
  }
}