utils.hpp

Go to the documentation of this file.

#pragma once
 
/// @file userver/storages/mysql/tests/utils.hpp
/// @brief Utilities for testing logic working with MySQL.
 
#include <userver/clients/dns/resolver.hpp>
#include <userver/engine/deadline.hpp>
#include <userver/utils/uuid4.hpp>
 
#include <userver/storages/mysql.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace storages::mysql::tests {
 
std::uint32_t GetMysqlPort();
 
std::chrono::system_clock::time_point ToMariaDBPrecision(std::chrono::system_clock::time_point tp);
 
class ClusterWrapper final {
public:
    ClusterWrapper();
    ~ClusterWrapper();
 
    storages::mysql::Cluster& operator*() const;
    storages::mysql::Cluster* operator->() const;
 
    engine::Deadline GetDeadline() const;
 
    template <typename... Args>
    StatementResultSet DefaultExecute(const std::string& query, const Args&... args) const;
 
private:
    clients::dns::Resolver resolver_;
    std::shared_ptr<storages::mysql::Cluster> cluster_;
 
    engine::Deadline deadline_;
};
 
template <typename... Args>
StatementResultSet ClusterWrapper::DefaultExecute(const std::string& query, const Args&... args) const {
    return cluster_->Execute(ClusterHostType::kPrimary, query, args...);
}
 
class TmpTable final {
public:
    explicit TmpTable(std::string_view definition);
    TmpTable(ClusterWrapper& cluster, std::string_view definition);
    ~TmpTable();
 
    template <typename... Args>
    std::string FormatWithTableName(std::string_view source, const Args&... args) const;
 
    template <typename... Args>
    StatementResultSet DefaultExecute(std::string_view source, const Args&... args);
 
    ClusterWrapper& GetCluster() const;
 
    Transaction Begin();
 
    engine::Deadline GetDeadline() const;
 
private:
    static constexpr std::string_view kCreateTableQueryTemplate = "CREATE TABLE {} ({})";
 
    void CreateTable(std::string_view definition);
 
    std::optional<ClusterWrapper> owned_cluster_;
    ClusterWrapper& cluster_;
    std::string table_name_;
};
 
template <typename... Args>
std::string TmpTable::FormatWithTableName(std::string_view source, const Args&... args) const {
    return fmt::format(fmt::runtime(source), table_name_, args...);
}
 
template <typename... Args>
StatementResultSet TmpTable::DefaultExecute(std::string_view source, const Args&... args) {
    return cluster_.DefaultExecute(FormatWithTableName(source, table_name_), args...);
}
 
}  // namespace storages::mysql::tests
 
USERVER_NAMESPACE_END