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

http://mobicents.googlecode.com/ · Java · 86 lines · 45 code · 14 blank · 27 comment · 4 complexity · d22dcb6bc2732774eb94fa10291b202e MD5 · raw file 

    

  1. /*

    2. 

  2.  * JBoss, Home of Professional Open Source

    3. 

  3.  * Copyright 2011, Red Hat, Inc. and individual contributors

    4. 

  4.  * by the @authors tag. See the copyright.txt in the distribution for a

    5. 

  5.  * full listing of individual contributors.

    6. 

  6.  *

    7. 

  7.  * This is free software; you can redistribute it and/or modify it

    8. 

  8.  * under the terms of the GNU Lesser General Public License as

    9. 

  9.  * published by the Free Software Foundation; either version 2.1 of

    10. 

  10.  * the License, or (at your option) any later version.

    11. 

  11.  *

    12. 

  12.  * This software is distributed in the hope that it will be useful,

    13. 

  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

    14. 

  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

    15. 

  15.  * Lesser General Public License for more details.

    16. 

  16.  *

    17. 

  17.  * You should have received a copy of the GNU Lesser General Public

    18. 

  18.  * License along with this software; if not, write to the Free

    19. 

  19.  * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA

    20. 

  20.  * 02110-1301 USA, or see the FSF site: http://www.fsf.org.

    21. 

  21.  */

    22. 

  22. 

    23. 

  23. package org.mobicents.protocols.ss7.map.functional;

    24. 

  24. 

    25. 

  25. import org.mobicents.protocols.ss7.map.api.MAPServiceBase;

    26. 

  26. import org.mobicents.protocols.ss7.map.api.MAPException;

    27. 

  27. import org.mobicents.protocols.ss7.map.api.dialog.MAPProviderAbortReason;

    28. 

  28. import org.mobicents.protocols.ss7.map.MAPProviderImpl;

    29. 

  29. import org.mobicents.protocols.ss7.map.api.service.supplementary.MAPServiceSupplementary;

    30. 

  30. import org.mobicents.protocols.ss7.map.primitives.MAPExtensionContainerTest;

    31. 

  31. import org.mobicents.protocols.ss7.tcap.api.TCAPProvider;

    32. 

  32. import org.mobicents.protocols.ss7.tcap.api.tc.dialog.events.TCBeginIndication;

    33. 

  33. import org.mobicents.protocols.ss7.tcap.asn.ApplicationContextName;

    34. 

  34. import org.mobicents.protocols.ss7.tcap.asn.comp.Component;

    35. 

  35. 

    36. 

  36. 

    37. 

  37. /**

    38. 

  38.  * 

    39. 

  39.  * @author amit bhayani

    40. 

  40.  * @author sergey vetyutnev

    41. 

  41.  * 

    42. 

  42.  */

    43. 

  43. public class MAPProviderImplWrapper extends MAPProviderImpl {

    44. 

  44. 

    45. 

  45. 	private int testMode = 0;

    46. 

  46. 

    47. 

  47. 	private final MAPServiceSupplementary mapServiceSupplementaryTest = new MAPServiceSupplementaryImplWrapper(this);

    48. 

  48. 

    49. 

  49. 	public MAPProviderImplWrapper(TCAPProvider tcapProvider) {

    50. 

  50. 		super(tcapProvider);

    51. 

  51. 		

    52. 

  52. 		for(MAPServiceBase serv : this.mapServices) {

    53. 

  53. 			if( serv instanceof MAPServiceSupplementary ) {

    54. 

  54. 				this.mapServices.remove(serv);

    55. 

  55. 				break;

    56. 

  56. 			}

    57. 

  57. 		}

    58. 

  58. 		

    59. 

  59. 		this.mapServices.add(this.mapServiceSupplementaryTest);

    60. 

  60. 	}

    61. 

  61. 

    62. 

  62. 	public MAPServiceSupplementary getMAPServiceSupplementary() {

    63. 

  63. 		return this.mapServiceSupplementaryTest;

    64. 

  64. 	}

    65. 

  65. 

    66. 

  66. 	public void setTestMode(int testMode) {

    67. 

  67. 		this.testMode = testMode;

    68. 

  68. 	}

    69. 

  69. 

    70. 

  70. 	public void onTCBegin(TCBeginIndication tcBeginIndication) {

    71. 

  71. 		ApplicationContextName acn = tcBeginIndication.getApplicationContextName();

    72. 

  72. 		Component[] comps = tcBeginIndication.getComponents();

    73. 

  73. 

    74. 

  74. 		if (this.testMode == 1) {

    75. 

  75. 			try {

    76. 

  76. 				this.fireTCAbortProvider(tcBeginIndication.getDialog(), MAPProviderAbortReason.invalidPDU,

    77. 

  77. 						MAPExtensionContainerTest.GetTestExtensionContainer());

    78. 

  78. 			} catch (MAPException e) {

    79. 

  79. 				loger.error("Error while firing TC-U-ABORT. ", e);

    80. 

  80. 			}

    81. 

  81. 			return;

    82. 

  82. 		}

    83. 

  83. 

    84. 

  84. 		super.onTCBegin(tcBeginIndication);

    85. 

  85. 	}

    86. 

  86. }
    87.