{

use TestTrait, EventuallyConsistentTestTrait, ExecuteCommandTrait {

ExecuteCommandTrait::runCommand as traitRunCommand;

}

private static $commandFile = __DIR__ . '/../spanner.php';

/** @var string instanceId */

protected static $instanceId;

/** @var string databaseId */

protected static $databaseId;

/** @var $instance Instance */

protected static $instance;

/** @var $lastUpdateData int */

protected static $lastUpdateDataTimestamp;

public static function setUpBeforeClass()

{

self::checkProjectEnvVars();

if (!extension_loaded('grpc')) {

self::markTestSkipped('Must enable grpc extension.');

}

$instanceId = self::requireEnv('GOOGLE_SPANNER_INSTANCE_ID');

$spanner = new SpannerClient([

'projectId' => self::$projectId,

]);

self::$databaseId = 'test-' . time() . rand();

self::$instanceId = $instanceId;

self::$instance = $spanner->instance(self::$instanceId);

}

public function testCreateDatabase()

{

$output = $this->runCommand('create-database');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Created database test-', $output);

}

/**

public function testInsertData()

{

$output = $this->runCommand('insert-data');

$this->assertEquals('Inserted data.' . PHP_EOL, $output);

}

/**

public function testQueryData()

{

$output = $this->runCommand('query-data');

$this->assertContains('SingerId: 1, AlbumId: 1, AlbumTitle: Total Junk', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, AlbumTitle: Green', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, AlbumTitle: Terrified', $output);

}

/**

public function testBatchQueryData()

{

$output = $this->runCommand('batch-query-data');

$this->assertContains('SingerId: 1, FirstName: Marc, LastName: Richards', $output);

$this->assertContains('SingerId: 2, FirstName: Catalina, LastName: Smith', $output);

$this->assertContains('SingerId: 3, FirstName: Alice, LastName: Trentor', $output);

$this->assertContains('SingerId: 4, FirstName: Lea, LastName: Martin', $output);

$this->assertContains('SingerId: 5, FirstName: David, LastName: Lomond', $output);

$this->assertContains('Total Partitions:', $output);

$this->assertContains('Total Records: 5', $output);

$this->assertContains('Average Records Per Partition:', $output);

}

/**

public function testReadData()

{

$output = $this->runCommand('read-data');

$this->assertContains('SingerId: 1, AlbumId: 1, AlbumTitle: Total Junk', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, AlbumTitle: Green', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, AlbumTitle: Terrified', $output);

}

/**

public function testAddColumn()

{

$output = $this->runCommand('add-column');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Added the MarketingBudget column.', $output);

}

/**

public function testUpdateData()

{

$output = $this->runCommand('update-data');

self::$lastUpdateDataTimestamp = time();

$this->assertEquals('Updated data.' . PHP_EOL, $output);

}

/**

public function testQueryDataWithNewColumn()

{

$output = $this->runCommand('query-data-with-new-column');

$this->assertContains('SingerId: 1, AlbumId: 1, MarketingBudget:', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, MarketingBudget:', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, MarketingBudget:', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, MarketingBudget:', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, MarketingBudget:', $output);

}

/**

public function testReadWriteTransaction()

{

$this->runCommand('update-data');

$output = $this->runCommand('read-write-transaction');

$this->assertContains('Setting first album\'s budget to 300000 and the second album\'s budget to 300000', $output);

$this->assertContains('Transaction complete.', $output);

}

/**

public function testCreateIndex()

{

$output = $this->runCommand('create-index');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Added the AlbumsByAlbumTitle index.', $output);

}

/**

public function testQueryDataWithIndex()

{

$output = $this->runCommand('query-data-with-index');

$this->assertContains('AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

$this->assertContains('AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

}

/**

public function testReadDataWithIndex()

{

$output = $this->runCommand('read-data-with-index');

$this->assertContains('AlbumId: 1, AlbumTitle: Total Junk', $output);

$this->assertContains('AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

$this->assertContains('AlbumId: 1, AlbumTitle: Green', $output);

$this->assertContains('AlbumId: 3, AlbumTitle: Terrified', $output);

$this->assertContains('AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

}

/**

public function testCreateStoringIndex()

{

$output = $this->runCommand('create-storing-index');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Added the AlbumsByAlbumTitle2 index.', $output);

}

/**

public function testReadDataWithStoringIndex()

{

$output = $this->runCommand('read-data-with-storing-index');

$this->assertContains('AlbumId: 2, AlbumTitle: Forever Hold Your Peace, MarketingBudget:', $output);

$this->assertContains('AlbumId: 2, AlbumTitle: Go, Go, Go, MarketingBudget:', $output);

$this->assertContains('AlbumId: 1, AlbumTitle: Green, MarketingBudget:', $output);

$this->assertContains('AlbumId: 3, AlbumTitle: Terrified, MarketingBudget:', $output);

$this->assertContains('AlbumId: 1, AlbumTitle: Total Junk, MarketingBudget:', $output);

}

/**

public function testReadOnlyTransaction()

{

$output = $this->runCommand('read-only-transaction');

$this->assertContains('SingerId: 1, AlbumId: 1, AlbumTitle: Total Junk', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, AlbumTitle: Green', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, AlbumTitle: Terrified', $output);

}

/**

public function testReadStaleData()

{

// read-stale-data reads data that is exactly 15 seconds old. So, make sure 15 seconds

// have elapsed since testUpdateData().

$elapsed = time() - self::$lastUpdateDataTimestamp;

if ($elapsed < 16) {

sleep(16 - $elapsed);

}

$output = $this->runCommand('read-stale-data');

$this->assertContains('SingerId: 1, AlbumId: 1, AlbumTitle: Total Junk', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, AlbumTitle: Go, Go, Go', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, AlbumTitle: Green', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, AlbumTitle: Forever Hold Your Peace', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, AlbumTitle: Terrified', $output);

}

/**

public function testCreateTableTimestamp()

{

$output = $this->runCommand('create-table-timestamp');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Created Performances table in database test-', $output);

}

/**

public function testInsertDataTimestamp()

{

$output = $this->runCommand('insert-data-timestamp');

$this->assertEquals('Inserted data.' . PHP_EOL, $output);

}

/**

public function testAddTimestampColumn()

{

$output = $this->runCommand('add-timestamp-column');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Added LastUpdateTime as a commit timestamp column in Albums table', $output);

}

/**

public function testUpdateDataTimestamp()

{

$output = $this->runCommand('update-data-timestamp');

$this->assertEquals('Updated data.' . PHP_EOL, $output);

}

/**

public function testQueryDataTimestamp()

{

$output = $this->runCommand('query-data-timestamp');

$this->assertContains('SingerId: 1, AlbumId: 1, MarketingBudget: 1000000, LastUpdateTime: 20', $output);

$this->assertContains('SingerId: 2, AlbumId: 2, MarketingBudget: 750000, LastUpdateTime: 20', $output);

$this->assertContains('SingerId: 1, AlbumId: 2, MarketingBudget: NULL, LastUpdateTime: NULL', $output);

$this->assertContains('SingerId: 2, AlbumId: 1, MarketingBudget: NULL, LastUpdateTime: NULL', $output);

$this->assertContains('SingerId: 2, AlbumId: 3, MarketingBudget: NULL, LastUpdateTime: NULL', $output);

}

/**

public function testInsertStructData()

{

$output = $this->runCommand('insert-struct-data');

$this->assertEquals('Inserted data.' . PHP_EOL, $output);

}

/**

public function testQueryDataWithStruct()

{

$output = $this->runCommand('query-data-with-struct');

$this->assertContains('SingerId: 6', $output);

}

/**

public function testQueryDataWithArrayOfStruct()

{

$output = $this->runCommand('query-data-with-array-of-struct');

$this->assertContains('SingerId: 6', $output);

$this->assertContains('SingerId: 7', $output);

$this->assertContains('SingerId: 8', $output);

}

/**

public function testQueryDataWithStructField()

{

$output = $this->runCommand('query-data-with-struct-field');

$this->assertContains('SingerId: 6', $output);

}

/**

public function testQueryDataWithNestedStructField()

{

$output = $this->runCommand('query-data-with-nested-struct-field');

$this->assertContains('SingerId: 6 SongName: Imagination', $output);

$this->assertContains('SingerId: 9 SongName: Imagination', $output);

}

/**

public function testInsertDataWithDml()

{

$output = $this->runCommand('insert-data-with-dml');

$this->assertContains('Inserted 1 row(s)', $output);

}

/**

public function testUpdateDataWithDml()

{

$output = $this->runCommand('update-data-with-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Updated 1 row(s)', $output);

}

/**

public function testDeleteDataWithDml()

{

$output = $this->runCommand('delete-data-with-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Deleted 1 row(s)', $output);

}

/**

public function testUpdateDataWithDmlTimestamp()

{

$output = $this->runCommand('update-data-with-dml-timestamp');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Updated 2 row(s)', $output);

}

/**

public function testWriteReadWithDml()

{

$output = $this->runCommand('write-read-with-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Timothy Campbell', $output);

}

/**

public function testUpdateDataWithDmlStructs()

{

$output = $this->runCommand('update-data-with-dml-structs');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Updated 1 row(s)', $output);

}

/**

public function testWriteDataWithDML()

{

$output = $this->runCommand('write-data-with-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Inserted 4 row(s)', $output);

}

/**

public function testQueryDataWithParameter()

{

$output = $this->runCommand('query-data-with-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('SingerId: 12, FirstName: Melissa, LastName: Garcia', $output);

}

/**

public function testUpdateDataWithDmlTransaction()

{

$output = $this->runCommand('write-data-with-dml-transaction');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Transaction complete', $output);

}

/**

public function testUpdateDataWithPartitionedDML()

{

$output = $this->runCommand('update-data-with-partitioned-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Updated 3 row(s)', $output);

}

/**

public function testDeleteDataWithPartitionedDML()

{

$output = $this->runCommand('deleted-data-with-partitioned-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Deleted 5 row(s)', $output);

}

/**

public function testUpdateDataWithBatchDML()

{

$output = $this->runCommand('update-data-with-batch-dml');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('Executed 2 SQL statements using Batch DML', $output);

}

/**

public function testCreateTableDatatypes()

{

$output = $this->runCommand('create-table-with-datatypes');

$this->assertContains('Waiting for operation to complete...', $output);

$this->assertContains('Created Venues table in database test-', $output);

}

/**

public function testInsertDataWithDatatypes()

{

$output = $this->runCommand('insert-data-with-datatypes');

$this->assertEquals('Inserted data.' . PHP_EOL, $output);

}

/**

public function testQueryDataWithArrayParameter()

{

$output = $this->runCommand('query-data-with-array-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 19, VenueName: Venue 19, AvailableDate: 2020-11-01', $output);

$this->assertContains('VenueId: 42, VenueName: Venue 42, AvailableDate: 2020-10-01', $output);

}

/**

public function testQueryDataWithBoolParameter()

{

$output = $this->runCommand('query-data-with-bool-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 19, VenueName: Venue 19, OutdoorVenue: True', $output);

}

/**

public function testQueryDataWithBytesParameter()

{

$output = $this->runCommand('query-data-with-bytes-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 4, VenueName: Venue 4', $output);

}

/**

public function testQueryDataWithDateParameter()

{

$output = $this->runCommand('query-data-with-date-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 4, VenueName: Venue 4, LastContactDate: 2018-09-02', $output);

$this->assertContains('VenueId: 42, VenueName: Venue 42, LastContactDate: 2018-10-01', $output);

}

/**

public function testQueryDataWithFloatParameter()

{

$output = $this->runCommand('query-data-with-float-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 4, VenueName: Venue 4, PopularityScore: 0.8', $output);

$this->assertContains('VenueId: 19, VenueName: Venue 19, PopularityScore: 0.9', $output);

}

/**

public function testQueryDataWithIntParameter()

{

$output = $this->runCommand('query-data-with-int-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 19, VenueName: Venue 19, Capacity: 6300', $output);

$this->assertContains('VenueId: 42, VenueName: Venue 42, Capacity: 3000', $output);

}

/**

public function testQueryDataWithStringParameter()

{

$output = $this->runCommand('query-data-with-string-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 42, VenueName: Venue 42', $output);

}

/**

public function testQueryDataWithTimestampParameter()

{

$this->runEventuallyConsistentTest(function () {

$output = $this->runCommand('query-data-with-timestamp-parameter');

self::$lastUpdateDataTimestamp = time();

$this->assertContains('VenueId: 4, VenueName: Venue 4, LastUpdateTime:', $output);

$this->assertContains('VenueId: 19, VenueName: Venue 19, LastUpdateTime:', $output);

$this->assertContains('VenueId: 42, VenueName: Venue 42, LastUpdateTime:', $output);

});

}

private function runCommand($commandName)

{

return $this->traitRunCommand($commandName, [

'instance_id' => self::$instanceId,

'database_id' => self::$databaseId,

]);

}

public static function tearDownAfterClass()

{

if (self::$instance) {

// Clean up database

$database = self::$instance->database(self::$databaseId);

$database->drop();

}

}