Exemplos de SqliteDataStore em Python

Exemplo n.º 1

Arquivo: http_data_store_test.py Projeto: sikopet/grr

def _StartServers():
    global HTTP_DB
    global STARTED_SERVER
    logging.info("Using TMP_DIR:" + TMP_DIR)
    temp_dir_1 = TMP_DIR + "/1"
    temp_dir_2 = TMP_DIR + "/2"
    os.mkdir(temp_dir_1)
    os.mkdir(temp_dir_2)
    HTTP_DB = [
        sqlite_data_store.SqliteDataStore(temp_dir_1),
        sqlite_data_store.SqliteDataStore(temp_dir_2)
    ]
    if ipaddr.IPAddress(_GetLocalIPAsString()).version == 6:
        af = socket.AF_INET6
    else:
        af = socket.AF_INET
    STARTED_SERVER = [
        threading.Thread(target=data_server.Start,
                         args=(HTTP_DB[0], PORT[0], af, True,
                               StoppableHTTPServer, MockRequestHandler1)),
        threading.Thread(target=data_server.Start,
                         args=(HTTP_DB[1], PORT[1], af, False,
                               StoppableHTTPServer, MockRequestHandler2))
    ]
    STARTED_SERVER[0].start()
    STARTED_SERVER[1].start()

Exemplo n.º 2

def _StartServer(temp_dir):
  global HTTP_DB
  global STARTED_SERVER
  HTTP_DB = sqlite_data_store.SqliteDataStore(temp_dir)
  STARTED_SERVER = threading.Thread(
      target=data_server.Start,
      args=(HTTP_DB, PORT, True, StoppableHTTPServer, MockRequestHandler))
  STARTED_SERVER.start()

Exemplo n.º 3

Arquivo: sqlite_data_store_test.py Projeto: wprelic/grr

    def InitDatastore(self):
        self.token = access_control.ACLToken(username="******",
                                             reason="Running tests")
        self.root_path = utils.SmartStr("%s/sqlite_test/" % self.temp_dir)

        config_lib.CONFIG.Set("Datastore.location", self.root_path)

        self.DestroyDatastore()

        data_store.DB = sqlite_data_store.SqliteDataStore()
        data_store.DB.security_manager = test_lib.MockSecurityManager()

Exemplo n.º 4

Arquivo: sqlite_data_store_test.py Projeto: sikopet/grr

  def InitDatastore(self):
    self.token = access_control.ACLToken(
        username="******", reason="Running tests")
    self.root_path = utils.SmartStr("%s/sqlite_test/" % self.temp_dir)

    with test_lib.ConfigOverrider({"Datastore.location": self.root_path}):

      self.DestroyDatastore()

      data_store.DB = sqlite_data_store.SqliteDataStore()
      data_store.DB.Initialize()

Exemplo n.º 5

Arquivo: http_data_store_test.py Projeto: stevensfwang-via/grr

def _StartServers():
    global HTTP_DB
    global STARTED_SERVER
    logging.info("Using TMP_DIR:" + TMP_DIR)
    temp_dir_1 = TMP_DIR + "/1"
    temp_dir_2 = TMP_DIR + "/2"
    os.mkdir(temp_dir_1)
    os.mkdir(temp_dir_2)
    HTTP_DB = [
        sqlite_data_store.SqliteDataStore(temp_dir_1),
        sqlite_data_store.SqliteDataStore(temp_dir_2)
    ]
    STARTED_SERVER = [
        threading.Thread(target=data_server.Start,
                         args=(HTTP_DB[0], PORT[0], True, StoppableHTTPServer,
                               MockRequestHandler1)),
        threading.Thread(target=data_server.Start,
                         args=(HTTP_DB[1], PORT[1], False, StoppableHTTPServer,
                               MockRequestHandler2))
    ]
    STARTED_SERVER[0].start()
    STARTED_SERVER[1].start()