Better tests for global tax rate and country/state tax rate. #600

awesomemotive · Sep 5, 2013 · 8ff0260 · 8ff0260
1 parent 0778386
commit 8ff0260
Showing 1 changed file with 32 additions and 3 deletions.
diff --git a/tests/unit-tests/tax.php b/tests/unit-tests/tax.php
@@ -27,11 +27,40 @@ public function test_taxes_after_discounts() {
 
 	public function test_get_tax_rate() {
 		global $edd_options;
+
+		// Setup global tax rate
 		$options = array();
 		$options['tax_rate'] = '3.6';
-		update_option( 'edd_options', array_merge( $options, $edd_options ) );
-		$this->assertInternalType( 'float', edd_get_tax_rate() );
-		$this->assertEquals( '0.036', edd_get_tax_rate() );
+		$edd_options = array_merge( $options, $edd_options );
+
+		// Setup country / state tax rates
+		$tax_rates = array();
+		$tax_rates[] = array( 'country' => 'US', 'state' => 'AL', 'rate' => 15 );
+
+		update_option( 'edd_options', $edd_options );
+		update_option( 'edd_tax_rates', $tax_rates );
+
+		$this->assertInternalType( 'float', edd_get_tax_rate( 'US', 'AL' ) );
+		$this->assertEquals( '0.15', edd_get_tax_rate( 'US', 'AL' ) );
+	}
+
+	public function test_get_global_tax_rate() {
+		global $edd_options;
+
+		// Setup global tax rate
+		$options = array();
+		$options['tax_rate'] = '3.6';
+		$edd_options = array_merge( $options, $edd_options );
+
+		// Setup country / state tax rates
+		$tax_rates = array();
+		$tax_rates[] = array( 'country' => 'US', 'state' => 'AL', 'rate' => 15 );
+
+		update_option( 'edd_options', $edd_options );
+		update_option( 'edd_tax_rates', $tax_rates );
+
+		$this->assertInternalType( 'float', edd_get_tax_rate( 'CA', 'AB' ) );
+		$this->assertEquals( '0.036', edd_get_tax_rate( 'CA', 'AB' ) );
 	}
 
 	public function test_calculate_tax() {