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 _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")
