1 /*
2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23 /*
24 * @test
25 * @bug 6921664
26 * @summary Verifies number of copies and the job name are passed to a
27 * 3rd party PrintService.
28 * @run main DummyPrintTest
29 */
30 import java.awt.Graphics;
31 import java.awt.print.PageFormat;
32 import java.awt.print.Printable;
33 import java.awt.print.PrinterException;
34 import java.awt.print.PrinterJob;
35 import java.util.HashSet;
36 import java.util.Set;
37 import javax.print.Doc;
38 import javax.print.DocFlavor;
39 import javax.print.DocPrintJob;
40 import javax.print.PrintException;
41 import javax.print.PrintService;
42 import javax.print.PrintServiceLookup;
43 import javax.print.ServiceUIFactory;
44 import javax.print.attribute.Attribute;
45 import javax.print.attribute.AttributeSet;
46 import javax.print.attribute.HashPrintJobAttributeSet;
47 import javax.print.attribute.HashPrintServiceAttributeSet;
48 import javax.print.attribute.PrintJobAttributeSet;
49 import javax.print.attribute.PrintRequestAttributeSet;
50 import javax.print.attribute.PrintServiceAttribute;
51 import javax.print.attribute.PrintServiceAttributeSet;
52 import javax.print.attribute.standard.Copies;
53 import javax.print.attribute.standard.JobName;
54 import javax.print.attribute.standard.PrinterName;
55 import javax.print.attribute.standard.PrinterState;
56 import javax.print.event.PrintJobAttributeListener;
57 import javax.print.event.PrintJobListener;
58 import javax.print.event.PrintServiceAttributeListener;
59
60
61 public class DummyPrintTest {
62
63 public static void main(String[] args) throws Exception {
64 // register custom print service implementation
65 String printerName = "myDummyPrintService";
66 PrintServiceLookup.registerService(new DummyPrintService(printerName));
67 // calling third party print logic
68 thirdPartyPrintLogic(printerName);
69 }
70
71 static void thirdPartyPrintLogic(String printerName) throws Exception {
72 PrinterJob printerjob = PrinterJob.getPrinterJob();
73 printerjob.setCopies(2);
74 printerjob.setJobName("myJobName");
75 printerjob.setPrintable(new DummyPrintable());
76 for (PrintService printService : PrinterJob.lookupPrintServices()) {
77 System.out.println("check printer name of service " + printService);
78 if (printerName.equals(printService.getName())) {
79 System.out.println("correct printer service do print...");
80 printerjob.setPrintService(printService);
81 printerjob.print();
82 break;
83 }
84 }
85 }
86 }
87
88 class DummyPrintService implements PrintService {
89 private final String _name;
90 private final Set<DocFlavor> _supportedFlavors;
91 private final PrintServiceAttributeSet _printServiceAttributeSet;
92
93 public DummyPrintService(String name) {
94 _name = name;
95 _supportedFlavors = new HashSet<DocFlavor>();
96 _supportedFlavors.add(DocFlavor.SERVICE_FORMATTED.PAGEABLE);
97 _supportedFlavors.add(DocFlavor.SERVICE_FORMATTED.PRINTABLE);
98 _printServiceAttributeSet = new HashPrintServiceAttributeSet();
99 _printServiceAttributeSet.add(new PrinterName(name, null));
100 _printServiceAttributeSet.add(PrinterState.IDLE);
101 }
102
103 @Override
104 public String toString() {
105 return "Dummy Printer : " + getName();
106 }
107
108 @Override
109 public String getName() {
110 return _name;
111 }
112
113 @Override
114 public DocPrintJob createPrintJob() {
115 return new DummyDocPrintJob(this);
116 }
117
118 @Override
119 public boolean isDocFlavorSupported(DocFlavor flavor) {
120 return _supportedFlavors.contains(flavor);
121 }
122
123 @Override
124 public <T extends PrintServiceAttribute> T getAttribute(Class<T> category) {
125 return category.cast(_printServiceAttributeSet.get(category));
126 }
127
128 @Override
129 public PrintServiceAttributeSet getAttributes() {
130 return _printServiceAttributeSet;
131 }
132
133 @Override
134 public DocFlavor[] getSupportedDocFlavors() {
135 return _supportedFlavors.toArray(new DocFlavor[_supportedFlavors.size()]);
136 }
137
138 @Override
139 public Object getDefaultAttributeValue(Class<? extends Attribute> category) {
140 return null;
141 }
142
143 @Override
144 public ServiceUIFactory getServiceUIFactory() {
145 return null;
146 }
147
148 @Override
149 public Class<?>[] getSupportedAttributeCategories() {
150 return null;
151 }
152
153 @Override
154 public Object getSupportedAttributeValues(Class<? extends Attribute> category,
155 DocFlavor flavor, AttributeSet attributes) {
156 return null;
157 }
158
159 @Override
160 public AttributeSet getUnsupportedAttributes(DocFlavor flavor,
161 AttributeSet attributes) {
162 return null;
163 }
164
165 @Override
166 public boolean isAttributeCategorySupported(Class<? extends Attribute> category) {
167 return false;
168 }
169
170 @Override
171 public boolean isAttributeValueSupported(Attribute attrval,
172 DocFlavor flavor,
173 AttributeSet attributes) {
174 return false;
175 }
176
177 @Override
178 public void addPrintServiceAttributeListener(PrintServiceAttributeListener listener) {
179 }
180
181 @Override
182 public void removePrintServiceAttributeListener(PrintServiceAttributeListener listener) {
183 }
184 }
185
186 class DummyDocPrintJob implements DocPrintJob {
187 private static int _counter;
188 private final PrintService _printService;
189 private final PrintJobAttributeSet _printJobAttributeSet;
190
191 public DummyDocPrintJob(PrintService printService) {
192 _counter++;
193 _printService = printService;
194 _printJobAttributeSet = new HashPrintJobAttributeSet();
195 }
196
197 @Override
198 public PrintService getPrintService() {
199 return _printService;
200 }
201
202 @Override
203 public PrintJobAttributeSet getAttributes() {
204 return _printJobAttributeSet;
205 }
206
207 @Override
208 public void addPrintJobAttributeListener(PrintJobAttributeListener listener,
209 PrintJobAttributeSet printJobAttributeSet) {
210 }
211
212 @Override
213 public void removePrintJobAttributeListener(PrintJobAttributeListener listener) {
214 }
215
216 @Override
217 public void addPrintJobListener(PrintJobListener listener) {
218 }
219
220 @Override
221 public void removePrintJobListener(PrintJobListener listener) {
222 }
223
224 @Override
225 public void print(Doc doc,
226 PrintRequestAttributeSet printRequestAttributeSet)
227 throws PrintException {
228 System.out.println("job name: " + printRequestAttributeSet.get(JobName.class));
229 System.out.println("copies: " + printRequestAttributeSet.get(Copies.class));
230 if(printRequestAttributeSet.get(JobName.class) == null ||
231 printRequestAttributeSet.get(Copies.class) == null) {
232 throw new RuntimeException("Copies and JobName is not passed correctly");
233 }
234 }
235 }
236
237 class DummyPrintable implements Printable {
238 @Override
239 public int print(Graphics graphics, PageFormat pageFormat, int pageIndex)
240 throws PrinterException {
241 if (pageIndex == 0) {
242 return Printable.PAGE_EXISTS;
243 } else {
244 return Printable.NO_SUCH_PAGE;
245 }
246 }
247 }