Merge branch 'main' into lucasmmota/dae-63-handle-exception

quintoandar · Jan 27, 2021 · 74f975a · 74f975a
2 parents d247abe + 41eca7b
commit 74f975a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 12 deletions.
diff --git a/docs/source/getstarted.md b/docs/source/getstarted.md
@@ -37,4 +37,5 @@ the [`HiveMetastoreClient`](https://github.com/quintoandar/hive-metastore-client
 
 - [`add_columns_to_table`](https://hive-metastore-client.readthedocs.io/en/latest/hive_metastore_client.html#hive_metastore_client.hive_metastore_client.HiveMetastoreClient.add_columns_to_table)
 - [`drop_columns_from_table`](https://hive-metastore-client.readthedocs.io/en/latest/hive_metastore_client.html#hive_metastore_client.hive_metastore_client.HiveMetastoreClient.drop_columns_from_table)
-- [`add_partitions_if_not_exists`](https://hive-metastore-client.readthedocs.io/en/latest/hive_metastore_client.html#hive_metastore_client.hive_metastore_client.HiveMetastoreClient.add_partitions_if_not_exists)
+- [`add_partitions_if_not_exists`](https://hive-metastore-client.readthedocs.io/en/latest/hive_metastore_client.html#hive_metastore_client.hive_metastore_client.HiveMetastoreClient.add_partitions_if_not_exists)
+- [`create_database_if_not_exists`](https://hive-metastore-client.readthedocs.io/en/latest/hive_metastore_client.html#hive_metastore_client.hive_metastore_client.HiveMetastoreClient.create_database_if_not_exists)
diff --git a/hive_metastore_client/hive_metastore_client.py b/hive_metastore_client/hive_metastore_client.py
@@ -144,7 +144,7 @@ def add_partitions_if_not_exists(
         except AlreadyExistsException:
             pass
 
-    def create_database_if_not_exists(self, database: Database) -> bool:
+    def create_database_if_not_exists(self, database: Database) -> None:
         """
         Creates the table in Hive Metastore if it does not exist.
 
@@ -156,9 +156,8 @@ def create_database_if_not_exists(self, database: Database) -> bool:
         """
         try:
             self.create_database(database)
-            return True
         except AlreadyExistsException:
-            return False
+            pass
 
     @staticmethod
     def _format_partitions_location(

diff --git a/tests/unit/hive_metastore_client/test_hive_metastore_client.py b/tests/unit/hive_metastore_client/test_hive_metastore_client.py
@@ -277,12 +277,9 @@ def test_create_database_if_not_exists_with_nonexistent_database(
         mocked_database_obj = Mock()
 
         # act
-        return_value = hive_metastore_client.create_database_if_not_exists(
-            mocked_database_obj
-        )
+        hive_metastore_client.create_database_if_not_exists(mocked_database_obj)
 
         # assert
-        assert return_value
         mocked_create_database.assert_called_once_with(mocked_database_obj)
 
     @mock.patch.object(HiveMetastoreClient, "create_database")
@@ -294,10 +291,7 @@ def test_create_database_if_not_exists_with_existent_database(
         mocked_create_database.side_effect = AlreadyExistsException()
 
         # act
-        return_value = hive_metastore_client.create_database_if_not_exists(
-            mocked_database_obj
-        )
+        hive_metastore_client.create_database_if_not_exists(mocked_database_obj)
 
         # assert
-        assert not return_value
         mocked_create_database.assert_called_once_with(mocked_database_obj)