Created test fixture for SQLPTRepresentationTest3hotfix/store_policy_db_in_storage_folder

1 files changed, 32 insertions, 36 deletions

diff --git a/src/components/policy/test/sql_pt_representation_test.cc b/src/components/policy/test/sql_pt_representation_test.cc
index fbbe14c755..42f7a9cc33 100644
--- a/src/components/policy/test/sql_pt_representation_test.cc
+++ b/src/components/policy/test/sql_pt_representation_test.cc
@@ -386,6 +386,25 @@ class SQLPTRepresentationTest2 : public ::testing::Test {
   const uint16_t kAttemptsToOpenPolicyDB;
 };
 
+class SQLPTRepresentationTest3 : public ::testing::Test {
+ protected:
+  SQLPTRepresentationTest3() : kAppStorageFolder("storage") {}
+
+  void SetUp() OVERRIDE {
+    file_system::CreateDirectory(kAppStorageFolder);
+    reps = new SQLPTRepresentation;
+  }
+
+  void TearDown() OVERRIDE {
+    file_system::RemoveDirectory(kAppStorageFolder, true);
+    delete reps;
+  }
+
+  SQLPTRepresentation* reps;
+  NiceMock<policy_handler_test::MockPolicySettings> policy_settings_;
+  const std::string kAppStorageFolder;
+};
+
 // {AKozoriz} : Unknown behavior (must try 8 times, tried 2 and opened)
 TEST_F(SQLPTRepresentationTest2,
        DISABLED_OpenAttemptTimeOut_ExpectCorrectNumber) {
@@ -992,56 +1011,38 @@ TEST_F(SQLPTRepresentationTest,
   EXPECT_EQ("EMERGENCY", priority);
 }
 
-namespace {
-const std::string kAppStorageFolder = "storage";
-}
-
-TEST(SQLPTRepresentationTest3, Init_InitNewDataBase_ExpectResultSuccess) {
-  file_system::CreateDirectory(kAppStorageFolder);
+TEST_F(SQLPTRepresentationTest3, Init_InitNewDataBase_ExpectResultSuccess) {
   // Arrange
-  NiceMock<policy_handler_test::MockPolicySettings> policy_settings_;
-  SQLPTRepresentation* reps;
-  reps = new SQLPTRepresentation;
-  // Checks
   ON_CALL(policy_settings_, app_storage_folder())
       .WillByDefault(ReturnRef(kAppStorageFolder));
+  // Checks
   EXPECT_EQ(::policy::SUCCESS, reps->Init(&policy_settings_));
   EXPECT_EQ(::policy::EXISTS, reps->Init(&policy_settings_));
   reps->RemoveDB();
-  delete reps;
-  file_system::RemoveDirectory(kAppStorageFolder);
 }
 
-TEST(SQLPTRepresentationTest3,
-     Init_TryInitNotExistingDataBase_ExpectResultFail) {
+TEST_F(SQLPTRepresentationTest3,
+       Init_TryInitNotExistingDataBase_ExpectResultFail) {
   const std::string kEmptyDirectory = "";
-  file_system::CreateDirectory(kAppStorageFolder);
   // Arrange
-  NiceMock<policy_handler_test::MockPolicySettings> policy_settings_;
   ON_CALL(policy_settings_, app_storage_folder())
       .WillByDefault(ReturnRef(kEmptyDirectory));
-  SQLPTRepresentation reps;
-  (reps.db())->set_path("/home/");
+  (reps->db())->set_path("/home/");
   // Check
-  EXPECT_EQ(::policy::FAIL, reps.Init(&policy_settings_));
-  file_system::RemoveDirectory(kAppStorageFolder);
+  EXPECT_EQ(::policy::FAIL, reps->Init(&policy_settings_));
 }
 
-TEST(SQLPTRepresentationTest3,
-     Close_InitNewDataBaseThenClose_ExpectResultSuccess) {
-  file_system::CreateDirectory(kAppStorageFolder);
+TEST_F(SQLPTRepresentationTest3,
+       Close_InitNewDataBaseThenClose_ExpectResultSuccess) {
   // Arrange
-  NiceMock<policy_handler_test::MockPolicySettings> policy_settings_;
   ON_CALL(policy_settings_, app_storage_folder())
       .WillByDefault(ReturnRef(kAppStorageFolder));
-  SQLPTRepresentation reps;
-  EXPECT_EQ(::policy::SUCCESS, reps.Init(&policy_settings_));
-  EXPECT_TRUE(reps.Close());
+  EXPECT_EQ(::policy::SUCCESS, reps->Init(&policy_settings_));
+  EXPECT_TRUE(reps->Close());
   utils::dbms::SQLError error(utils::dbms::Error::OK);
   // Checks
-  EXPECT_EQ(error.number(), (reps.db()->LastError().number()));
-  reps.RemoveDB();
-  file_system::RemoveDirectory(kAppStorageFolder);
+  EXPECT_EQ(error.number(), (reps->db()->LastError().number()));
+  reps->RemoveDB();
 }
 
 TEST_F(SQLPTRepresentationTest,
@@ -1484,11 +1485,8 @@ TEST_F(SQLPTRepresentationTest,
   EXPECT_EQ(0, dbms->FetchOneInt(query_select));
 }
 
-TEST(SQLPTRepresentationTest3, RemoveDB_RemoveDB_ExpectFileDeleted) {
-  file_system::CreateDirectory(kAppStorageFolder);
+TEST_F(SQLPTRepresentationTest3, RemoveDB_RemoveDB_ExpectFileDeleted) {
   // Arrange
-  policy_handler_test::MockPolicySettings policy_settings_;
-  SQLPTRepresentation* reps = new SQLPTRepresentation;
   ON_CALL(policy_settings_, app_storage_folder())
       .WillByDefault(ReturnRef(kAppStorageFolder));
   EXPECT_EQ(::policy::SUCCESS, reps->Init(&policy_settings_));
@@ -1498,8 +1496,6 @@ TEST(SQLPTRepresentationTest3, RemoveDB_RemoveDB_ExpectFileDeleted) {
   reps->RemoveDB();
   // Check
   EXPECT_FALSE(file_system::FileExists(path));
-  delete reps;
-  file_system::RemoveDirectory(kAppStorageFolder);
 }
 
 // TODO {AKozoriz} : Snapshot must have module meta section, but test