Make authentication testable

Author: lexmag

.travis.yml

@@ -10,19 +10,25 @@ elixir:
 
 env:
   matrix:
-    - CASSANDRA_VERSION="2.2.8"
-    - CASSANDRA_VERSION="3.9"
+    - CASSANDRA_VERSION=2.2.8 AUTHENTICATION=true
+    - CASSANDRA_VERSION=2.2.8
+    - CASSANDRA_VERSION=3.9 AUTHENTICATION=true
+    - CASSANDRA_VERSION=3.9
 
 matrix:
   include:
     otp_release: 18.3
     elixir: 1.4.0
-    env: CASSANDRA_VERSION="3.9"
+    env: CASSANDRA_VERSION=3.9
 
 # These steps are taken from: http://cassandra.apache.org/doc/latest/getting_started/installing.html#installation-from-binary-tarball-files.
 before_install:
   - java -version
   - wget https://archive.apache.org/dist/cassandra/$CASSANDRA_VERSION/apache-cassandra-$CASSANDRA_VERSION-bin.tar.gz
   - tar -xzf apache-cassandra-$CASSANDRA_VERSION-bin.tar.gz
+  - |-
+    if [ $AUTHENTICATION = true ]; then
+      sed -i -e "s/\(authenticator: \)AllowAllAuthenticator/\1PasswordAuthenticator/" apache-cassandra-$CASSANDRA_VERSION/conf/cassandra.yaml
+    fi
   - sh apache-cassandra-$CASSANDRA_VERSION/bin/cassandra
   - for try in {1..12}; do (nodetool version > /dev/null 2>&1 && break) || sleep 5; done

test/integration/authentication_test.exs

@@ -0,0 +1,12 @@
+defmodule AuthenticationTest do
+  start_options = [
+    authentication: {Xandra.Authenticator.Password, [username: "cassandra", password: "cassandra"]},
+  ]
+  use XandraTest.IntegrationCase, start_options: start_options
+
+  @moduletag :authentication
+
+  test "challenge is passed", %{conn: conn, keyspace: keyspace} do
+    assert Xandra.execute!(conn, "USE #{keyspace}")
+  end
+end

test/test_helper.exs

@@ -1,31 +1,38 @@
-ExUnit.start()
+if System.get_env("AUTHENTICATION") == "true" do
+  ExUnit.start(exclude: [:test], include: [:authentication])
+else
+  ExUnit.start()
+end
 
 defmodule XandraTest.IntegrationCase do
   use ExUnit.CaseTemplate
 
-  using do
-    quote do
+  using options do
+    start_options = Keyword.get(options, :start_options, [])
+    quote [bind_quoted: [start_options: start_options, case_template: __MODULE__]] do
       setup_all do
         keyspace = "xandra_test_" <> String.replace(inspect(__MODULE__), ".", "")
-        unquote(__MODULE__).setup_keyspace(keyspace)
+        start_options = unquote(start_options)
+        case_template = unquote(case_template)
 
+        case_template.setup_keyspace(keyspace, start_options)
         on_exit(fn ->
-          unquote(__MODULE__).drop_keyspace(keyspace)
+          case_template.drop_keyspace(keyspace, start_options)
         end)
 
-        %{keyspace: keyspace}
+        %{keyspace: keyspace, start_options: start_options}
       end
     end
   end
 
-  setup %{keyspace: keyspace} do
-    {:ok, conn} = Xandra.start_link()
+  setup %{keyspace: keyspace, start_options: start_options} do
+    {:ok, conn} = Xandra.start_link(start_options)
     Xandra.execute!(conn, "USE #{keyspace}", [])
     %{conn: conn}
   end
 
-  def setup_keyspace(keyspace) do
-    {:ok, conn} = Xandra.start_link()
+  def setup_keyspace(keyspace, start_options) do
+    {:ok, conn} = Xandra.start_link(start_options)
     Xandra.execute!(conn, "DROP KEYSPACE IF EXISTS #{keyspace}", [])
     statement = """
     CREATE KEYSPACE #{keyspace}
@@ -34,8 +41,8 @@ defmodule XandraTest.IntegrationCase do
     Xandra.execute!(conn, statement, [])
   end
 
-  def drop_keyspace(keyspace) do
-    {:ok, conn} = Xandra.start_link()
+  def drop_keyspace(keyspace, start_options) do
+    {:ok, conn} = Xandra.start_link(start_options)
     Xandra.execute!(conn, "DROP KEYSPACE IF EXISTS #{keyspace}", [])
   end
 end