def test_list_flows_in_bucket(fixture_versioned_flow):
test_rc, test_rb, test_pg, test_p, test_vf = fixture_versioned_flow
r1 = versioning.list_flows_in_bucket(test_rb.identifier)
assert isinstance(r1, list)
assert isinstance(r1[0], registry.VersionedFlow)
with pytest.raises(ValueError, match='Bucket does not exist'):
_ = versioning.list_flows_in_bucket('NiPyApi-FakeNews')
def nifireg_delete_flows(identifier, identifier_type='name'):
bucket = versioning.get_registry_bucket(identifier, identifier_type)
if bucket:
for flow in versioning.list_flows_in_bucket(bucket.identifier):
endpoint = '/buckets/{bucketId}/flows/{flowId}'.format(
bucketId=flow.bucket_identifier, flowId=flow.identifier)
resp = nifireg_api_delete(
endpoint, headers={'Content-Type': 'application/json'})
def test_nifi_registry_iot(run_id):
bucket = versioning.get_registry_bucket('IoT')
assert bucket
flows = versioning.list_flows_in_bucket(bucket.identifier)
assert len(flows) == 1
flow = flows[0]
versions = versioning.list_flow_versions(bucket.identifier, flow.identifier)
assert len(versions) >= 2
assert versions[0].comments == 'Second version - ' + run_id, 'Comments: ' + versions[0].comments
assert versions[1].comments == 'First version - ' + run_id, 'Comments: ' + versions[1].comments
def test_nifi_registry_sensorflows(run_id):
bucket = versioning.get_registry_bucket('SensorFlows')
assert bucket
flows = versioning.list_flows_in_bucket(bucket.identifier)
assert len(flows) == 1
flow = flows[0]
versions = versioning.list_flow_versions(bucket.identifier, flow.identifier)
assert len(versions) == 3
assert versions[0].comments == 'Second version - ' + run_id, 'Comments: ' + versions[0].comments
assert versions[1].comments == 'First version - ' + run_id, 'Comments: ' + versions[1].comments
assert versions[2].comments == 'Enabled version control - ' + run_id, 'Comments: ' + versions[2].comments
def test_list_flows_in_bucket(regress_flow_reg, fix_ver_flow):
r1 = versioning.list_flows_in_bucket(fix_ver_flow.bucket.identifier)
assert isinstance(r1, list)
assert isinstance(r1[0], registry.VersionedFlow)
with pytest.raises(ValueError, match='does not exist'):
_ = versioning.list_flows_in_bucket('NiPyApi-FakeNews')
def delete_flows(identifier, identifier_type='name'):
bucket = versioning.get_registry_bucket(identifier, identifier_type)
if bucket:
for flow in versioning.list_flows_in_bucket(bucket.identifier):
BucketFlowsApi().delete_flow(flow.bucket_identifier,
flow.identifier)
