Example #1
0
  def test_main(self, java_mock, exists_mock, expandWebApp_mock, atlasDir_mock, executeEnvSh_mock, writePid_mock, exist_pid_mock, grep_mock, configure_hbase_mock, runProcess_mock):
    sys.argv = []
    exists_mock.return_value = True
    expandWebApp_mock.return_value = "webapp"
    atlasDir_mock.return_value = "atlas_home"

    exist_pid_mock(789)
    exist_pid_mock.assert_called_with(789)
    grep_mock.return_value = "hbase"

    atlas.main()
    self.assertTrue(configure_hbase_mock.called)
    runProcess_mock.assert_called_with(['atlas_home/hbase/bin/hbase-daemon.sh', '--config', 'atlas_home/hbase/conf', 'start', 'master'], 'atlas_home/logs', False, True)
    self.assertTrue(java_mock.called)
    if IS_WINDOWS:
      
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'atlas_home\\server\\webapp\\atlas'],
        'atlas_home\\conf;atlas_home\\server\\webapp\\atlas\\WEB-INF\\classes;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\atlas-titan-${project.version}.jar;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;atlas_home\\libext\\*;atlas_home\\hbase\\conf',
        ['-Datlas.log.dir=atlas_home\\logs', '-Datlas.log.file=application.log', '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home\\conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml', '-Djava.net.preferIPv4Stack=true'], 'atlas_home\\logs')
      
    else:
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'atlas_home/server/webapp/atlas'],
        'atlas_home/conf:atlas_home/server/webapp/atlas/WEB-INF/classes:atlas_home/server/webapp/atlas/WEB-INF/lib/atlas-titan-${project.version}.jar:atlas_home/server/webapp/atlas/WEB-INF/lib/*:atlas_home/libext/*:atlas_home/hbase/conf',
        ['-Datlas.log.dir=atlas_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home/conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml', '-Djava.net.preferIPv4Stack=true'],  'atlas_home/logs')

    pass
  def test_main(self, java_mock, exists_mock, expandWebApp_mock, metadataDir_mock, executeEnvSh_mock, writePid_mock, unix_exist_pid_mock, win_exist_pid_mock):
    sys.argv = []
    exists_mock.return_value = True
    expandWebApp_mock.return_value = "webapp"
    metadataDir_mock.return_value = "metadata_home"

    win_exist_pid_mock("789")
    win_exist_pid_mock.assert_called_with((str)(789))
    unix_exist_pid_mock(789)
    unix_exist_pid_mock.assert_called_with(789)
    metadata.main()
    self.assertTrue(java_mock.called)
    if IS_WINDOWS:
      
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'metadata_home\\server\\webapp\\atlas'],
        'metadata_home\\conf;metadata_home\\server\\webapp\\atlas\\WEB-INF\\classes;metadata_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;metadata_home\\libext\\*;metadata_home\\hbase\\conf',
        ['-Datlas.log.dir=metadata_home\\logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home\\conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml'], 'metadata_home\\logs')
      
      
      
        
    else:
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'metadata_home/server/webapp/atlas'],
        'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib/*:metadata_home/libext/*:metadata_home/hbase/conf',
        ['-Datlas.log.dir=metadata_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home/conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml'],  'metadata_home/logs')
      
      
    pass
    def test_main(
        self,
        java_mock,
        exists_mock,
        expandWebApp_mock,
        atlasDir_mock,
        executeEnvSh_mock,
        writePid_mock,
        unix_exist_pid_mock,
        win_exist_pid_mock,
    ):
        sys.argv = []
        exists_mock.return_value = True
        expandWebApp_mock.return_value = "webapp"
        atlasDir_mock.return_value = "atlas_home"

        win_exist_pid_mock("789")
        win_exist_pid_mock.assert_called_with((str)(789))
        unix_exist_pid_mock(789)
        unix_exist_pid_mock.assert_called_with(789)
        atlas.main()
        self.assertTrue(java_mock.called)
        if IS_WINDOWS:

            java_mock.assert_called_with(
                "org.apache.atlas.Atlas",
                ["-app", "atlas_home\\server\\webapp\\atlas"],
                "atlas_home\\conf;atlas_home\\server\\webapp\\atlas\\WEB-INF\\classes;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\atlas-titan-${project.version}.jar;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;atlas_home\\libext\\*;atlas_home\\hbase\\conf",
                [
                    "-Datlas.log.dir=atlas_home\\logs",
                    "-Datlas.log.file=application.log",
                    "-Datlas.home=atlas_home",
                    "-Datlas.conf=atlas_home\\conf",
                    "-Xmx1024m",
                    "-XX:MaxPermSize=512m",
                    "-Dlog4j.configuration=atlas-log4j.xml",
                    "-Djava.net.preferIPv4Stack=true",
                ],
                "atlas_home\\logs",
            )

        else:
            java_mock.assert_called_with(
                "org.apache.atlas.Atlas",
                ["-app", "atlas_home/server/webapp/atlas"],
                "atlas_home/conf:atlas_home/server/webapp/atlas/WEB-INF/classes:atlas_home/server/webapp/atlas/WEB-INF/lib/atlas-titan-${project.version}.jar:atlas_home/server/webapp/atlas/WEB-INF/lib/*:atlas_home/libext/*:atlas_home/hbase/conf",
                [
                    "-Datlas.log.dir=atlas_home/logs",
                    "-Datlas.log.file=application.log",
                    "-Datlas.home=atlas_home",
                    "-Datlas.conf=atlas_home/conf",
                    "-Xmx1024m",
                    "-XX:MaxPermSize=512m",
                    "-Dlog4j.configuration=atlas-log4j.xml",
                    "-Djava.net.preferIPv4Stack=true",
                ],
                "atlas_home/logs",
            )

        pass
