Example #1
0
                "exception_severity": 0,
            }),
        )
        self.assertEqual(
            result[ids[4]].get_info(),
            JobInfo({
                "id": ids[4],
                "result": flux.constants.FLUX_JOB_RESULT_CANCELED,
                "t_submit": 0.0,
                "exception_occurred": True,
                "exception_type": "cancel",
                "exception_note": "",
                "exception_severity": 0,
            }),
        )

        # synchronous job.result() test
        self.assertEqual(job.result(self.fh, ids[3]),
                         result[ids[3]].get_info())


if __name__ == "__main__":
    from subflux import rerun_under_flux

    if rerun_under_flux(size=__flux_size(), personality="job"):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())

# vi: ts=4 sw=4 expandtab
Example #2
0
        self.assertEqual(r[u"\xa3"], u"value")
        self.assertEqual(r["key"], u"\u32db \u263a \u32e1")
        self.assertTrue(isinstance(r["key"], six.text_type))

    def test_rpc_with(self):
        """Sending a ping"""
        with self.f.rpc("cmb.ping", {"seq": 1, "pad": "stuff"}) as r:
            j = r.get()
            self.assertEqual(j["seq"], 1)
            self.assertEqual(j["pad"], "stuff")

    def test_rpc_null_payload(self):
        """Sending a request that receives a NULL response"""
        resp = self.f.rpc("attr.set", {
            "name": "attr-that-doesnt-exist",
            "value": "foo"
        }).get()
        self.assertIsNone(resp)

    def test_get_rank(self):
        """Get flux rank"""
        rank = self.f.get_rank()
        self.assertEqual(rank, 0)


if __name__ == "__main__":
    if rerun_under_flux(__flux_size()):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())
Example #3
0
        self.assertEqual(r, "")
        self.assertEqual(len(list(ds)), 14)
        self.assertEqual(len(list(fs)), 14)

        for r, ds, fs in walk_gen:
            pass

    def test_put_mkdir(self):
        flux.kvs.put_mkdir(self.f, "txn_mkdir")
        flux.kvs.commit(self.f)
        self.assertTrue(flux.kvs.exists(self.f, "txn_mkdir"))

    def test_put_unlink(self):
        flux.kvs.put(self.f, "txn_unlink", 1)
        flux.kvs.commit(self.f)
        flux.kvs.put_unlink(self.f, "txn_unlink")
        flux.kvs.commit(self.f)
        self.assertFalse(flux.kvs.exists(self.f, "txn_unlink"))

    def test_put_symlink(self):
        flux.kvs.put_symlink(self.f, "txn_symlink", "txn_target")
        flux.kvs.commit(self.f)
        self.assertFalse(flux.kvs.exists(self.f, "txn_symlink"))


if __name__ == "__main__":
    if rerun_under_flux(__flux_size()):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())
Example #4
0
                                    ensure_ascii=False).encode("utf-8")

        for topic in [unicode_topic, binary_topic]:
            resp = self.fh.mrpc_create(topic=topic,
                                       payload=json_payload,
                                       rankset=[0]).get()
            self.assertEqual(resp["foo"], u"bar")

        for payload in [json_payload, unicode_payload, binary_payload]:
            resp = self.fh.mrpc_create(binary_topic,
                                       payload=payload,
                                       rankset=[0]).get()
            self.assertEqual(resp["foo"], u"bar")

    def test_06_null_response_payload(self):
        resp = self.fh.mrpc_create("attr.set", {
            "name": "attr-that-doesnt-exist",
            "value": "foo"
        },
                                   rankset=[0]).get()
        self.assertEqual(resp, None)


if __name__ == "__main__":
    from subflux import rerun_under_flux

    if rerun_under_flux(size=_flux_size()):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())
Example #5
0
    def test_00_null_submit(self):
        with self.assertRaises(EnvironmentError) as error:
            job.submit(ffi.NULL, self.jobspec)
        self.assertEqual(error.exception.errno, errno.EINVAL)

        with self.assertRaises(EnvironmentError) as error:
            job.submit_get_id(ffi.NULL)
        self.assertEqual(error.exception.errno, errno.EINVAL)

        with self.assertRaises(EnvironmentError) as error:
            job.submit(self.fh, ffi.NULL)
        self.assertEqual(error.exception.errno, errno.EINVAL)

    def test_01_nonstring_submit(self):
        with self.assertRaises(TypeError):
            job.submit(self.fh, 0)

    def test_02_sync_submit(self):
        jobid = job.submit(self.fh, self.jobspec)
        self.assertGreater(jobid, 0)


if __name__ == "__main__":
    from subflux import rerun_under_flux

    if rerun_under_flux(size=__flux_size(), personality="job"):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())
Example #6
0
        unicode_topic = u"cmb.ping"

        json_payload = {"foo": u"bar"}
        unicode_payload = json.dumps(json_payload, ensure_ascii=False)
        binary_payload = json.dumps(json_payload, ensure_ascii=False).encode("utf-8")

        for topic in [unicode_topic, binary_topic]:
            resp = self.fh.mrpc_create(
                topic=topic, payload=json_payload, rankset=[0]
            ).get()
            self.assertEqual(resp["foo"], u"bar")

        for payload in [json_payload, unicode_payload, binary_payload]:
            resp = self.fh.mrpc_create(binary_topic, payload=payload, rankset=[0]).get()
            self.assertEqual(resp["foo"], u"bar")

    def test_06_null_response_payload(self):
        resp = self.fh.mrpc_create(
            "attr.set", {"name": "attr-that-doesnt-exist", "value": "foo"}, rankset=[0]
        ).get()
        self.assertEqual(resp, None)


if __name__ == "__main__":
    from subflux import rerun_under_flux

    if rerun_under_flux(size=_flux_size()):
        from pycotap import TAPTestRunner

        unittest.main(testRunner=TAPTestRunner())