Use the build object directly instead of its JSON in the modules handler

This commit is contained in:
mprahl
2018-03-01 19:43:43 -05:00
parent d3f92bf5d3
commit f19f76872c
3 changed files with 16 additions and 20 deletions

View File

@@ -59,6 +59,11 @@ class TestModuleWait:
builder.module_build_tag = {'name': 'some-tag-build'}
create_builder.return_value = builder
mocked_module_build = mock.Mock()
mocked_module_build.name = 'foo'
mocked_module_build.stream = 'stream'
mocked_module_build.version = '1'
mocked_module_build.state = 1
mocked_module_build.id = 1
mocked_module_build.json.return_value = {
'name': 'foo',
'stream': '1',
@@ -66,7 +71,6 @@ class TestModuleWait:
'state': 'some state',
'id': 1
}
mocked_module_build.extended_json = mocked_module_build.json
formatted_testmodule_yml_path = os.path.join(
base_dir, 'staged_data', 'formatted_testmodule.yaml')