Remove table from repository.

Author: MartinRixham

src/repository/repository.h

@@ -19,6 +19,8 @@ namespace repository
 			virtual bool has_table(const std::string &table_name) const = 0;
 
 			virtual table::table read_table(const std::string &table_name) const = 0;
+
+			virtual void delete_table(const std::string &table_name) = 0;
 	};
 }
 

src/repository/rocksdb_repository.cpp

@@ -68,6 +68,22 @@ table::table repository::rocksdb_repository::read_table(const std::string &table
 	}
 }
 
+void repository::rocksdb_repository::delete_table(const std::string &table_name)
+{
+	std::string value;
+
+	database->Get(rocksdb::ReadOptions(), "TABLE_" + table_name, &value);
+
+	if (value.empty())
+	{
+		return;
+	}
+	else
+	{
+		database->Delete(rocksdb::WriteOptions(), "TABLE_" + table_name);
+	}
+}
+
 repository::rocksdb_repository::~rocksdb_repository()
 {
 	delete database;

src/repository/rocksdb_repository.h

@@ -20,6 +20,8 @@ namespace repository
 
 			table::table read_table(const std::string &table_name) const override;
 
+			void delete_table(const std::string &table_name) override;
+
 			~rocksdb_repository();
 	};
 }

test/repository/fake_repository.cpp

@@ -24,19 +24,27 @@ std::set<table::table> repository::fake_repository::list_tables() const
 	return table_list;
 }
 
-bool repository::fake_repository::has_table(const std::string &tableName) const
+bool repository::fake_repository::has_table(const std::string &table_name) const
 {
-	return tables.count(tableName) > 0;
+	return tables.count(table_name) > 0;
 }
 
-table::table repository::fake_repository::read_table(const std::string &tableName) const
+table::table repository::fake_repository::read_table(const std::string &table_name) const
 {
-	if (tables.count(tableName))
+	if (tables.count(table_name))
 	{
-		return table::to_table(tables.at(tableName));
+		return table::to_table(tables.at(table_name));
 	}
 	else
 	{
-		return table::invalid_table("Fake data store didn't contain table " + tableName + ".");
+		return table::invalid_table("Fake data store didn't contain table " + table_name + ".");
+	}
+}
+
+void repository::fake_repository::delete_table(const std::string &table_name)
+{
+	if (tables.count(table_name))
+	{
+		return;
 	}
 }

test/repository/fake_repository.h

@@ -22,6 +22,8 @@ namespace repository
 			bool has_table(const std::string &tableName) const override;
 
 			table::table read_table(const std::string &tableName) const override;
+
+			void delete_table(const std::string &table_name) override;
 	};
 }
 

test/repository/rocksdb_repository_test.cpp

@@ -85,3 +85,25 @@ TEST_F(repository_test, does_not_have_table)
 
 	EXPECT_FALSE(has_table);
 }
+
+TEST_F(repository_test, does_not_delete_invalid_table)
+{
+	repository.create_table(table::valid_table("a table", std::vector<std::string>()));
+
+	repository.delete_table("not a table");
+
+	bool has_table = repository.has_table("a table");
+
+	EXPECT_TRUE(has_table);
+}
+
+TEST_F(repository_test, delete_table_from_repository)
+{
+	repository.create_table(table::valid_table("a table", std::vector<std::string>()));
+
+	repository.delete_table("a table");
+
+	bool has_table = repository.has_table("a table");
+
+	EXPECT_FALSE(has_table);
+}