@@ -215,15 +215,15 @@ fn update_add_gitlab() {
215215#[ test]
216216fn update_add_azure_devops ( ) {
217217 // Add a azure devops badge
218- let ( conn , krate, test_badges) = set_up ( ) ;
218+ let ( krate, test_badges) = set_up ( ) ;
219219
220220 let mut badges = HashMap :: new ( ) ;
221221 badges. insert (
222222 String :: from ( "azure-devops" ) ,
223223 test_badges. azure_devops_attributes ,
224224 ) ;
225- Badge :: update_crate ( & conn , & krate, Some ( & badges) ) . unwrap ( ) ;
226- assert_eq ! ( krate. badges( & conn ) . unwrap ( ) , vec![ test_badges. azure_devops] ) ;
225+ krate. update ( & badges) ;
226+ assert_eq ! ( krate. badges( ) , vec![ test_badges. azure_devops] ) ;
227227}
228228
229229#[ test]
@@ -429,7 +429,7 @@ fn gitlab_required_keys() {
429429#[ test]
430430fn azure_devops_required_keys ( ) {
431431 // Add a azure devops badge missing a required field
432- let ( conn , krate, mut test_badges) = set_up ( ) ;
432+ let ( krate, mut test_badges) = set_up ( ) ;
433433
434434 let mut badges = HashMap :: new ( ) ;
435435
@@ -440,10 +440,10 @@ fn azure_devops_required_keys() {
440440 test_badges. azure_devops_attributes ,
441441 ) ;
442442
443- let invalid_badges = Badge :: update_crate ( & conn , & krate, Some ( & badges) ) . unwrap ( ) ;
443+ let invalid_badges = krate. update ( & badges) ;
444444 assert_eq ! ( invalid_badges. len( ) , 1 ) ;
445- assert ! ( invalid_badges. contains ( & "azure-devops" ) ) ;
446- assert_eq ! ( krate. badges( & conn ) . unwrap ( ) , vec![ ] ) ;
445+ assert_eq ! ( invalid_badges. first ( ) . unwrap ( ) , "azure-devops" ) ;
446+ assert_eq ! ( krate. badges( ) , vec![ ] ) ;
447447}
448448
449449#[ test]
0 commit comments