11from tests .test_helper import *
2+ from braintree .test .credit_card_numbers import CreditCardNumbers
23
34class TestCreditCard (unittest .TestCase ):
45 def test_create_adds_credit_card_to_existing_customer (self ):
@@ -860,11 +861,76 @@ def test_expiring_between(self):
860861
861862 self .assertEquals (set (['2010' ]), TestHelper .unique ([credit_card .expiration_year for credit_card in collection .items ]))
862863
864+ def test_commercial_card (self ):
865+ customer = Customer .create ().customer
866+ result = CreditCard .create ({
867+ "customer_id" : customer .id ,
868+ "number" : CreditCardNumbers .CardTypeIndicators .Commercial ,
869+ "expiration_date" : "05/2014" ,
870+ "options" : {"verify_card" : True }
871+ })
872+
873+ credit_card = result .credit_card
874+
875+ self .assertEquals (CreditCard .Commercial .Yes , credit_card .commercial )
876+
877+ def test_durbin_regulated_card (self ):
878+ customer = Customer .create ().customer
879+ result = CreditCard .create ({
880+ "customer_id" : customer .id ,
881+ "number" : CreditCardNumbers .CardTypeIndicators .DurbinRegulated ,
882+ "expiration_date" : "05/2014" ,
883+ "options" : {"verify_card" : True }
884+ })
885+
886+ credit_card = result .credit_card
887+
888+ self .assertEquals (CreditCard .DurbinRegulated .Yes , credit_card .durbin_regulated )
889+
890+ def test_debit_card (self ):
891+ customer = Customer .create ().customer
892+ result = CreditCard .create ({
893+ "customer_id" : customer .id ,
894+ "number" : CreditCardNumbers .CardTypeIndicators .Debit ,
895+ "expiration_date" : "05/2014" ,
896+ "options" : {"verify_card" : True }
897+ })
898+
899+ credit_card = result .credit_card
900+
901+ self .assertEquals (CreditCard .Debit .Yes , credit_card .debit )
902+
903+ def test_healthcare_card (self ):
904+ customer = Customer .create ().customer
905+ result = CreditCard .create ({
906+ "customer_id" : customer .id ,
907+ "number" : CreditCardNumbers .CardTypeIndicators .Healthcare ,
908+ "expiration_date" : "05/2014" ,
909+ "options" : {"verify_card" : True }
910+ })
911+
912+ credit_card = result .credit_card
913+
914+ self .assertEquals (CreditCard .Healthcare .Yes , credit_card .healthcare )
915+
916+ def test_payroll_card (self ):
917+ customer = Customer .create ().customer
918+ result = CreditCard .create ({
919+ "customer_id" : customer .id ,
920+ "number" : CreditCardNumbers .CardTypeIndicators .Payroll ,
921+ "expiration_date" : "05/2014" ,
922+ "options" : {"verify_card" : True }
923+ })
924+
925+ credit_card = result .credit_card
926+
927+ self .assertEquals (CreditCard .Payroll .Yes , credit_card .payroll )
928+
863929 def test_prepaid_card (self ):
864930 customer = Customer .create ().customer
865931 result = CreditCard .create ({
866932 "customer_id" : customer .id ,
867- "number" : "4500600000000061" ,
933+ "number" : CreditCardNumbers . CardTypeIndicators . Prepaid ,
868934 "expiration_date" : "05/2014" ,
869935 "options" : {"verify_card" : True }
870936 })
@@ -877,25 +943,34 @@ def test_all_negative_card_type_identifiers(self):
877943 customer = Customer .create ().customer
878944 result = CreditCard .create ({
879945 "customer_id" : customer .id ,
880- "number" : "4111111111111111" ,
946+ "number" : CreditCardNumbers . CardTypeIndicators . No ,
881947 "expiration_date" : "05/2014" ,
882948 "options" : {"verify_card" : True }
883949 })
884950
885951 credit_card = result .credit_card
886952
953+ self .assertEquals (CreditCard .Debit .No , credit_card .debit )
954+ self .assertEquals (CreditCard .DurbinRegulated .No , credit_card .durbin_regulated )
887955 self .assertEquals (CreditCard .Prepaid .No , credit_card .prepaid )
956+ self .assertEquals (CreditCard .Payroll .No , credit_card .payroll )
957+ self .assertEquals (CreditCard .Commercial .No , credit_card .commercial )
958+ self .assertEquals (CreditCard .Healthcare .No , credit_card .healthcare )
888959
889960 def test_card_without_card_type_identifiers (self ):
890961 customer = Customer .create ().customer
891962 result = CreditCard .create ({
892963 "customer_id" : customer .id ,
893- "number" : "378282246310005" ,
964+ "number" : CreditCardNumbers . CardTypeIndicators . Unknown ,
894965 "expiration_date" : "05/2014" ,
895966 "options" : {"verify_card" : True }
896967 })
897968
898969 credit_card = result .credit_card
899970
971+ self .assertEquals (CreditCard .Debit .Unknown , credit_card .debit )
972+ self .assertEquals (CreditCard .DurbinRegulated .Unknown , credit_card .durbin_regulated )
900973 self .assertEquals (CreditCard .Prepaid .Unknown , credit_card .prepaid )
901-
974+ self .assertEquals (CreditCard .Payroll .Unknown , credit_card .payroll )
975+ self .assertEquals (CreditCard .Commercial .Unknown , credit_card .commercial )
976+ self .assertEquals (CreditCard .Healthcare .Unknown , credit_card .healthcare )
0 commit comments