restful/tests/Runner_Resolve_Auth_Test.php at a693f3a4190b8ec232808f41d6e697e1c1552463 · wp-cli/restful · GitHub

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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
<?php

use WP_CLI\Tests\TestCase;
use PHPUnit\Framework\Attributes\DataProvider;

class Runner_Resolve_Auth_Test extends TestCase {

	private $saved_env = array();

	public function set_up() {
		foreach ( array( 'WP_REST_CLI_AUTH_USER', 'WP_REST_CLI_AUTH_PASSWORD' ) as $var ) {
			$val                     = getenv( $var );
			$this->saved_env[ $var ] = false === $val ? false : $val;
			putenv( $var );
		}
	}

	public function tear_down() {
		foreach ( $this->saved_env as $var => $val ) {
			if ( false === $val ) {
				putenv( $var );
			} else {
				putenv( "{$var}={$val}" );
			}
		}
	}

	public function test_no_auth_when_nothing_set() {
		$auth = \WP_REST_CLI\Runner::resolve_auth( 'example.com' );
		$this->assertSame( array(), $auth );
	}

	public function test_auth_from_config() {
		$auth = \WP_REST_CLI\Runner::resolve_auth(
			'example.com',
			array(
				'http_user'     => 'admin',
				'http_password' => 'secret',
			)
		);
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'admin',
				'password' => 'secret',
			),
			$auth
		);
	}

	public function test_config_allows_empty_password() {
		$auth = \WP_REST_CLI\Runner::resolve_auth(
			'example.com',
			array( 'http_user' => 'admin' )
		);
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'admin',
				'password' => '',
			),
			$auth
		);
	}

	public function test_env_vars_override_config() {
		putenv( 'WP_REST_CLI_AUTH_USER=envuser' );
		putenv( 'WP_REST_CLI_AUTH_PASSWORD=envpass' );
		$auth = \WP_REST_CLI\Runner::resolve_auth(
			'example.com',
			array(
				'http_user'     => 'cfguser',
				'http_password' => 'cfgpass',
			)
		);
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'envuser',
				'password' => 'envpass',
			),
			$auth
		);
	}

	public function test_env_user_without_env_password_uses_empty_password() {
		putenv( 'WP_REST_CLI_AUTH_USER=envuser' );
		$auth = \WP_REST_CLI\Runner::resolve_auth( 'example.com' );
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'envuser',
				'password' => '',
			),
			$auth
		);
	}

	public function test_empty_env_user_skips_env_auth() {
		putenv( 'WP_REST_CLI_AUTH_USER=' );
		putenv( 'WP_REST_CLI_AUTH_PASSWORD=envpass' );
		$auth = \WP_REST_CLI\Runner::resolve_auth( 'example.com' );
		$this->assertSame( array(), $auth );
	}

	public function test_url_credentials_override_env_vars() {
		putenv( 'WP_REST_CLI_AUTH_USER=envuser' );
		putenv( 'WP_REST_CLI_AUTH_PASSWORD=envpass' );
		$auth = \WP_REST_CLI\Runner::resolve_auth( 'http://urluser:urlpass@example.com' );
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'urluser',
				'password' => 'urlpass',
			),
			$auth
		);
	}

	/**
	 * @dataProvider provide_url_credentials
	 */
	#[DataProvider( 'provide_url_credentials' )] // phpcs:ignore PHPCompatibility.Attributes.NewAttributes.PHPUnitAttributeFound
	public function test_url_credentials( $url, $expected_user, $expected_pass ) {
		$auth = \WP_REST_CLI\Runner::resolve_auth( $url );
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => $expected_user,
				'password' => $expected_pass,
			),
			$auth
		);
	}

	public static function provide_url_credentials() {
		return array(
			'no scheme'        => array( 'urluser:urlpass@example.com', 'urluser', 'urlpass' ),
			'https scheme'     => array( 'https://urluser:urlpass@example.com', 'urluser', 'urlpass' ),
			'user only'        => array( 'urluser@example.com', 'urluser', '' ),
			'user only, https' => array( 'https://urluser@example.com', 'urluser', '' ),
		);
	}

	public function test_url_credentials_override_config() {
		$auth = \WP_REST_CLI\Runner::resolve_auth(
			'http://urluser:urlpass@example.com',
			array(
				'http_user'     => 'cfguser',
				'http_password' => 'cfgpass',
			)
		);
		$this->assertSame(
			array(
				'type'     => 'basic',
				'username' => 'urluser',
				'password' => 'urlpass',
			),
			$auth
		);
	}
}