From 8094254ff973fb0663a82988ed3a20484c1c780c Mon Sep 17 00:00:00 2001
From: Alexander Couzens <lynxis@fe80.eu>
Date: Tue, 6 Apr 2021 15:53:10 +0200
Subject: [PATCH] pcu: PCU_Tests_NS: add support for vty

Allow to use the vty by test cases. f_init_vty(id) must be called
before using the vty.

Related: SYS#5416
Change-Id: Ib2011babb123e1e1bb576e72b9bfbda90dce2df7
---
 pcu/PCU_Tests_NS.ttcn | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pcu/PCU_Tests_NS.ttcn b/pcu/PCU_Tests_NS.ttcn
index e2bad4f9d..eea8d3f5f 100644
--- a/pcu/PCU_Tests_NS.ttcn
+++ b/pcu/PCU_Tests_NS.ttcn
@@ -12,6 +12,8 @@ module PCU_Tests_NS {
 
 import from General_Types all;
 import from Osmocom_Types all;
+import from Osmocom_VTY_Functions all;
+import from TELNETasp_PortType all;
 import from PCU_Tests all;
 import from SGSN_Components all;
 import from Osmocom_Gb_Types all;
@@ -29,11 +31,19 @@ type component RAW_PCU_CT {
 	/* PCUIF (we emulate the BTS part) */
 	port PCUIF_CODEC_PT PCU;
 	var ConnectionId g_pcu_conn_id := -1;
+	/* VTY connection to the PCU */
+	port TELNETasp_PT PCUVTY;
 }
 
 type component RAW_Test_CT extends RAW_NS_CT, RAW_PCU_CT {
 }
 
+function f_init_vty(charstring id) runs on RAW_PCU_CT {
+	map(self:PCUVTY, system:PCUVTY);
+	f_vty_set_prompts(PCUVTY);
+	f_vty_transceive(PCUVTY, "enable");
+}
+
 function f_init_pcuif() runs on RAW_PCU_CT {
 	var PCUIF_info_ind info_ind;
 	map(self:PCU, system:PCU);