/protocols/ss7/map/map-impl/src/test/java/org/mobicents/protocols/ss7/map/functional/MAPProviderImplWrapper.java

 1/*
 2 * JBoss, Home of Professional Open Source
 3 * Copyright 2011, Red Hat, Inc. and individual contributors
 4 * by the @authors tag. See the copyright.txt in the distribution for a
 5 * full listing of individual contributors.
 6 *
 7 * This is free software; you can redistribute it and/or modify it
 8 * under the terms of the GNU Lesser General Public License as
 9 * published by the Free Software Foundation; either version 2.1 of
10 * the License, or (at your option) any later version.
11 *
12 * This software is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this software; if not, write to the Free
19 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21 */
22
23package org.mobicents.protocols.ss7.map.functional;
24
25import org.mobicents.protocols.ss7.map.api.MAPServiceBase;
26import org.mobicents.protocols.ss7.map.api.MAPException;
27import org.mobicents.protocols.ss7.map.api.dialog.MAPProviderAbortReason;
28import org.mobicents.protocols.ss7.map.MAPProviderImpl;
29import org.mobicents.protocols.ss7.map.api.service.supplementary.MAPServiceSupplementary;
30import org.mobicents.protocols.ss7.map.primitives.MAPExtensionContainerTest;
31import org.mobicents.protocols.ss7.tcap.api.TCAPProvider;
32import org.mobicents.protocols.ss7.tcap.api.tc.dialog.events.TCBeginIndication;
33import org.mobicents.protocols.ss7.tcap.asn.ApplicationContextName;
34import org.mobicents.protocols.ss7.tcap.asn.comp.Component;
35
36
37/**
38 * 
39 * @author amit bhayani
40 * @author sergey vetyutnev
41 * 
42 */
43public class MAPProviderImplWrapper extends MAPProviderImpl {
44
45	private int testMode = 0;
46
47	private final MAPServiceSupplementary mapServiceSupplementaryTest = new MAPServiceSupplementaryImplWrapper(this);
48
49	public MAPProviderImplWrapper(TCAPProvider tcapProvider) {
50		super(tcapProvider);
51		
52		for(MAPServiceBase serv : this.mapServices) {
53			if( serv instanceof MAPServiceSupplementary ) {
54				this.mapServices.remove(serv);
55				break;
56			}
57		}
58		
59		this.mapServices.add(this.mapServiceSupplementaryTest);
60	}
61
62	public MAPServiceSupplementary getMAPServiceSupplementary() {
63		return this.mapServiceSupplementaryTest;
64	}
65
66	public void setTestMode(int testMode) {
67		this.testMode = testMode;
68	}
69
70	public void onTCBegin(TCBeginIndication tcBeginIndication) {
71		ApplicationContextName acn = tcBeginIndication.getApplicationContextName();
72		Component[] comps = tcBeginIndication.getComponents();
73
74		if (this.testMode == 1) {
75			try {
76				this.fireTCAbortProvider(tcBeginIndication.getDialog(), MAPProviderAbortReason.invalidPDU,
77						MAPExtensionContainerTest.GetTestExtensionContainer());
78			} catch (MAPException e) {
79				loger.error("Error while firing TC-U-ABORT. ", e);
80			}
81			return;
82		}
83
84		super.onTCBegin(tcBeginIndication);
85	}
86}