Blame view

sources/tests/lib/dbschema.php 3.87 KB
31b7f2792   Kload   Upgrade to ownclo...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
  <?php
  /**
   * Copyright (c) 2012 Bart Visscher <bartv@thisnet.nl>
   * This file is licensed under the Affero General Public License version 3 or
   * later.
   * See the COPYING-README file.
   */
  
  class Test_DBSchema extends PHPUnit_Framework_TestCase {
  	protected $schema_file = 'static://test_db_scheme';
  	protected $schema_file2 = 'static://test_db_scheme2';
  	protected $table1;
  	protected $table2;
  
  	public function setUp() {
  		$dbfile = OC::$SERVERROOT.'/tests/data/db_structure.xml';
  		$dbfile2 = OC::$SERVERROOT.'/tests/data/db_structure2.xml';
  
  		$r = '_'.OC_Util::generateRandomBytes('4').'_';
  		$content = file_get_contents( $dbfile );
  		$content = str_replace( '*dbprefix*', '*dbprefix*'.$r, $content );
  		file_put_contents( $this->schema_file, $content );
  		$content = file_get_contents( $dbfile2 );
  		$content = str_replace( '*dbprefix*', '*dbprefix*'.$r, $content );
  		file_put_contents( $this->schema_file2, $content );
  
  		$prefix = OC_Config::getValue( "dbtableprefix", "oc_" );
  		
  		$this->table1 = $prefix.$r.'cntcts_addrsbks';
  		$this->table2 = $prefix.$r.'cntcts_cards';
  	}
  
  	public function tearDown() {
  		unlink($this->schema_file);
  		unlink($this->schema_file2);
  	}
  
  	// everything in one test, they depend on each other
  	/**
  	 * @medium
  	 */
  	public function testSchema() {
  		$this->doTestSchemaCreating();
  		$this->doTestSchemaChanging();
  		$this->doTestSchemaDumping();
  		$this->doTestSchemaRemoving();
  	}
  
  	public function doTestSchemaCreating() {
  		OC_DB::createDbFromStructure($this->schema_file);
  		$this->assertTableExist($this->table1);
  		$this->assertTableExist($this->table2);
  	}
  
  	public function doTestSchemaChanging() {
  		OC_DB::updateDbFromStructure($this->schema_file2);
  		$this->assertTableExist($this->table2);
  	}
  
  	public function doTestSchemaDumping() {
  		$outfile = 'static://db_out.xml';
  		OC_DB::getDbStructure($outfile);
  		$content = file_get_contents($outfile);
  		$this->assertContains($this->table1, $content);
  		$this->assertContains($this->table2, $content);
  	}
  
  	public function doTestSchemaRemoving() {
  		OC_DB::removeDBStructure($this->schema_file);
  		$this->assertTableNotExist($this->table1);
  		$this->assertTableNotExist($this->table2);
  	}
  
  	public function tableExist($table) {
  
  		switch (OC_Config::getValue( 'dbtype', 'sqlite' )) {
  			case 'sqlite':
  			case 'sqlite3':
  				$sql = "SELECT name FROM sqlite_master "
  					.  "WHERE type = 'table' AND name = ? "
  					.  "UNION ALL SELECT name FROM sqlite_temp_master "
  					.  "WHERE type = 'table' AND name = ?";
  				$result = \OC_DB::executeAudited($sql, array($table, $table));
  				break;
  			case 'mysql':
  				$sql = 'SHOW TABLES LIKE ?';
  				$result = \OC_DB::executeAudited($sql, array($table));
  				break;
  			case 'pgsql':
  				$sql = 'SELECT tablename AS table_name, schemaname AS schema_name '
  					.  'FROM pg_tables WHERE schemaname NOT LIKE \'pg_%\' '
  					.  'AND schemaname != \'information_schema\' '
  					.  'AND tablename = ?';
  				$result = \OC_DB::executeAudited($sql, array($table));
  				break;
  			case 'oci':
  				$sql = 'SELECT TABLE_NAME FROM USER_TABLES WHERE TABLE_NAME = ?';
  				$result = \OC_DB::executeAudited($sql, array($table));
  				break;
  			case 'mssql':
  				$sql = 'SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = ?';
  				$result = \OC_DB::executeAudited($sql, array($table));
  				break;
  		}
  		
  		$name = $result->fetchOne(); //FIXME checking with '$result->numRows() === 1' does not seem to work?
  		if ($name === $table) {
  			return true;
  		} else {
  			return false;
  		}
  	}
  
  	public function assertTableExist($table) {
  		$this->assertTrue($this->tableExist($table), 'Table ' . $table . ' does not exist');
  	}
  
  	public function assertTableNotExist($table) {
  		$type=OC_Config::getValue( "dbtype", "sqlite" );
  		if( $type == 'sqlite' || $type == 'sqlite3' ) {
  			// sqlite removes the tables after closing the DB
  		} else {
  			$this->assertFalse($this->tableExist($table), 'Table ' . $table . ' exists.');
  		}
  	}
  }