Пример #1
0
def connectionTest():
    cListA = range(26, 235)
    pCfgFiles = {
        0: ("obj.cfg", "a lot of obj cfg info"),
        1: ("fs.cfg", "some forcedSource info"),
        2: ("junk_cfg", "blah blah junk\n more stuff")
    }
    ingest_dict = {
        'host': 'mt.st.com',
        'port': 2461,
        'auth': '1234',
        'db': 'afake_db',
        'skip': False,
        'keep': True
    }
    pregen_dict = {
        0: ("visit_ccd_test.csv", "1,2,3,55,22,10.5,something,4"),
        1: ("junk_file.txt", "some other stuff"),
        2: ("type.txt", "The quick brown fox jumped over the lazy dog."),
        3: ("skey", "asjd43rauydfsf4baeuyrf784r;;!@")
    }
    timing_dict = TimingDict()
    timing_dict.add('gen_o', 345.23)
    timing_dict.add('gen_fs', 981.23)
    timing_dict.add('conv', 12.9999)
    timing_dict.add('trans', 42.1)
    timing_dict.add('del', 0.123)
    timing_dict.increment()
    success, s_warn1, c_warn1 = testDataGenConnection(
        14242, 'qt', 10000, 30, 178, 'bunch of json file entries', 28, cListA,
        pCfgFiles, ingest_dict, timing_dict, pregen_dict, 4367)
    if not success:
        print("First test failed")
        exit(1)

    ingest_dict = {
        'host': 'mt.st.edu',
        'port': 0,
        'auth': '',
        'db': 'diff_db',
        'skip': True,
        'keep': False
    }
    pregen_dict = {}
    timing_dict = TimingDict()
    success, s_warn2, c_warn2 = testDataGenConnection(
        14242, 'qt', 10000, 30, 1, 'bunch of json file entries', 28, cListA,
        pCfgFiles, ingest_dict, timing_dict, pregen_dict, 92)

    print("success=", success, "serv_warn=", s_warn1, s_warn2, "client_warn=",
          c_warn1, c_warn2)
    def servRecvTiming(self):
        """Receive the dictionary of timing information from the client

        Return
        ------
        timing_dict : TimingDict
            TimingDict object
        """
        print("servRecvTiming C_TIMDCT")
        msg_id, msg, msg_len = self._recv_msg()
        if msg_id != self._C_TIMDCT:
            self.warnings += 1
            raise DataGenError(
                f'ERROR servRecvTiming {str(msg_id)}:{str(msg_len)} {msg}')
        msg_split = msg.split(self.COMPLEXSEP)
        timing_dict = TimingDict()
        timing_dict.count = int(msg_split[0])

        # msg_split contains an alternating sequence of key then
        # value starting at msg_split[1]
        for j in range(1, len(msg_split) - 1, 2):
            timing_dict.add(msg_split[j], msg_split[j + 1])
        return timing_dict