| #!/usr/bin/env python |
| """ |
| Licensed to the Apache Software Foundation (ASF) under one |
| or more contributor license agreements. See the NOTICE file |
| distributed with this work for additional information |
| regarding copyright ownership. The ASF licenses this file |
| to you under the Apache License, Version 2.0 (the |
| "License"); you may not use this file except in compliance |
| with the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| |
| """ |
| |
| from resource_management import * |
| |
| class OozieServiceCheck(Script): |
| def service_check(self, env): |
| import params |
| env.set_params(params) |
| |
| # on HDP1 this file is different |
| smoke_test_file_name = 'oozieSmoke2.sh' |
| |
| oozie_smoke_shell_file( smoke_test_file_name) |
| |
| def oozie_smoke_shell_file( |
| file_name |
| ): |
| import params |
| |
| File( format("/tmp/{file_name}"), |
| content = StaticFile(file_name), |
| mode = 0755 |
| ) |
| |
| if params.security_enabled: |
| sh_cmd = format("sh /tmp/{file_name} {conf_dir} {hadoop_conf_dir} {smokeuser} {security_enabled} {smokeuser_keytab} {kinit_path_local}") |
| else: |
| sh_cmd = format("sh /tmp/{file_name} {conf_dir} {hadoop_conf_dir} {smokeuser} {security_enabled}") |
| |
| Execute( format("/tmp/{file_name}"), |
| command = sh_cmd, |
| tries = 3, |
| try_sleep = 5, |
| logoutput = True |
| ) |
| |
| def main(): |
| import sys |
| command_type = 'service_check' |
| command_data_file = '/root/workspace/Oozie/input.json' |
| basedir = '/root/workspace/Oozie/main' |
| sys.argv = ["", command_type, command_data_file, basedir] |
| |
| OozieServiceCheck().execute() |
| |
| if __name__ == "__main__": |
| OozieServiceCheck().execute() |
| #main() |
| |