tainacan/tests/test-api-roles.php

<?php

namespace Tainacan\Tests;

/**
 * @group api
 *
 */
class TAINACAN_REST_Roles_Controller extends TAINACAN_UnitApiTestCase {

	/**
	 * just while we dont refactor capabilities
	 */
	public static function setUpBeforeClass() {
		parent::setUpBeforeClass();
		$role = get_role('administrator');
		$role->add_cap('tnc_rep_edit_users');
		global $current_user;
		$current_user->get_role_caps();
	}

	public function setUp() {
		parent::setUp();
		// reset WP_Roles object. Possible bug was cleaning database between tests, but not the object
		global $wpdb;
		wp_roles()->roles = get_option($wpdb->prefix . 'user_roles');
		wp_roles()->init_roles();
	}

	public function test_create_get_roles() {

		$request = new \WP_REST_Request('POST', $this->namespace . '/roles');

		$request->set_query_params(['name' => 'New role']);

		$create = $this->server->dispatch($request);

		$this->assertEquals( 201, $create->get_status() );

		$request = new \WP_REST_Request('GET', $this->namespace . '/roles');

		//$request->set_query_params($name_query);

		$name_response = $this->server->dispatch($request);
		$data = $name_response->get_data();
		$this->assertArrayHasKey('tainacan-new-role', $data);
		$this->assertEquals('New role', $data['tainacan-new-role']['name']);

	}

	public function test_create_remove_roles() {
		$request = new \WP_REST_Request('POST', $this->namespace . '/roles');

		$request->set_query_params(['name' => 'Super role']);

		$create = $this->server->dispatch($request);

		$this->assertEquals( 201, $create->get_status() );

		$request = new \WP_REST_Request('DELETE', $this->namespace . '/roles/tainacan-super-role');

		$delete_response = $this->server->dispatch($request);

		$this->assertEquals( 200, $delete_response->get_status() );

		$request = new \WP_REST_Request('GET', $this->namespace . '/roles');

		$name_response = $this->server->dispatch($request);
		$data = $name_response->get_data();
		$this->assertArrayNotHasKey('tainacan-super-role', $data);



	}

	public function test_edit_role() {

		$request = new \WP_REST_Request('POST', $this->namespace . '/roles');

		$request->set_query_params(['name' => 'New role']);

		$create = $this->server->dispatch($request);
		//var_dump($create);
		$this->assertEquals( 201, $create->get_status() );

		$request = new \WP_REST_Request('PATCH', $this->namespace . '/roles/new-role');

		$request->set_query_params(
			[
				'name' => 'Changed name',
				'add_cap' => 'tnc_rep_edit_collections'
			]
		);

		$response = $this->server->dispatch($request);

		$this->assertEquals( 200, $response->get_status() );

		$role = \wp_roles()->roles['tainacan-new-role'];
		$this->assertArrayHasKey('tnc_rep_edit_collections', $role['capabilities']);
		$this->assertTrue($role['capabilities']['tnc_rep_edit_collections']);
		$this->assertEquals('Changed name', $role['name']);

		$request = new \WP_REST_Request('PATCH', $this->namespace . '/roles/new-role');

		$request->set_query_params(
			[
				'add_cap' => 'manage_options'
			]
		);

		$response = $this->server->dispatch($request);

		$this->assertEquals( 400, $response->get_status() );

	}

	public function test_get_role() {
		$request = new \WP_REST_Request('GET', $this->namespace . '/roles/administrator');

		$response = $this->server->dispatch($request);

		$this->assertEquals( 200, $response->get_status() );

		$data = $response->get_data();

		$this->assertEquals( translate_user_role('Administrator'), $data['name'] );
		$this->assertArrayHasKey('manage_options', $data['capabilities']);
		$this->assertTrue($data['capabilities']['manage_options']);

	}

	public function test_add_dependencies() {
		$request = new \WP_REST_Request('POST', $this->namespace . '/roles');

		$request->set_query_params(['name' => 'New role']);

		$create = $this->server->dispatch($request);
		//var_dump($create);
		$this->assertEquals( 201, $create->get_status() );

		$request = new \WP_REST_Request('PATCH', $this->namespace . '/roles/new-role');

		$request->set_query_params(
			[
				'name' => 'Changed name',
				'add_cap' => 'tnc_col_12_edit_items'
			]
		);

		$response = $this->server->dispatch($request);

		$this->assertEquals( 200, $response->get_status() );

		$role = \wp_roles()->roles['tainacan-new-role'];

		$this->assertArrayHasKey('tnc_col_12_edit_items', $role['capabilities']);
		$this->assertTrue($role['capabilities']['tnc_col_12_edit_items']);
		$this->assertArrayHasKey('upload_files', $role['capabilities']);
		$this->assertTrue($role['capabilities']['upload_files']);
	}

	public function test_get_collection_roles() {

		$collection = $this->tainacan_entity_factory->create_entity(
			'collection',
			array(
				'name' => 'teste',
				'status' => 'publish'
			),
			true
		);

		$editor = get_role('editor');
		$contributor = get_role('contributor');
		$author = get_role('author');

		$editor->add_cap('manage_tainacan_collection_' . $collection->get_id());

		$author->add_cap( 'tnc_col_' . $collection->get_id() . '_edit_items' );
		$author->add_cap( 'tnc_col_' . $collection->get_id() . '_edit_metadata' );
		$author->add_cap( 'tnc_col_' . $collection->get_id() . '_edit_filters' );

		$contributor->add_cap( 'tnc_col_all_edit_items' );
		$contributor->add_cap( 'tnc_col_all_edit_published_items' );


		$request = new \WP_REST_Request('GET', $this->namespace . '/collection/' . $collection->get_id() . '/roles');

		$response = $this->server->dispatch($request);
		//var_dump($create);
		$this->assertEquals( 200, $response->get_status() );

		$caps = $response->get_data()['capabilities'];

		$this->assertArrayHasKey('editor', $caps['manage_tainacan_collection_%d']['roles']);

		$this->assertArrayHasKey('author', $caps['tnc_col_%d_edit_items']['roles']);
		$this->assertArrayHasKey('author', $caps['tnc_col_%d_edit_metadata']['roles']);
		$this->assertArrayHasKey('author', $caps['tnc_col_%d_edit_filters']['roles']);

		$this->assertArrayHasKey('contributor', $caps['tnc_col_%d_edit_items']['roles']);
		$this->assertArrayHasKey('contributor', $caps['tnc_col_%d_edit_published_items']['roles']);




	}

}

?>