60 lines
1.6 KiB
PHP
60 lines
1.6 KiB
PHP
<?php
|
|
/**
|
|
* @package WooCommerce\Tests\PaymentGateways
|
|
*/
|
|
|
|
/**
|
|
* Class WC_Tests_Payment_Gateway.
|
|
*/
|
|
class WC_Tests_Payment_Gateway extends WC_Unit_Test_Case {
|
|
|
|
/**
|
|
* Setup, enable payment gateways Cash on delivery and direct bank deposit.
|
|
*/
|
|
public function setUp() {
|
|
parent::setUp();
|
|
WC()->session = null;
|
|
$wc_payment_gateways = WC_Payment_Gateways::instance();
|
|
$wc_payment_gateways->init();
|
|
foreach ( $wc_payment_gateways->payment_gateways() as $name => $gateway ) {
|
|
if ( in_array( $name, array( 'cod', 'bacs' ) ) ) {
|
|
$gateway->enabled = 'yes';
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Initialize session that some tests might have removed.
|
|
*/
|
|
public function tearDown() {
|
|
parent::tearDown();
|
|
WC()->initialize_session();
|
|
}
|
|
|
|
/**
|
|
* Test that we can set a current gateway from session. Basic smoke test.
|
|
*/
|
|
public function test_wc_set_current_gateway_from_session() {
|
|
WC()->initialize_session();
|
|
wp_set_current_user( 1 );
|
|
|
|
$gateways = WC()->payment_gateways()->get_available_payment_gateways();
|
|
$gateways['bacs']->chosen = false;
|
|
WC()->session->set( 'chosen_payment_method', 'bacs' );
|
|
WC()->payment_gateways()->set_current_gateway( $gateways );
|
|
$this->assertTrue( $gateways['bacs']->chosen );
|
|
}
|
|
|
|
/**
|
|
* Test that we can set a current gateway without session.
|
|
*/
|
|
public function test_wc_set_current_gateway_without_session() {
|
|
$gateways = WC()->payment_gateways()->get_available_payment_gateways();
|
|
$current_gateway = current( $gateways );
|
|
$current_gateway->chosen = false;
|
|
WC()->payment_gateways()->set_current_gateway( $gateways );
|
|
$this->assertTrue( $current_gateway->chosen );
|
|
}
|
|
|
|
}
|