Skip to content

Commit a54ede2

Browse files
committed
INSERT and DELETE working
1 parent 62cea5a commit a54ede2

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

query-builder/src/test/java/com/datastax/oss/driver/api/querybuilder/delete/DeleteSelectorTest.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424

2525
import org.junit.Test;
2626

27+
import java.util.Arrays;
28+
2729
public class DeleteSelectorTest {
2830

2931
@Test
@@ -34,6 +36,12 @@ public void should_generate_column_deletion() {
3436
.hasCql("DELETE v FROM ks.foo WHERE k=?");
3537
}
3638

39+
@Test
40+
public void should_generate_vector_deletion() {
41+
assertThat(deleteFrom("foo").column("v").whereColumn("k").isEqualTo(literal(Arrays.asList(0.1, 0.2))))
42+
.hasCql("DELETE v FROM foo WHERE k=[0.1,0.2]");
43+
}
44+
3745
@Test
3846
public void should_generate_field_deletion() {
3947
assertThat(

query-builder/src/test/java/com/datastax/oss/driver/api/querybuilder/insert/RegularInsertTest.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@
2626
import com.datastax.oss.driver.api.querybuilder.term.Term;
2727
import com.datastax.oss.driver.internal.querybuilder.insert.DefaultInsert;
2828
import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableMap;
29+
30+
import java.util.Arrays;
2931
import java.util.Map;
3032
import org.junit.Test;
3133

@@ -41,6 +43,12 @@ public void should_generate_column_assignments() {
4143
.hasCql("INSERT INTO foo (a,b) VALUES (?,?)");
4244
}
4345

46+
@Test
47+
public void should_generate_vector_literals() {
48+
assertThat(insertInto("foo").value("a", literal(Arrays.asList(0.1, 0.2, 0.3))))
49+
.hasCql("INSERT INTO foo (a) VALUES ([0.1,0.2,0.3])");
50+
}
51+
4452
@Test
4553
public void should_keep_last_assignment_if_column_listed_twice() {
4654
assertThat(

0 commit comments

Comments
 (0)