Merge pull request #195 from datastax/ruby-187_tests

[RUBY-187] Integration tests and cuke features for trigger meta

Author: kishkaru

features/basics/schema_metadata.feature

@@ -80,6 +80,39 @@ Feature: Schema Metadata
       )
       """
 
+  Scenario: Getting trigger metadata
+    Given the following schema:
+      """cql
+      CREATE KEYSPACE simplex WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 3};
+      CREATE TABLE simplex.test_table (a text primary key, b text);
+      CREATE TABLE simplex.audit (key timeuuid, keyspace_name text, table_name text, primary_key text, PRIMARY KEY(key));
+      CREATE TRIGGER trigger1 ON simplex.test_table USING 'org.apache.cassandra.triggers.AuditTrigger';
+      """
+    And the following example:
+      """ruby
+      require 'cassandra'
+
+      cluster = Cassandra.cluster
+
+      # AuditTrigger is available in cassandra/conf/triggers. See cassandra/examples/triggers in Cassandra
+      trigger = cluster.keyspace('simplex').table('test_table').trigger('trigger1')
+      puts trigger.to_cql
+
+      puts ""
+      puts "Name: #{trigger.name}"
+      puts "Table name: #{trigger.table.name}"
+      puts "Class: #{trigger.options['class']}"
+      """
+    When it is executed
+    Then its output should contain:
+      """cql
+      CREATE TRIGGER "trigger1" ON simplex.test_table USING 'org.apache.cassandra.triggers.AuditTrigger';
+
+      Name: trigger1
+      Table name: test_table
+      Class: org.apache.cassandra.triggers.AuditTrigger
+      """
+
   Scenario: Getting index metadata
     Given the following schema:
       """cql

integration/metadata_test.rb

@@ -39,6 +39,12 @@ def setup
     @session.execute("CREATE TABLE simplex.custom (f1 int PRIMARY KEY," \
     " f2 'org.apache.cassandra.db.marshal.CompositeType(org.apache.cassandra.db.marshal.UUIDType,org.apache.cassandra.db.marshal.UTF8Type)')")
     @listener.wait_for_table('simplex', 'custom')
+    @session.execute("CREATE TABLE simplex.test1 (key text, value text, PRIMARY KEY(key))")
+    @listener.wait_for_table('simplex', 'test1')
+    @session.execute("CREATE TABLE simplex.test2 (key text, value text, PRIMARY KEY(key))")
+    @listener.wait_for_table('simplex', 'test2')
+    @session.execute("CREATE TABLE simplex.audit (key timeuuid, keyspace_name text, table_name text, primary_key text, PRIMARY KEY(key))")
+    @listener.wait_for_table('simplex', 'audit')
   end
 
   def teardown
@@ -63,7 +69,7 @@ def test_can_retrieve_keyspace_metadata
     assert_equal 1, ks_meta.replication.options['replication_factor'].to_i
     assert ks_meta.durable_writes?
     assert ks_meta.has_table?('users')
-    assert_equal 4, ks_meta.tables.size
+    assert_equal 7, ks_meta.tables.size
 
     ks_cql = Regexp.new(/CREATE KEYSPACE simplex WITH replication = {'class': 'SimpleStrategy', \
 'replication_factor': '1'} AND durable_writes = true;/)
@@ -342,4 +348,42 @@ def test_table_metadata_contains_extensions
     assert_empty table_meta.options.extensions
   end
 
+  # Test for retrieving trigger metadata
+  #
+  # test_can_retrieve_trigger_metadata tests that all pieces of trigger metadata can be retrieved. It first creates a
+  # simple trigger. It then goes through each piece of the trigger metadata and verifies that each piece is as expected.
+  # It finally creates another trigger with the same name, on a different table and verifies that it is retrieved and
+  # complete.
+  #
+  # @since 3.1.0
+  # @jira_ticket RUBY-187
+  # @expected_result trigger metadata should be retrieved.
+  #
+  # @test_category metadata
+  #
+  def test_can_retrieve_trigger_metadata
+    skip("Triggers were introduced in Cassandra 2.0") if CCM.cassandra_version < '2.0.0'
+
+    # trigger1, on test1 table
+    @session.execute("CREATE TRIGGER trigger1 ON simplex.test1 USING 'org.apache.cassandra.triggers.AuditTrigger'")
+    @listener.wait_for_trigger('simplex', 'test1', 'trigger1')
+
+    assert @cluster.keyspace('simplex').table('test1').has_trigger?('trigger1')
+    trigger_meta = @cluster.keyspace('simplex').table('test1').trigger('trigger1')
+    assert_equal 'trigger1', trigger_meta.name
+    assert_equal 'test1', trigger_meta.table.name
+    assert_equal 'org.apache.cassandra.triggers.AuditTrigger', trigger_meta.options['class']
+
+    # trigger1, on test2 table
+    @session.execute("CREATE TRIGGER trigger1 ON simplex.test2 USING 'org.apache.cassandra.triggers.AuditTrigger'")
+    @listener.wait_for_trigger('simplex', 'test2', 'trigger1')
+
+    assert @cluster.keyspace('simplex').table('test2').has_trigger?('trigger1')
+    trigger_meta2 = @cluster.keyspace('simplex').table('test2').trigger('trigger1')
+    assert_equal 'trigger1', trigger_meta2.name
+    assert_equal 'test2', trigger_meta2.table.name
+    assert_equal 'org.apache.cassandra.triggers.AuditTrigger', trigger_meta2.options['class']
+
+    refute_equal trigger_meta, trigger_meta2
+  end
 end

integration/schema_change_listener.rb

@@ -73,6 +73,13 @@ def wait_for_materialized_view(keyspace_name, view_name, *args)
     end
   end
 
+  def wait_for_trigger(keyspace_name, table_name, trigger_name, *args)
+    self.wait_for_table(keyspace_name, table_name)
+    wait_for_change(keyspace_name, 2) do |ks|
+      ks.table(table_name).has_trigger?(trigger_name, *args)
+    end
+  end
+
   def keyspace_changed(keyspace)
     # This looks a little strange, but here's the idea: if we don't have
     # Condition's for this keyspace, immediately return. Otherwise, for each

support/ccm.rb

@@ -455,6 +455,8 @@ def start(jvm_arg=nil)
 
       options[:load_balancing_policy] = SameOrderLoadBalancingPolicy.new
 
+      enable_triggers
+
       total_attempts = 1
       until @nodes.all?(&:up?) && @cluster && @cluster.hosts.select(&:up?).count == @nodes.size
         attempts = 1
@@ -714,6 +716,17 @@ def change_tombstone_thresholds
       start
     end
 
+    def enable_triggers
+      trigger_root = File.expand_path(File.dirname(__FILE__) + '/../support/triggers')
+      ccm_node_conf_dir = "~/.ccm/ruby-driver-cassandra-#{CCM.cassandra_version}-test-cluster"
+
+      (1..@nodes.size).each do |n|
+        `mkdir #{ccm_node_conf_dir}/node#{n}/conf/triggers`
+        `cp #{trigger_root}/AuditTrigger.properties #{ccm_node_conf_dir}/node#{n}/conf`
+        `cp #{trigger_root}/trigger-example.jar #{ccm_node_conf_dir}/node#{n}/conf/triggers`
+      end
+    end
+
     def setup_schema(schema)
       schema.strip!
       schema.chomp!(";")

support/triggers/AuditTrigger.properties

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+#--
+# Copyright 2013-2016 DataStax, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#++
+
+keyspace=simplex
+table=audit