Automattic / jetpack

	/**

	 * Test attach_stored_licenses_on_connection() for the master user.

	 */

	public function test_attach_stored_licenses_on_connection__master_user() {

		$connection = $this->createMock( Connection_Manager::class );

		$connection->method( 'is_connection_owner' )->willReturn( true );

		$licensing = $this->createPartialMock(

			Licensing::class,

			array( 'connection', 'attach_stored_licenses' )

		);

		$licensing->method( 'connection' )->willReturn( $connection );

		$licensing->expects( $this->once() )

			->method( 'attach_stored_licenses' );

		$licensing->attach_stored_licenses_on_connection();

	}

	/**

	 * Test attach_stored_licenses_on_connection() for a secondary user.

	/**

	 * Test attach_stored_licenses_on_connection() for a secondary user.

	 */

	public function test_attach_stored_licenses_on_connection__secondary_user() {

		$connection = $this->createMock( Connection_Manager::class );

		$connection->method( 'is_connection_owner' )->willReturn( false );

		$licensing = $this->createPartialMock(

			Licensing::class,

			array( 'connection', 'attach_stored_licenses' )

		);

		$licensing->method( 'connection' )->willReturn( $connection );

		$licensing->expects( $this->never() )

			->method( 'attach_stored_licenses' );

		$licensing->attach_stored_licenses_on_connection();

	}

}