fixes for null check

Impetus · Sep 21, 2012 · 3284d87 · 3284d87
1 parent d1f61b0
commit 3284d87
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/kundera-tests/src/test/java/com/impetus/kundera/tests/cli/CassandraCli.java b/kundera-tests/src/test/java/com/impetus/kundera/tests/cli/CassandraCli.java
@@ -286,16 +286,19 @@ public static void truncateColumnFamily(String keyspace, String... columns)
     {
         try
         {
-            client.set_keyspace(keyspace);
-            for (String column : columns)
+            if (client != null)
             {
-                if(columnFamilyExist(column, keyspace))
+                client.set_keyspace(keyspace);
+                for (String column : columns)
                 {
-                    client.truncate(column);
+                    if (columnFamilyExist(column, keyspace))
+                    {
+                        client.truncate(column);
+                    }
                 }
             }
         }
-        catch(IllegalArgumentException iex)
+        catch (IllegalArgumentException iex)
         {
             // do nothing.
         }