Search in sources :

Example 1 with Column

use of com.airbnb.spinaltap.mysql.mutation.schema.Column in project SpinalTap by airbnb.

the class MysqlKeyProviderTest method testGetKey.

@Test
public void testGetKey() throws Exception {
    Row row = new Row(TABLE, ImmutableMap.of(ID_COLUMN, new Column(TABLE.getColumns().get(ID_COLUMN), 1234)));
    MysqlMutation mutation = new MysqlInsertMutation(MUTATION_METADATA, row);
    assertEquals("test:users:1234", MysqlKeyProvider.INSTANCE.get(mutation));
}
Also used : Column(com.airbnb.spinaltap.mysql.mutation.schema.Column) Row(com.airbnb.spinaltap.mysql.mutation.schema.Row) Test(org.junit.Test)

Example 2 with Column

use of com.airbnb.spinaltap.mysql.mutation.schema.Column in project SpinalTap by airbnb.

the class MysqlUpdateMutationTest method testAsColumnValues.

@Test
public void testAsColumnValues() throws Exception {
    Table table = new Table(1, "table_name", "db_name", ImmutableList.of(new ColumnMetadata("id", ColumnDataType.LONGLONG, false, 0)), ImmutableList.of());
    Row row = new Row(table, ImmutableMap.of("id", new Column(table.getColumns().get("id"), 2)));
    assertEquals(ImmutableMap.of("id", 2), MysqlUpdateMutation.asColumnValues(row));
}
Also used : ColumnMetadata(com.airbnb.spinaltap.mysql.mutation.schema.ColumnMetadata) Table(com.airbnb.spinaltap.mysql.mutation.schema.Table) Column(com.airbnb.spinaltap.mysql.mutation.schema.Column) Row(com.airbnb.spinaltap.mysql.mutation.schema.Row) Test(org.junit.Test)

Example 3 with Column

use of com.airbnb.spinaltap.mysql.mutation.schema.Column in project SpinalTap by airbnb.

the class RowTest method testSinglePrimaryKey.

@Test
public void testSinglePrimaryKey() throws Exception {
    Table table = new Table(TABLE_ID, TABLE_NAME, DB_NAME, ImmutableList.of(new ColumnMetadata(ID_COLUMN, ColumnDataType.LONGLONG, true, 0), new ColumnMetadata(NAME_COLUMN, ColumnDataType.VARCHAR, false, 1)), ImmutableList.of(ID_COLUMN));
    Row row = new Row(table, ImmutableMap.of(ID_COLUMN, new Column(table.getColumns().get(ID_COLUMN), 1), NAME_COLUMN, new Column(table.getColumns().get(NAME_COLUMN), "Bob")));
    assertEquals("1", row.getPrimaryKeyValue());
}
Also used : ColumnMetadata(com.airbnb.spinaltap.mysql.mutation.schema.ColumnMetadata) Table(com.airbnb.spinaltap.mysql.mutation.schema.Table) Column(com.airbnb.spinaltap.mysql.mutation.schema.Column) Row(com.airbnb.spinaltap.mysql.mutation.schema.Row) Test(org.junit.Test)

Example 4 with Column

use of com.airbnb.spinaltap.mysql.mutation.schema.Column in project SpinalTap by airbnb.

the class RowTest method testCompositePrimaryKey.

@Test
public void testCompositePrimaryKey() throws Exception {
    Table table = new Table(TABLE_ID, TABLE_NAME, DB_NAME, ImmutableList.of(new ColumnMetadata(ID_COLUMN, ColumnDataType.LONGLONG, true, 0), new ColumnMetadata(NAME_COLUMN, ColumnDataType.VARCHAR, true, 1)), ImmutableList.of(ID_COLUMN, NAME_COLUMN));
    Row row = new Row(table, ImmutableMap.of(ID_COLUMN, new Column(table.getColumns().get(ID_COLUMN), 1), NAME_COLUMN, new Column(table.getColumns().get(NAME_COLUMN), "Bob")));
    assertEquals("1Bob", row.getPrimaryKeyValue());
}
Also used : ColumnMetadata(com.airbnb.spinaltap.mysql.mutation.schema.ColumnMetadata) Table(com.airbnb.spinaltap.mysql.mutation.schema.Table) Column(com.airbnb.spinaltap.mysql.mutation.schema.Column) Row(com.airbnb.spinaltap.mysql.mutation.schema.Row) Test(org.junit.Test)

Example 5 with Column

use of com.airbnb.spinaltap.mysql.mutation.schema.Column in project SpinalTap by airbnb.

the class RowTest method testNullPrimaryKey.

@Test
public void testNullPrimaryKey() throws Exception {
    Table table = new Table(TABLE_ID, TABLE_NAME, DB_NAME, ImmutableList.of(new ColumnMetadata(ID_COLUMN, ColumnDataType.LONGLONG, true, 0)), ImmutableList.of(ID_COLUMN));
    Row row = new Row(table, ImmutableMap.of(ID_COLUMN, new Column(table.getColumns().get(ID_COLUMN), null)));
    assertEquals("null", row.getPrimaryKeyValue());
}
Also used : ColumnMetadata(com.airbnb.spinaltap.mysql.mutation.schema.ColumnMetadata) Table(com.airbnb.spinaltap.mysql.mutation.schema.Table) Column(com.airbnb.spinaltap.mysql.mutation.schema.Column) Row(com.airbnb.spinaltap.mysql.mutation.schema.Row) Test(org.junit.Test)

Aggregations

Column (com.airbnb.spinaltap.mysql.mutation.schema.Column)9 Row (com.airbnb.spinaltap.mysql.mutation.schema.Row)8 ColumnMetadata (com.airbnb.spinaltap.mysql.mutation.schema.ColumnMetadata)7 Test (org.junit.Test)7 Table (com.airbnb.spinaltap.mysql.mutation.schema.Table)6 Mutation (com.airbnb.jitney.event.spinaltap.v1.Mutation)1 BinlogFilePos (com.airbnb.spinaltap.mysql.BinlogFilePos)1 DataSource (com.airbnb.spinaltap.mysql.DataSource)1 MysqlDeleteMutation (com.airbnb.spinaltap.mysql.mutation.MysqlDeleteMutation)1 MysqlInsertMutation (com.airbnb.spinaltap.mysql.mutation.MysqlInsertMutation)1 MysqlMutation (com.airbnb.spinaltap.mysql.mutation.MysqlMutation)1 MysqlMutationMetadata (com.airbnb.spinaltap.mysql.mutation.MysqlMutationMetadata)1 MysqlUpdateMutation (com.airbnb.spinaltap.mysql.mutation.MysqlUpdateMutation)1 ImmutableMap (com.google.common.collect.ImmutableMap)1