diff --git a/lib/kafkaex_lag_exporter/behaviours/kafka_utils_behaviour.ex b/lib/kafkaex_lag_exporter/behaviours/kafka_utils_behaviour.ex
new file mode 100644
index 0000000..3f69662
--- /dev/null
+++ b/lib/kafkaex_lag_exporter/behaviours/kafka_utils_behaviour.ex
@@ -0,0 +1,43 @@
+defmodule KafkaexLagExporter.KafkaUtils.Behaviour do
+  @moduledoc false
+
+  @callback connection(atom) :: {list({charlist, non_neg_integer}), Keyword.t()}
+
+  @callback resolve_offsets(binary, :earliest | :latest, atom) ::
+              list({non_neg_integer, integer})
+
+  @callback fetch_committed_offsets(binary, binary, atom) ::
+              list({non_neg_integer, non_neg_integer})
+
+  @callback lag(binary, binary, atom) :: list({non_neg_integer, integer})
+
+  @callback lag_total(binary, binary, atom) :: non_neg_integer
+
+  @callback get_consumer_group_names({host :: atom, port :: non_neg_integer}) :: list(binary)
+
+  @callback topic_names_for_consumer_groups(
+              {host :: atom, port :: non_neg_integer},
+              list(binary),
+              list(binary)
+            ) :: list(binary)
+
+  def connection(client), do: impl().connection(client)
+
+  def resolve_offsets(topic, type, client), do: impl().resolve_offsets(topic, type, client)
+
+  def fetch_committed_offsets(topic, consumer_group, client),
+    do: impl().fetch_committed_offsets(topic, consumer_group, client)
+
+  def lag(topic, consumer_group, client), do: impl().lag(topic, consumer_group, client)
+
+  def lag_total(topic, consumer_group, client),
+    do: impl().lag_total(topic, consumer_group, client)
+
+  def get_consumer_group_names({host, port}), do: impl().get_consumer_group_names({host, port})
+
+  def topic_names_for_consumer_groups(endpoint, list, consumer_group_names),
+    do: impl().topic_names_for_consumer_groups(endpoint, list, consumer_group_names)
+
+  defp impl,
+    do: Application.get_env(:kafkaex_lag_exporter, :kafka_utils, KafkaexLagExporter.KafkaUtils)
+end
diff --git a/lib/kafkaex_lag_exporter/behaviours/kafka_wrapper_behaviour.ex b/lib/kafkaex_lag_exporter/behaviours/kafka_wrapper_behaviour.ex
new file mode 100644
index 0000000..bd5e227
--- /dev/null
+++ b/lib/kafkaex_lag_exporter/behaviours/kafka_wrapper_behaviour.ex
@@ -0,0 +1,45 @@
+defmodule KafkaexLagExporter.KafkaWrapper.Behaviour do
+  @moduledoc false
+
+  @type client() :: :brod.client()
+  @type endpoint() :: :brod.endpoint()
+  @type conn_config() :: :brod.conn_config()
+  @type group_id() :: :kpro.group_id()
+  @type kpro_struct() :: :kpro.struct()
+  @type topic() :: :kpro.topic()
+  @type partition() :: :kpro.partition()
+  @type offset_time() :: :kpro.msg_ts() | :earliest | :latest
+  @type kpro_config() :: [{atom, term()}] | :kpro.conn_config()
+  @type offset() :: :kpro.offset()
+  @type cg() :: any()
+
+  @callback fetch_committed_offsets(list(endpoint()), conn_config(), group_id()) ::
+              {:ok, list(kpro_struct())} | {:error, any()}
+
+  @callback get_partitions_count(client(), topic()) :: {:ok, pos_integer} | {:error, any()}
+
+  @callback resolve_offset(list([endpoint()]), topic(), partition(), offset_time(), kpro_config()) ::
+              {:ok, offset()} | {:error, any()}
+
+  @callback list_all_groups(list(endpoint()), conn_config()) ::
+              list({endpoint(), list(cg())} | {:error, any()})
+
+  @callback describe_groups(endpoint(), conn_config(), list(group_id())) ::
+              {:ok, list(kpro_struct())} | {:error, any()}
+
+  def fetch_committed_offsets(endpoints, sock_opts, consumer_group),
+    do: impl().fetch_committed_offsets(endpoints, sock_opts, consumer_group)
+
+  def get_partitions_count(client, topic), do: impl().get_partitions_count(client, topic)
+
+  def resolve_offset(endpoints, topic, i, type, sock_opts),
+    do: impl().resolve_offset(endpoints, topic, i, type, sock_opts)
+
+  def list_all_groups(endpoints, sock_opts),
+    do: impl().list_all_groups(endpoints, sock_opts)
+
+  def describe_groups(endpoint, sock_opts, consumer_group_names),
+    do: impl().describe_groups(endpoint, sock_opts, consumer_group_names)
+
+  defp impl, do: Application.get_env(:kafkaex_lag_exporter, :kafka_wrapper, :brod)
+end