1 #include <boost/test/unit_test.hpp>
8 extern std::string sqlHost, sqlUser, sqlPass, sqlDB;
10 BOOST_AUTO_TEST_SUITE( TestSQL )
12 BOOST_AUTO_TEST_CASE( testSQL ) {
13 std::ifstream conf("config.txt");
15 BOOST_WARN_MESSAGE( 0, "Config file is missing. Exiting." );
18 BOOST_REQUIRE( parseConfig("config.txt") == 0 );
19 std::shared_ptr<MySQLJobProvider> jp( new MySQLJobProvider( sqlHost, sqlUser, sqlPass, sqlDB ) );
20 BOOST_REQUIRE( jp->query( "TRUNCATE TABLE profiles" ).first == 0 );
21 BOOST_REQUIRE( jp->query( "TRUNCATE TABLE certs" ).first == 0 );
22 BOOST_REQUIRE( jp->query( "TRUNCATE TABLE certAvas" ).first == 0 );
23 BOOST_REQUIRE( jp->query( "TRUNCATE TABLE subjectAlternativeNames" ).first == 0 );
24 BOOST_REQUIRE( jp->query( "TRUNCATE TABLE jobs" ).first == 0 );
25 BOOST_REQUIRE( jp->query( "INSERT INTO profiles SET id='1', keyname='assured', keyUsage='', extendedKeyUsage='', name='assured'" ).first == 0 );
26 BOOST_REQUIRE( jp->query( "INSERT INTO jobs SET task='sign', targetId='1'" ).first == 0 );
28 std::shared_ptr<Job> job = jp->fetchJob();
31 BOOST_REQUIRE_EQUAL( job->target, "1" );
32 BOOST_REQUIRE_EQUAL( job->task, "sign" );
35 std::shared_ptr<TBSCertificate> cert = jp->fetchTBSCert(job);
36 BOOST_REQUIRE( !cert );
37 BOOST_REQUIRE( jp->query( "INSERT INTO certs SET csr_type='CSR', id='1', profile='1'" ).first == 0 );
38 BOOST_REQUIRE( jp->query( "INSERT INTO subjectAlternativeNames SET certId='1', contents='example.org', type='DNS'" ).first == 0 );
39 BOOST_REQUIRE( jp->query( "INSERT INTO certAvas SET certid='1', name='CN', value='example.org'" ).first == 0 );
40 cert = jp->fetchTBSCert(job);
41 BOOST_REQUIRE( cert );
43 std::shared_ptr<SignedCertificate> fcert( new SignedCertificate() );
44 fcert->certificate="CERT";
46 fcert->crt_name="crt.name.crt";
47 fcert->ca_name="assured";
48 jp->writeBack( job, fcert );
50 BOOST_REQUIRE( !jp->fetchJob() );
51 BOOST_REQUIRE( jp->query( "INSERT INTO jobs SET task='revoke', targetId='1'" ).first == 0 );
53 BOOST_REQUIRE_EQUAL( job->target, "1" );
54 BOOST_REQUIRE_EQUAL( job->task, "revoke" );
55 std::pair<std::string, std::string> revocationInfo = jp->getRevocationInfo( job );
56 BOOST_REQUIRE_EQUAL( revocationInfo.first, "1234");
57 BOOST_REQUIRE_EQUAL( revocationInfo.second, "assured");
58 jp->writeBackRevocation( job, "2000-01-01 01:01:01" );
62 BOOST_AUTO_TEST_CASE( testSQLDisconnected ) {
63 std::ifstream conf("config.txt");
65 BOOST_WARN_MESSAGE( 0, "Config file is missing. Exiting." );
69 //BOOST_REQUIRE( parseConfig("config.txt") == 0 );
70 std::shared_ptr<MySQLJobProvider> jp( new MySQLJobProvider( sqlHost, sqlUser, sqlPass, sqlDB ) );
73 BOOST_REQUIRE( jp->query("SELECT 1").first);
74 BOOST_REQUIRE_THROW( jp->escape_string("uia"), const char * );
75 BOOST_REQUIRE_THROW( jp->finishJob(std::shared_ptr<Job>()), const char * );
76 BOOST_REQUIRE_THROW( jp->failJob(std::shared_ptr<Job>()), const char * );
79 BOOST_AUTO_TEST_SUITE_END()