@@ -25,69 +25,75 @@ describe('features/classification/security-controls/utils', () => {
2525
2626 describe ( 'getShortSecurityControlsMessage()' , ( ) => {
2727 test ( 'should return null when there are no restrictions' , ( ) => {
28- expect ( getShortSecurityControlsMessage ( { } ) ) . toBeNull ( ) ;
28+ expect ( getShortSecurityControlsMessage ( { } ) ) . toEqual ( [ ] ) ;
2929 } ) ;
3030
3131 test ( 'should not return messages when shared link restriction has a "public" access level' , ( ) => {
3232 accessPolicy = { sharedLink : { accessLevel : PUBLIC } } ;
33- expect ( getShortSecurityControlsMessage ( accessPolicy ) ) . toBeNull ( ) ;
33+ expect ( getShortSecurityControlsMessage ( accessPolicy ) ) . toEqual ( [ ] ) ;
3434 } ) ;
3535
36- test ( 'should return all restrictions message when all restrictions are present' , ( ) => {
37- accessPolicy = { sharedLink : { } , download : { } , externalCollab : { } , app : { } } ;
38- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortAllRestrictions ) ;
36+ test ( 'should return correct message when all restrictions are present' , ( ) => {
37+ accessPolicy = { sharedLink : { } , download : { } , externalCollab : { } , app : { } , watermark : { } } ;
38+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortAllRestrictions ) ;
39+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 1 ] . message ) . toBe ( messages . shortWatermarking ) ;
3940 } ) ;
4041
4142 test ( 'should return all restrictions message when download, app and either shared link, or external collab restrictions are present' , ( ) => {
4243 accessPolicy = { sharedLink : { } , download : { } , app : { } } ;
43- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortAllRestrictions ) ;
44+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortAllRestrictions ) ;
4445 accessPolicy = { externalCollab : { } , download : { } , app : { } } ;
45- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortAllRestrictions ) ;
46+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortAllRestrictions ) ;
4647 } ) ;
4748
4849 test ( 'should return correct message when download and either shared link, or external collab restrictions are present' , ( ) => {
4950 accessPolicy = { sharedLink : { } , download : { } } ;
50- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharingDownload ) ;
51+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharingDownload ) ;
5152 accessPolicy = { externalCollab : { } , download : { } } ;
52- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharingDownload ) ;
53+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharingDownload ) ;
5354 } ) ;
5455
5556 test ( 'should return correct message when app and either shared link, or external collab restrictions are present' , ( ) => {
5657 accessPolicy = { sharedLink : { } , app : { } } ;
57- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharingApp ) ;
58+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharingApp ) ;
5859 accessPolicy = { externalCollab : { } , app : { } } ;
59- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharingApp ) ;
60+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharingApp ) ;
6061 } ) ;
6162
6263 test ( 'should return correct message when app and download restrictions are present' , ( ) => {
6364 accessPolicy = { download : { } , app : { } } ;
64- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortDownloadApp ) ;
65+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortDownloadApp ) ;
6566 } ) ;
6667
6768 test ( 'should return correct message when there are shared link or external collab restrictions' , ( ) => {
6869 accessPolicy = { sharedLink : { } , externalCollab : { } } ;
69- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharing ) ;
70+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharing ) ;
7071
7172 accessPolicy = { sharedLink : { } } ;
72- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharing ) ;
73+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharing ) ;
7374
7475 accessPolicy = { externalCollab : { } } ;
75- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortSharing ) ;
76+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortSharing ) ;
7677 } ) ;
7778
7879 test ( 'should return correct message when there is a download restriction' , ( ) => {
7980 accessPolicy = { download : { } } ;
80- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortDownload ) ;
81+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortDownload ) ;
8182 } ) ;
8283
8384 test ( 'should return correct message when there is a download restriction' , ( ) => {
8485 accessPolicy = { app : { } } ;
85- expect ( getShortSecurityControlsMessage ( accessPolicy ) . message ) . toBe ( messages . shortApp ) ;
86+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortApp ) ;
87+ } ) ;
88+
89+ test ( 'should return correct message when there is a watermark restriction' , ( ) => {
90+ accessPolicy = { watermark : { } } ;
91+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . message ) . toBe ( messages . shortWatermarking ) ;
8692 } ) ;
8793
8894 test ( 'should not return tooltipMessage' , ( ) => {
8995 accessPolicy = { sharedLink : { } , download : { } , externalCollab : { } , app : { } } ;
90- expect ( getShortSecurityControlsMessage ( accessPolicy ) . tooltipMessage ) . toBeUndefined ( ) ;
96+ expect ( getShortSecurityControlsMessage ( accessPolicy ) [ 0 ] . tooltipMessage ) . toBeUndefined ( ) ;
9197 } ) ;
9298 } ) ;
9399
@@ -112,6 +118,15 @@ describe('features/classification/security-controls/utils', () => {
112118 ] ) ;
113119 } ) ;
114120
121+ test ( 'should include correct message when watermark is applied' , ( ) => {
122+ accessPolicy = {
123+ watermark : {
124+ enabled : true ,
125+ } ,
126+ } ;
127+ expect ( getFullSecurityControlsMessages ( accessPolicy ) ) . toEqual ( [ { message : messages . watermarkingApplied } ] ) ;
128+ } ) ;
129+
115130 test ( 'should include correct message when external collab is blocked' , ( ) => {
116131 accessPolicy = {
117132 externalCollab : {
0 commit comments