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')