@@ -736,9 +736,19 @@ public void ReturnsExpectedUser(Package package, User currentUser, string expect
736736 }
737737 }
738738
739+ public enum SeverityString_State
740+ {
741+ None ,
742+ Low ,
743+ Medium ,
744+ High ,
745+ Critical
746+ }
747+
739748 public static IEnumerable < object [ ] > DeprecationFieldsAreSetAsExpected_Data =
740749 MemberDataHelper . Combine (
741750 MemberDataHelper . FlagEnumDataSet < PackageDeprecationStatus > ( ) ,
751+ MemberDataHelper . EnumDataSet < SeverityString_State > ( ) ,
742752 MemberDataHelper . BooleanDataSet ( ) ,
743753 MemberDataHelper . BooleanDataSet ( ) ,
744754 MemberDataHelper . BooleanDataSet ( ) ,
@@ -748,16 +758,39 @@ public void ReturnsExpectedUser(Package package, User currentUser, string expect
748758 [ MemberData ( nameof ( DeprecationFieldsAreSetAsExpected_Data ) ) ]
749759 public void DeprecationFieldsAreSetAsExpected (
750760 PackageDeprecationStatus status ,
761+ SeverityString_State severity ,
751762 bool hasCves ,
752763 bool hasCwes ,
753764 bool hasAlternateRegistration ,
754765 bool hasAlternatePackage )
755766 {
756767 // Arrange
768+ decimal ? cvss ;
769+ switch ( severity )
770+ {
771+ case SeverityString_State . Critical :
772+ cvss = ( decimal ) 9.5 ;
773+ break ;
774+ case SeverityString_State . High :
775+ cvss = ( decimal ) 7.5 ;
776+ break ;
777+ case SeverityString_State . Medium :
778+ cvss = ( decimal ) 5 ;
779+ break ;
780+ case SeverityString_State . Low :
781+ cvss = ( decimal ) 2 ;
782+ break ;
783+ case SeverityString_State . None :
784+ cvss = null ;
785+ break ;
786+ default :
787+ throw new ArgumentOutOfRangeException ( nameof ( severity ) ) ;
788+ }
789+
757790 var deprecation = new PackageDeprecation
758791 {
759792 Status = status ,
760- CvssRating = ( decimal ) 5.5 ,
793+ CvssRating = cvss ,
761794 CustomMessage = "hello"
762795 } ;
763796
@@ -825,9 +858,32 @@ public void DeprecationFieldsAreSetAsExpected(
825858
826859 // Assert
827860 Assert . Equal ( status , model . DeprecationStatus ) ;
828- Assert . Equal ( deprecation . CvssRating , model . CvssRating ) ;
829861 Assert . Equal ( deprecation . CustomMessage , model . CustomMessage ) ;
830862
863+ string expectedString ;
864+ switch ( severity )
865+ {
866+ case SeverityString_State . Critical :
867+ expectedString = "Critical" ;
868+ break ;
869+ case SeverityString_State . High :
870+ expectedString = "High" ;
871+ break ;
872+ case SeverityString_State . Medium :
873+ expectedString = "Medium" ;
874+ break ;
875+ case SeverityString_State . Low :
876+ expectedString = "Low" ;
877+ break ;
878+ case SeverityString_State . None :
879+ expectedString = null ;
880+ break ;
881+ default :
882+ throw new ArgumentOutOfRangeException ( nameof ( severity ) ) ;
883+ }
884+
885+ Assert . Equal ( expectedString , model . Severity ) ;
886+
831887 if ( hasCves )
832888 {
833889 Assert . Equal ( cveIds , model . CveIds ) ;
0 commit comments