Skip to content

Commit e2639e3

Browse files
committed
Fixes Cassandra Tests
1 parent 3260887 commit e2639e3

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

cassandra-driver/src/test/java/org/jnosql/diana/cassandra/column/CassandraColumnFamilyManagerTest.java

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848
import static java.util.Collections.singletonList;
4949
import static java.util.stream.Collectors.toList;
5050
import static org.hamcrest.Matchers.containsInAnyOrder;
51+
import static org.jnosql.diana.api.column.ColumnCondition.eq;
5152
import static org.jnosql.diana.cassandra.column.Constants.COLUMN_FAMILY;
5253
import static org.jnosql.diana.cassandra.column.Constants.KEY_SPACE;
5354
import static org.junit.Assert.assertEquals;
@@ -113,7 +114,7 @@ public void shouldInsertColumnsWithConsistencyLevel() {
113114
public void shouldFindById() {
114115

115116
columnEntityManager.insert(getColumnFamily());
116-
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(ColumnCondition.eq(Columns.of("id", 10L)));
117+
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(eq(Columns.of("id", 10L)));
117118
List<ColumnEntity> columnEntity = columnEntityManager.select(query);
118119
assertFalse(columnEntity.isEmpty());
119120
List<Column> columns = columnEntity.get(0).getColumns();
@@ -126,7 +127,7 @@ public void shouldFindById() {
126127
public void shouldFindByIdWithConsistenceLevel() {
127128

128129
columnEntityManager.insert(getColumnFamily());
129-
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(ColumnCondition.eq(Columns.of("id", 10L)));
130+
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(eq(Columns.of("id", 10L)));
130131
List<ColumnEntity> columnEntity = columnEntityManager.select(query, CONSISTENCY_LEVEL);
131132
assertFalse(columnEntity.isEmpty());
132133
List<Column> columns = columnEntity.get(0).getColumns();
@@ -160,7 +161,7 @@ public void shouldPrepareStatment() {
160161
public void shouldDeleteColumnFamily() {
161162
columnEntityManager.insert(getColumnFamily());
162163
ColumnEntity.of(COLUMN_FAMILY, singletonList(Columns.of("id", 10L)));
163-
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(ColumnCondition.eq(Columns.of("id", 10L)));
164+
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(eq(Columns.of("id", 10L)));
164165
columnEntityManager.delete(ColumnDeleteQuery.of(query.getColumnFamily(), query.getCondition().get()));
165166
List<ColumnEntity> entities = columnEntityManager.cql("select * from newKeySpace.newColumnFamily where id=10;");
166167
Assert.assertTrue(entities.isEmpty());
@@ -170,7 +171,7 @@ public void shouldDeleteColumnFamily() {
170171
public void shouldDeleteColumnFamilyWithConsistencyLevel() {
171172
columnEntityManager.insert(getColumnFamily());
172173
ColumnEntity.of(COLUMN_FAMILY, singletonList(Columns.of("id", 10L)));
173-
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(ColumnCondition.eq(Columns.of("id", 10L)));
174+
ColumnQuery query = ColumnQuery.of(COLUMN_FAMILY).and(eq(Columns.of("id", 10L)));
174175
columnEntityManager.delete(ColumnDeleteQuery.of(query.getColumnFamily(), query.getCondition().get()), CONSISTENCY_LEVEL);
175176
List<ColumnEntity> entities = columnEntityManager.cql("select * from newKeySpace.newColumnFamily where id=10;");
176177
Assert.assertTrue(entities.isEmpty());
@@ -236,6 +237,8 @@ public void shouldSupportAnUDTElement() {
236237
entity.add(udt);
237238
columnEntityManager.insert(entity);
238239
ColumnQuery query = ColumnQuery.of("users");
240+
query.with(eq(Column.of("nickname", "Ioda")));
241+
239242
ColumnEntity columnEntity = columnEntityManager.singleResult(query).get();
240243
Column column = columnEntity.find("name").get();
241244
udt = UDT.class.cast(column);
@@ -257,7 +260,7 @@ public void shouldSupportDate() {
257260
entity.add(Column.of("dateEnd", dateEnd));
258261
columnEntityManager.insert(entity);
259262
ColumnQuery query = ColumnQuery.of("history");
260-
query.and(ColumnCondition.eq(Column.of("name", "World war II")));
263+
query.and(eq(Column.of("name", "World war II")));
261264
ColumnEntity entity1 = columnEntityManager.singleResult(query).get();
262265
Assert.assertNotNull(entity1);
263266

0 commit comments

Comments
 (0)