def test_get(self):
rpcs = [
call("[XWB]113021.108XWB GET VARIABLE VALUE51028$O(^PRST(459,\"B\",\"15-22\",1))f"),
call("[XWB]113021.108\nDDR LISTER52006\"FILE\"006459.01t006\"IENS\"005,618,t008\"FIELDS\"087@;.01E;3;4;10;6;11;29;34;37;40;61;63;64;66;85;88;94;118;131;132;133;134;135;136;137;138t007\"FLAGS\"002IPt006\"XREF\"001Bt008\"SCREEN\"021I $P(^(0),U,10)=\"015\"f"),
call("[XWB]113021.108\nDDR LISTER52006\"FILE\"006459.01t006\"IENS\"005,618,t008\"FIELDS\"087@;.01E;3;4;10;6;11;29;34;37;40;61;63;64;66;85;88;94;118;131;132;133;134;135;136;137;138t007\"FLAGS\"002IPt006\"XREF\"001Bt008\"SCREEN\"021I $P(^(0),U,10)=\"016\"f")
]
f = open('run618_015.txt', 'rb')
response_015 = f.read().decode()
f.close()
f = open('run618_016.txt', 'rb')
response_016 = f.read().decode()
f.close()
cxn = RpcServer(None, 0)
cxn.execute = MagicMock(side_effect=[
'618', response_015, response_016
])
run = FmsPayrun(cxn, '15-22')
rex = run.get_records(['015', '016'])
self.assertEqual(cxn.execute.call_args_list, rpcs)
self.assertEqual(len(rex), 2)
self.assertEqual(len(rex['015']), 39)
self.assertEqual(len(rex['016']), 91)
def _connect(self):
svr = RpcServer(self.host, self.port)
svr.open()
if not svr.is_open:
msg = 'Unable to connect to %s, port %d' % (self.host, self.port)
raise RpcException(msg)
visitor = RpcVisitor(self.user)
visitor.visit(svr)
return svr
def test_save_run(self):
f = open("../../pyvista/fms/tests/run560_015.txt", "rb")
response_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run560_016.txt", "rb")
response_016 = f.read().decode()
f.close()
cxn = RpcServer(None, 0)
cxn.execute = MagicMock(side_effect=["560", response_015, response_016])
fms_run = FmsPayrun(cxn, "13-16")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("13-16", my_run)
def test_save_runs(self):
f = open("../../pyvista/fms/tests/run559_015.txt", "rb")
response559_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run559_016.txt", "rb")
response559_016 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run560_015.txt", "rb")
response560_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run560_016.txt", "rb")
response560_016 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run616_015.txt", "rb")
response616_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run616_016.txt", "rb")
response616_016 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run617_015.txt", "rb")
response617_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run617_016.txt", "rb")
response617_016 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run618_015.txt", "rb")
response618_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run618_016.txt", "rb")
response618_016 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run619_015.txt", "rb")
response619_015 = f.read().decode()
f.close()
f = open("../../pyvista/fms/tests/run619_016.txt", "rb")
response619_016 = f.read().decode()
f.close()
cxn = RpcServer(None, 0)
cxn.execute = MagicMock(
side_effect=[
"559",
response559_015,
response559_016,
"560",
response560_015,
response560_016,
"616",
response616_015,
response616_016,
"617",
response617_015,
response617_016,
"618",
response618_015,
response618_016,
"619",
response619_015,
response619_016,
]
)
fms_run = FmsPayrun(cxn, "13-15")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("13-15", my_run)
fms_run = FmsPayrun(cxn, "13-16")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("13-16", my_run)
fms_run = FmsPayrun(cxn, "15-20")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("15-20", my_run)
fms_run = FmsPayrun(cxn, "15-21")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("15-21", my_run)
fms_run = FmsPayrun(cxn, "15-22")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("15-22", my_run)
fms_run = FmsPayrun(cxn, "15-23")
my_run = fms_run.get_records(["015", "016"])
self.controller.save_run("15-23", my_run)
def test_get_ien(self):
cxn = RpcServer(None, 0)
cxn.execute = MagicMock(return_value='618')
run = FmsPayrun(cxn, '15-22')
ien = run._get_ien()
self.assertEqual(ien, '618')
