def test_mark_step_done(self): self.create_bucket() t = MarkStepDoneTask(BUCKET, "steps/", "mystep") t.execute() self.assertEquals( storage.pull_content(BUCKET, "steps/mystep"), json.dumps(UNKNOWN_CONTEXT))
def test_mark_step_done(self): self.create_bucket() t = MarkStepDoneTask(BUCKET, "steps/", "mystep") t.execute() self.assertEqual( storage.pull_content(BUCKET, "steps/mystep"), json.dumps(UNKNOWN_CONTEXT))
def test_metrology(self): self.create_bucket() ex = Executor(MyWorkflow) ex.run(input={"args": [1], "kwargs": {}}) res = json.loads( storage.pull_content(settings.METROLOGY_BUCKET, "local/local/activity.0.json")) self.assertEquals(res[0]["name"], "Step1") self.assertEquals(res[0]["read"]["records"], 1) self.assertEquals(res[0]["metadata"]["num"], 1) res = json.loads( storage.pull_content(settings.METROLOGY_BUCKET, "local/local/metrology.json")) self.assertEquals(res[0][1]["metrology"][0]["name"], "Step1") self.assertEquals(res[0][1]["metrology"][0]["read"]["records"], 1) self.assertEquals(res[0][1]["metrology"][0]["metadata"]["num"], 1)
def test_metrology(self): self.create_bucket() ex = Executor(MyWorkflow) ex.run(input={"args": [1], "kwargs": {}}) res = json.loads(storage.pull_content( settings.METROLOGY_BUCKET, "local/local/activity.0.json")) self.assertEqual(res["meta"], "foo bar") steps = res["steps"] self.assertEqual(steps[0]["name"], "Step1") self.assertEqual(steps[0]["read"]["records"], 1) self.assertEqual(steps[0]["metadata"]["num"], 1) res = json.loads(storage.pull_content( settings.METROLOGY_BUCKET, "local/local/metrology.json")) self.assertEqual(res[0][1]["metrology"]["meta"], "foo bar") self.assertEqual(res[0][1]["metrology"]["steps"][0]["name"], "Step1") self.assertEqual(res[0][1]["metrology"]["steps"][0]["read"]["records"], 1) self.assertEqual(res[0][1]["metrology"]["steps"][0]["metadata"]["num"], 1)
def _pull_jumbo_field(location): bucket, path = location.replace(constants.JUMBO_FIELDS_PREFIX, "").split("/", 1) cached_value = _get_cached(path) if cached_value: return cached_value content = storage.pull_content(bucket, path) _set_cached(path, content) return content
def test_pull_content(self): self.create() storage.push(self.bucket, "mykey.txt", self.tmp_filename) self.assertEqual(storage.pull_content(self.bucket, "mykey.txt"), "42")
def test_pull_content(self): self.create() storage.push(self.bucket, "mykey.txt", self.tmp_filename) self.assertEqual( storage.pull_content(self.bucket, "mykey.txt"), "42")