1 use Test::More tests => '70';
4 use_ok('RT::ScripConditions');
5 use_ok('RT::ScripActions');
6 use_ok('RT::Template');
12 use File::Temp qw/tempfile/;
13 my ($fh, $filename) = tempfile( UNLINK => 1, SUFFIX => '.rt');
14 my $link_scrips_orig = $RT::LinkTransactionsRun1Scrip;
15 my $link_acl_chacks_orig = $RT::StrictLinkACL;
16 $RT::LinkTransactionsRun1Scrip = 1;
17 $RT::StrictLinkACL = 1;
19 my $condition = RT::ScripCondition->new( $RT::SystemUser );
20 $condition->Load('User Defined');
22 my $action = RT::ScripAction->new( $RT::SystemUser );
23 $action->Load('User Defined');
25 my $template = RT::Template->new( $RT::SystemUser );
26 $template->Load('Blank');
29 my $q1 = RT::Queue->new($RT::SystemUser);
30 my ($id,$msg) = $q1->Create(Name => "LinkTest1.$$");
32 my $q2 = RT::Queue->new($RT::SystemUser);
33 ($id,$msg) = $q2->Create(Name => "LinkTest2.$$");
36 my $commit_code = <<END;
37 open(FILE, "<$filename");
41 open(FILE, ">$filename");
42 if (\$self->TransactionObj->Type eq 'AddLink') {
43 print FILE \$data+1, "\n";
46 print FILE \$data-1, "\n";
52 my $Scrips = RT::Scrips->new( $RT::SystemUser );
54 while ( my $Scrip = $Scrips->Next ) {
55 $Scrip->Delete if $Scrip->Description =~ /Add or Delete Link \d+/;
59 my $scrip = RT::Scrip->new($RT::SystemUser);
60 ($id,$msg) = $scrip->Create( Description => "Add or Delete Link $$",
61 ScripCondition => $condition->id,
62 ScripAction => $action->id,
63 Template => $template->id,
64 Stage => 'TransactionCreate',
66 CustomIsApplicableCode => '$self->TransactionObj->Type =~ /(Add|Delete)Link/;',
67 CustomPrepareCode => '1;',
68 CustomCommitCode => $commit_code,
70 ok($id, "Scrip created");
72 my $u1 = RT::User->new($RT::SystemUser);
73 ($id,$msg) = $u1->Create(Name => "LinkTestUser.$$");
76 my $creator = RT::CurrentUser->new($u1->id);
78 ($id,$msg) = $u1->PrincipalObj->GrantRight ( Object => $q1, Right => 'CreateTicket');
81 diag('Create tickets without rights to link') if $ENV{'TEST_VERBOSE'};
83 # on q2 we have no rights, yet
84 my $parent = RT::Ticket->new( $RT::SystemUser );
85 ($id,$tid,$msg) = $parent->Create( Subject => 'Link test 1', Queue => $q2->id );
87 my $child = RT::Ticket->new( $creator );
88 ($id,$tid,$msg) = $child->Create( Subject => 'Link test 1', Queue => $q1->id, MemberOf => $parent->id );
90 $child->CurrentUser( $RT::SystemUser );
91 is($child->_Links('Base')->Count, 0, 'link was not created, no permissions');
92 is($child->_Links('Target')->Count, 0, 'link was not create, no permissions');
95 diag('Create tickets with rights checks on one end of a link') if $ENV{'TEST_VERBOSE'};
97 # on q2 we have no rights, but use checking one only on thing
98 local $RT::StrictLinkACL = 0;
99 my $parent = RT::Ticket->new( $RT::SystemUser );
100 ($id,$tid,$msg) = $parent->Create( Subject => 'Link test 1', Queue => $q2->id );
102 my $child = RT::Ticket->new( $creator );
103 ($id,$tid,$msg) = $child->Create( Subject => 'Link test 1', Queue => $q1->id, MemberOf => $parent->id );
105 $child->CurrentUser( $RT::SystemUser );
106 is($child->_Links('Base')->Count, 1, 'link was created');
107 is($child->_Links('Target')->Count, 0, 'link was created only one');
108 # no scrip run on second ticket accroding to config option
109 is(link_count($filename), 0, "scrips ok");
112 ($id,$msg) = $u1->PrincipalObj->GrantRight ( Object => $q1, Right => 'ModifyTicket');
115 diag('try to add link without rights') if $ENV{'TEST_VERBOSE'};
117 # on q2 we have no rights, yet
118 my $parent = RT::Ticket->new( $RT::SystemUser );
119 ($id,$tid,$msg) = $parent->Create( Subject => 'Link test 1', Queue => $q2->id );
121 my $child = RT::Ticket->new( $creator );
122 ($id,$tid,$msg) = $child->Create( Subject => 'Link test 1', Queue => $q1->id );
124 my ($id, $msg) = $child->AddLink(Type => 'MemberOf', Target => $parent->id);
126 is(link_count($filename), 0, "scrips ok");
127 $child->CurrentUser( $RT::SystemUser );
128 is($child->_Links('Base')->Count, 0, 'link was not created, no permissions');
129 is($child->_Links('Target')->Count, 0, 'link was not create, no permissions');
132 diag('add link with rights only on base') if $ENV{'TEST_VERBOSE'};
134 # on q2 we have no rights, but use checking one only on thing
135 local $RT::StrictLinkACL = 0;
136 my $parent = RT::Ticket->new( $RT::SystemUser );
137 ($id,$tid,$msg) = $parent->Create( Subject => 'Link test 1', Queue => $q2->id );
139 my $child = RT::Ticket->new( $creator );
140 ($id,$tid,$msg) = $child->Create( Subject => 'Link test 1', Queue => $q1->id );
142 my ($id, $msg) = $child->AddLink(Type => 'MemberOf', Target => $parent->id);
144 is(link_count($filename), 1, "scrips ok");
145 $child->CurrentUser( $RT::SystemUser );
146 is($child->_Links('Base')->Count, 1, 'link was created');
147 is($child->_Links('Target')->Count, 0, 'link was created only one');
148 $child->CurrentUser( $creator );
150 # turn off feature and try to delete link, we should fail
151 $RT::StrictLinkACL = 1;
152 my ($id, $msg) = $child->AddLink(Type => 'MemberOf', Target => $parent->id);
154 is(link_count($filename), 1, "scrips ok");
155 $child->CurrentUser( $RT::SystemUser );
156 $child->_Links('Base')->_DoCount;
157 is($child->_Links('Base')->Count, 1, 'link was not deleted');
158 $child->CurrentUser( $creator );
160 # try to delete link, we should success as feature is active
161 $RT::StrictLinkACL = 0;
162 my ($id, $msg) = $child->DeleteLink(Type => 'MemberOf', Target => $parent->id);
164 is(link_count($filename), 0, "scrips ok");
165 $child->CurrentUser( $RT::SystemUser );
166 $child->_Links('Base')->_DoCount;
167 is($child->_Links('Base')->Count, 0, 'link was deleted');
171 my $ticket = RT::Ticket->new( $creator);
172 ok($ticket->isa('RT::Ticket'));
173 ($id,$tid, $msg) = $ticket->Create(Subject => 'Link test 1', Queue => $q1->id);
176 diag('try link to itself') if $ENV{'TEST_VERBOSE'};
178 my ($id, $msg) = $ticket->AddLink(Type => 'RefersTo', Target => $ticket->id);
180 is(link_count($filename), 0, "scrips ok");
183 my $ticket2 = RT::Ticket->new($RT::SystemUser);
184 ($id, $tid, $msg) = $ticket2->Create(Subject => 'Link test 2', Queue => $q2->id);
186 ($id,$msg) =$ticket->AddLink(Type => 'RefersTo', Target => $ticket2->id);
188 ok(link_count($filename) == 0, "scrips ok");
190 ($id,$msg) = $u1->PrincipalObj->GrantRight ( Object => $q2, Right => 'CreateTicket');
192 ($id,$msg) = $u1->PrincipalObj->GrantRight ( Object => $q2, Right => 'ModifyTicket');
194 ($id,$msg) =$ticket->AddLink(Type => 'RefersTo', Target => $ticket2->id);
196 ok(link_count($filename) == 1, "scrips ok");
197 ($id,$msg) =$ticket->AddLink(Type => 'RefersTo', Target => -1);
199 ok(link_count($filename) == 1, "scrips ok");
200 ($id,$msg) = $ticket->AddLink(Type => 'RefersTo', Target => $ticket2->id);
202 is(link_count($filename), 1, "scrips ok");
204 my $transactions = $ticket2->Transactions;
205 $transactions->Limit( FIELD => 'Type', VALUE => 'AddLink' );
206 ok( $transactions->Count == 1, "Transaction found in other ticket" );
207 ok( $transactions->First->Field eq 'ReferredToBy');
208 ok( $transactions->First->NewValue eq $ticket->URI );
210 ($id,$msg) =$ticket->DeleteLink(Type => 'RefersTo', Target => $ticket2->id);
212 ok(link_count($filename) == 0, "scrips ok");
213 $transactions = $ticket2->Transactions;
214 $transactions->Limit( FIELD => 'Type', VALUE => 'DeleteLink' );
215 ok( $transactions->Count == 1, "Transaction found in other ticket" );
216 ok( $transactions->First->Field eq 'ReferredToBy');
217 ok( $transactions->First->OldValue eq $ticket->URI );
219 $RT::LinkTransactionsRun1Scrip = 0;
221 ($id,$msg) =$ticket->AddLink(Type => 'RefersTo', Target => $ticket2->id);
223 ok(link_count($filename) == 2, "scrips ok");
224 ($id,$msg) =$ticket->DeleteLink(Type => 'RefersTo', Target => $ticket2->id);
226 ok(link_count($filename) == 0, "scrips ok");
229 $RT::LinkTransactionsRun1Scrip = $link_scrips_orig;
230 $RT::StrictLinkACL = $link_acl_checks_orig;
237 open(FILE, "<$file");