Example #4
0
  def test_main_embedded(self, is_solr_local_mock, is_hbase_local_mock, java_mock, exists_mock, expandWebApp_mock,
                atlasDir_mock, executeEnvSh_mock, writePid_mock, exist_pid_mock, grep_mock, getConfig_mock,
                configure_hbase_mock, runProcess_mock):
    sys.argv = []
    exists_mock.return_value = True
    expandWebApp_mock.return_value = "webapp"
    atlasDir_mock.return_value = "atlas_home"
    is_hbase_local_mock.return_value = True
    is_solr_local_mock.return_value = True

    exist_pid_mock(789)
    exist_pid_mock.assert_called_with(789)
    grep_mock.return_value = "hbase"
    getConfig_mock.return_value = "localhost:9838"

    atlas.main()
    self.assertTrue(configure_hbase_mock.called)

    if IS_WINDOWS:
      calls = [call(['atlas_home\\hbase\\bin\\start-hbase.cmd', '--config', 'atlas_home\\hbase\\conf'], 'atlas_home\\logs', False, True),
               call(['atlas_home\\solr\\bin\\solr.cmd', 'start', '-z', 'localhost:9838', '-p', '9838'], 'atlas_home\\logs', False, True),
               call(['atlas_home\\solr\\bin\\solr.cmd', 'create', '-c', 'vertex_index', '-d', 'atlas_home\\solr\\server\\solr\\configsets\\basic_configs\\conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home\\logs', False, True),
               call(['atlas_home\\solr\\bin\\solr.cmd', 'create', '-c', 'edge_index', '-d', 'atlas_home\\solr\\server\\solr\\configsets\\basic_configs\\conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home\\logs', False, True),
               call(['atlas_home\\solr\\bin\\solr.cmd', 'create', '-c', 'fulltext_index', '-d', 'atlas_home\\solr\\server\\solr\\configsets\\basic_configs\\conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home\\logs', False, True)]

      runProcess_mock.assert_has_calls(calls)
    else:
      calls = [call(['atlas_home/hbase/bin/hbase-daemon.sh', '--config', 'atlas_home/hbase/conf', 'start', 'master'], 'atlas_home/logs', False, True),
               call(['atlas_home/solr/bin/solr', 'start', '-z', 'localhost:9838', '-p', '9838'], 'atlas_home/logs', False, True),
               call(['atlas_home/solr/bin/solr', 'create', '-c', 'vertex_index', '-d', 'atlas_home/solr/server/solr/configsets/basic_configs/conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home/logs', False, True),
               call(['atlas_home/solr/bin/solr', 'create', '-c', 'edge_index', '-d', 'atlas_home/solr/server/solr/configsets/basic_configs/conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home/logs', False, True),
               call(['atlas_home/solr/bin/solr', 'create', '-c', 'fulltext_index', '-d', 'atlas_home/solr/server/solr/configsets/basic_configs/conf', '-shards', '1', '-replicationFactor', '1'], 'atlas_home/logs', False, True)]

      runProcess_mock.assert_has_calls(calls)

    self.assertTrue(java_mock.called)
    if IS_WINDOWS:
      
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'atlas_home\\server\\webapp\\atlas'],
        'atlas_home\\conf;atlas_home\\server\\webapp\\atlas\\WEB-INF\\classes;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;atlas_home\\libext\\*;atlas_home\\hbase\\conf',
        ['-Datlas.log.dir=atlas_home\\logs', '-Datlas.log.file=application.log', '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home\\conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml', '-Djava.net.preferIPv4Stack=true'], 'atlas_home\\logs')
      
    else:
      java_mock.assert_called_with(
        'org.apache.atlas.Atlas',
        ['-app', 'atlas_home/server/webapp/atlas'],
        'atlas_home/conf:atlas_home/server/webapp/atlas/WEB-INF/classes:atlas_home/server/webapp/atlas/WEB-INF/lib/*:atlas_home/libext/*:atlas_home/hbase/conf',
        ['-Datlas.log.dir=atlas_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home/conf', '-Xmx1024m', '-XX:MaxPermSize=512m', '-Dlog4j.configuration=atlas-log4j.xml', '-Djava.net.preferIPv4Stack=true'],  'atlas_home/logs')

    pass
Example #5
0
    def test_main(self, java_mock, exists_mock, expandWebApp_mock,
                  metadataDir_mock, executeEnvSh_mock, writePid_mock,
                  unix_exist_pid_mock, win_exist_pid_mock):
        sys.argv = []
        exists_mock.return_value = True
        expandWebApp_mock.return_value = "webapp"
        metadataDir_mock.return_value = "metadata_home"

        win_exist_pid_mock("789")
        win_exist_pid_mock.assert_called_with((str)(789))
        unix_exist_pid_mock(789)
        unix_exist_pid_mock.assert_called_with(789)
        metadata.main()
        self.assertTrue(java_mock.called)
        if IS_WINDOWS:

            java_mock.assert_called_with(
                'org.apache.atlas.Atlas',
                ['-app', 'metadata_home\\server\\webapp\\atlas'],
                'metadata_home\\conf;metadata_home\\server\\webapp\\atlas\\WEB-INF\\classes;metadata_home\\server\\webapp\\atlas\\WEB-INF\\lib\\atlas-titan-${project.version}.jar;metadata_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;metadata_home\\libext\\*;metadata_home\\hbase\\conf',
                [
                    '-Datlas.log.dir=metadata_home\\logs',
                    '-Datlas.log.file=application.log',
                    '-Datlas.home=metadata_home',
                    '-Datlas.conf=metadata_home\\conf', '-Xmx1024m',
                    '-XX:MaxPermSize=512m',
                    '-Dlog4j.configuration=atlas-log4j.xml'
                ], 'metadata_home\\logs')

        else:
            java_mock.assert_called_with(
                'org.apache.atlas.Atlas',
                ['-app', 'metadata_home/server/webapp/atlas'],
                'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib/atlas-titan-${project.version}.jar:metadata_home/server/webapp/atlas/WEB-INF/lib/*:metadata_home/libext/*:metadata_home/hbase/conf',
                [
                    '-Datlas.log.dir=metadata_home/logs',
                    '-Datlas.log.file=application.log',
                    '-Datlas.home=metadata_home',
                    '-Datlas.conf=metadata_home/conf', '-Xmx1024m',
                    '-XX:MaxPermSize=512m',
                    '-Dlog4j.configuration=atlas-log4j.xml'
                ], 'metadata_home/logs')

        pass
Example #6
0
 def test_main(self, java_mock, exists_mock, expandWebApp_mock, metadataDir_mock, executeEnvSh_mock, writePid_mock):
   sys.argv = []
   exists_mock.return_value = True
   expandWebApp_mock.return_value = "webapp"
   metadataDir_mock.return_value = "metadata_home"
   metadata.main()
   self.assertTrue(java_mock.called)
   if IS_WINDOWS:
     java_mock.assert_called_with(
       'org.apache.atlas.Main',
       ['-app', 'metadata_home/server/webapp/atlas'],
       'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib\\*:metadata_home/libext\\*',
       ['-Datlas.log.dir=metadata_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home/conf', '-Xmx1024m'], 'metadata_home/logs')
   else:
     java_mock.assert_called_with(
       'org.apache.atlas.Main',
       ['-app', 'metadata_home/server/webapp/atlas'],
       'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib/*:metadata_home/libext/*',
       ['-Datlas.log.dir=metadata_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home/conf', '-Xmx1024m'],  'metadata_home/logs')
   pass
Example #7
0
 def test_main(self, java_mock, exists_mock, expandWebApp_mock, metadataDir_mock, executeEnvSh_mock, writePid_mock):
   sys.argv = []
   exists_mock.return_value = True
   expandWebApp_mock.return_value = "webapp"
   metadataDir_mock.return_value = "metadata_home"
   metadata.main()
   self.assertTrue(java_mock.called)
   if IS_WINDOWS:
     java_mock.assert_called_with(
       'org.apache.atlas.Main',
       ['-app', 'metadata_home/server/webapp/atlas'],
       'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib\\*:metadata_home/libext\\*',
       ['-Datlas.log.dir=metadata_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home/conf', '-Xmx1024m'], 'metadata_home/logs')
   else:
     java_mock.assert_called_with(
       'org.apache.atlas.Main',
       ['-app', 'metadata_home/server/webapp/atlas'],
       'metadata_home/conf:metadata_home/server/webapp/atlas/WEB-INF/classes:metadata_home/server/webapp/atlas/WEB-INF/lib/*:metadata_home/libext/*',
       ['-Datlas.log.dir=metadata_home/logs', '-Datlas.log.file=application.log', '-Datlas.home=metadata_home', '-Datlas.conf=metadata_home/conf', '-Xmx1024m'],  'metadata_home/logs')
   pass
Example #8
0
    def test_main_embedded(self, wait_for_startup_mock, is_solr_local_mock,
                           is_hbase_local_mock, java_mock, exists_mock,
                           expandWebApp_mock, atlasDir_mock, executeEnvSh_mock,
                           writePid_mock, exist_pid_mock, grep_mock,
                           getConfigWithDefault_mock, getConfig_mock,
                           configure_hbase_mock, runProcess_mock):
        sys.argv = []
        exists_mock.return_value = True
        expandWebApp_mock.return_value = "webapp"
        atlasDir_mock.return_value = "atlas_home"
        is_hbase_local_mock.return_value = True
        is_solr_local_mock.return_value = True
        wait_for_startup_mock.return_value = True

        exist_pid_mock(789)
        exist_pid_mock.assert_called_with(789)
        grep_mock.return_value = "hbase"
        getConfig_mock.side_effect = self.get_config_mock_side_effect
        getConfigWithDefault_mock.side_effect = self.get_default_config_mock_side_effect

        atlas.main()
        self.assertTrue(configure_hbase_mock.called)

        if IS_WINDOWS:
            calls = [
                call([
                    'atlas_home\\hbase\\bin\\start-hbase.cmd', '--config',
                    'atlas_home\\hbase\\conf'
                ], 'atlas_home\\logs', False, True),
                call([
                    'atlas_home\\solr\\bin\\solr.cmd', 'start', '-z',
                    'localhost:9838', '-p', '9838'
                ], 'atlas_home\\logs', False, True),
                call([
                    'atlas_home\\solr\\bin\\solr.cmd', 'create', '-c',
                    'vertex_index', '-d',
                    'atlas_home\\solr\\server\\solr\\configsets\\_default\\conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home\\logs', False, True),
                call([
                    'atlas_home\\solr\\bin\\solr.cmd', 'create', '-c',
                    'edge_index', '-d',
                    'atlas_home\\solr\\server\\solr\\configsets\\_default\\conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home\\logs', False, True),
                call([
                    'atlas_home\\solr\\bin\\solr.cmd', 'create', '-c',
                    'fulltext_index', '-d',
                    'atlas_home\\solr\\server\\solr\\configsets\\_default\\conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home\\logs', False, True)
            ]

            runProcess_mock.assert_has_calls(calls)
        else:
            calls = [
                call([
                    'atlas_home/hbase/bin/hbase-daemon.sh', '--config',
                    'atlas_home/hbase/conf', 'start', 'master'
                ], 'atlas_home/logs', False, True),
                call([
                    'atlas_home/solr/bin/solr', 'start', '-z',
                    'localhost:9838', '-p', '9838'
                ], 'atlas_home/logs', False, True),
                call([
                    'atlas_home/solr/bin/solr', 'create', '-c', 'vertex_index',
                    '-d',
                    'atlas_home/solr/server/solr/configsets/_default/conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home/logs', False, True),
                call([
                    'atlas_home/solr/bin/solr', 'create', '-c', 'edge_index',
                    '-d',
                    'atlas_home/solr/server/solr/configsets/_default/conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home/logs', False, True),
                call([
                    'atlas_home/solr/bin/solr', 'create', '-c',
                    'fulltext_index', '-d',
                    'atlas_home/solr/server/solr/configsets/_default/conf',
                    '-shards', '1', '-replicationFactor', '1'
                ], 'atlas_home/logs', False, True)
            ]

            runProcess_mock.assert_has_calls(calls)

        self.assertTrue(java_mock.called)
        if IS_WINDOWS:

            java_mock.assert_called_with(
                'org.apache.atlas.Atlas',
                ['-app', 'atlas_home\\server\\webapp\\atlas'],
                'atlas_home\\conf;atlas_home\\server\\webapp\\atlas\\WEB-INF\\classes;atlas_home\\server\\webapp\\atlas\\WEB-INF\\lib\\*;atlas_home\\libext\\*;atlas_home\\hbase\\conf',
                [
                    '-Datlas.log.dir=atlas_home\\logs',
                    '-Datlas.log.file=application.log',
                    '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home\\conf',
                    '-Xmx1024m', '-Dlog4j.configuration=atlas-log4j.xml',
                    '-Djava.net.preferIPv4Stack=true', '-server'
                ], 'atlas_home\\logs')

        else:
            java_mock.assert_called_with(
                'org.apache.atlas.Atlas',
                ['-app', 'atlas_home/server/webapp/atlas'],
                'atlas_home/conf:atlas_home/server/webapp/atlas/WEB-INF/classes:atlas_home/server/webapp/atlas/WEB-INF/lib/*:atlas_home/libext/*:atlas_home/hbase/conf',
                [
                    '-Datlas.log.dir=atlas_home/logs',
                    '-Datlas.log.file=application.log',
                    '-Datlas.home=atlas_home', '-Datlas.conf=atlas_home/conf',
                    '-Xmx1024m', '-Dlog4j.configuration=atlas-log4j.xml',
                    '-Djava.net.preferIPv4Stack=true', '-server'
                ], 'atlas_home/logs')

        pass