testDBLoadGold.c

Go to the documentation of this file.

/*********************************************************************
Copyright (C) 2011-2012 Hewlett-Packard Development Company, L.P.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
version 2 as published by the Free Software Foundation.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*********************************************************************/

/* cunit includes */
#include <CUnit/CUnit.h>
#include "wget_agent.h"
#include "../utility.h"
#include <string.h>
#include <ctype.h>
#include "libfodbreposysconf.h"

#define AGENT_DIR "../../"

static PGresult *result = NULL;
extern fo_conf* sysconfig;

static fo_dbManager* dbManager;
int  DBLoadGoldInit()
{
  char URL[URLMAX];
  char TempFileDir[STRMAX];
  char TempFile[STRMAX];

  dbManager = createTestEnvironment(AGENT_DIR, "wget_agent", 1);
  if (!dbManager) {
    LOG_FATAL("Unable to connect to database");
    return 1;
  }

  pgConn = fo_dbManager_getWrappedConnection(dbManager);

  strcpy(GlobalParam, "-l 1 -A *.list -R *.deb");
  strcpy(URL, "https://mirrors.kernel.org/fossology/releases/3.0.0/ubuntu/14.04/");
  strcpy(TempFileDir, "./test_result/");
  strcpy(TempFile, "./test_result/fossology.sources.list");
  GetURL(TempFile, URL, TempFileDir);
  strcpy(GlobalTempFile,"./test_result/fossology.sources.list");
  strcpy(GlobalURL, "https://mirrors.kernel.org/fossology/releases/3.0.0/ubuntu/14.04/");

  memset(SQL,'\0',STRMAX);
  snprintf(SQL,STRMAX, "DELETE FROM upload where upload_filename = 'fossology.sources.list';");
  result = PQexec(pgConn, SQL);
  if (fo_checkPQcommand(pgConn, result, SQL, __FILE__ ,__LINE__))
  {
    printf("Perpare delete information ERROR!\n");
    return 1;
  }
  PQclear(result);

  memset(SQL,'\0',STRMAX);
  snprintf(SQL,STRMAX,"INSERT INTO upload (upload_filename,upload_mode,upload_ts) VALUES ('fossology.sources.list',40,now());");
  result = PQexec(pgConn, SQL);
  if (fo_checkPQcommand(pgConn, result, SQL, __FILE__ ,__LINE__))
  {
    printf("Perpare upload information ERROR!\n");
    return 1;
  }
  PQclear(result);
  memset(SQL,'\0',STRMAX);
  snprintf(SQL,STRMAX,"SELECT upload_pk from upload where upload_filename = 'fossology.sources.list';");
  result = PQexec(pgConn, SQL);
  if (fo_checkPQresult(pgConn, result, SQL, __FILE__ ,__LINE__))
  {
    printf("Perpare upload information ERROR!\n");
    return 1;
  }
  GlobalUploadKey = atoi(PQgetvalue(result,0,0));
  PQclear(result);

  GError* error = NULL;
  char* foConf = get_confFile();

  char cmd[STRMAX+1];
  snprintf(cmd, STRMAX, "sed -i 's|depth.*|depth=3|' %s", foConf);
  if (system(cmd) != 0) {
    printf("cannot reset depth to 3 with %s\n", cmd);
    return 1;
  }

  sysconfig = fo_config_load(foConf, &error);

  if (error) {
    printf("cannot load config from '%s' error: %s\n", foConf, error->message);
    return 1;
  }

  return 0;
}
int DBLoadGoldClean()
{
  memset(GlobalTempFile, 0, STRMAX);
  memset(GlobalURL, 0, URLMAX);
  memset(GlobalParam, 0, STRMAX);
  char TempFileDir[STRMAX];

  strcpy(TempFileDir, "./test_result");
  if (file_dir_existed(TempFileDir))
  {
    RemoveDir(TempFileDir);
  }

  if (sysconfig) {
    fo_config_free(sysconfig);
  }

  char repoDir[STRMAX+1];
  if (snprintf(repoDir, STRMAX, "%s/repo", get_sysconfdir())>0) {
    RemoveDir(repoDir);
  }

  dropTestEnvironment(dbManager, AGENT_DIR, "wget_agent");
  GlobalUploadKey = -1;

  return 0;
}

void string_tolower(char *string)
{
  int length = strlen(string);
  int i = 0;
  for (i = 0; i < length; i++)
  {
    string[i] = tolower(string[i]);
  }
}

/* test functions */

void testDBLoadGold()
{
  //printf("db start\n");
  DBLoadGold();
  //printf("db end\n");
  char SQL[STRMAX];
  char *pfile_sha1;
  char *pfile_md5;
  memset(SQL, 0, STRMAX);
  PGresult *result;
  snprintf(SQL, STRMAX-1, "select pfile_sha1, pfile_md5 from pfile where pfile_pk in (select pfile_fk from "
      "upload where upload_pk = %ld);", GlobalUploadKey);
  result =  PQexec(pgConn, SQL); /* SELECT */
  if (fo_checkPQresult(pgConn, result, SQL, __FILE__, __LINE__))
  {
    if (pgConn) PQfinish(pgConn);
    SafeExit(-1);
  }
  pfile_sha1 = PQgetvalue(result,0,0);
  pfile_md5 = PQgetvalue(result,0,1);
  //printf("pfile_sha1, pfile_md5 are:%s, %s\n", pfile_sha1, pfile_md5 );
  string_tolower(pfile_sha1);
  string_tolower(pfile_md5);
  //printf("pfile_sha1, pfile_md5 are:%s, %s\n", pfile_sha1, pfile_md5 );
  char file_name_file[STRMAX] = {0};
  char file_name_gold[STRMAX] = {0};
  char string0[3] = {0};
  char string1[3] = {0};
  char string2[3] = {0};
  char *string4 = get_sysconfdir();
  strncpy(string0, pfile_sha1, 2);
  strncpy(string1, pfile_sha1 + 2, 2);
  strncpy(string2, pfile_sha1 + 4, 2);
  //printf("string0, string1, string2 are:%s, %s, %s\n", string0, string1, string2);
  sprintf(file_name_file, "%s/repo/files/%s/%s/%s/%s.%s.10240", string4, string0, string1, string2, pfile_sha1, pfile_md5);
  sprintf(file_name_gold, "%s/repo/gold/%s/%s/%s/%s.%s.10240", string4, string0, string1, string2, pfile_sha1, pfile_md5);
  int existed = file_dir_existed(file_name_file);
  CU_ASSERT_EQUAL(existed, 1); /* the file into repo? */
  if (existed)
  {
    RemoveDir(file_name_file);
  }
  existed = 0;
  existed = file_dir_existed(file_name_gold);
  CU_ASSERT_EQUAL(existed, 1); /* the file into repo? */
  //printf("file_name_file, file_name_gold are:%s,%s\n", file_name_file, file_name_gold);
  if (existed)
  {
    RemoveDir(file_name_gold);
  }
  PQclear(result);
  //printf("testDBLoadGold end\n");
}

CU_TestInfo testcases_DBLoadGold[] =
{
#if 0
#endif
{"DBLoadGold:Insert", testDBLoadGold},
  CU_TEST_INFO_NULL
};