From 9033ef83ec91e9821fed2a2a6aed5623fac8d731 Mon Sep 17 00:00:00 2001
From: Nathan Ziebart <nziebart@palantir.com>
Date: Fri, 20 Oct 2017 12:56:30 +0100
Subject: [PATCH] fix tests

---
 .../com/palantir/atlasdb/table/description/ApiTestSchema.java | 4 ++++
 .../atlasdb/table/description/SchemaApiTestV2Impl.java        | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/ApiTestSchema.java b/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/ApiTestSchema.java
index 931c9cd6297..da674cd5f65 100644
--- a/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/ApiTestSchema.java
+++ b/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/ApiTestSchema.java
@@ -38,6 +38,8 @@ private static Schema generateSchema() {
 
         schema.addTableDefinition("SchemaApiTest", new TableDefinition() {
             {
+                javaTableName("SchemaApiTest");
+
                 rowName();
                 rowComponent("component1", ValueType.STRING);
 
@@ -52,6 +54,8 @@ private static Schema generateSchema() {
 
         schema.addTableDefinition("HashComponentsTest", new TableDefinition() {
             {
+                javaTableName("HashComponentsTest");
+
                 rowName();
                 hashFirstNRowComponents(2);
                 rowComponent("component1", ValueType.VAR_LONG);
diff --git a/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/SchemaApiTestV2Impl.java b/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/SchemaApiTestV2Impl.java
index 3695451a8b8..025a751280f 100644
--- a/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/SchemaApiTestV2Impl.java
+++ b/atlasdb-client/src/test/java/com/palantir/atlasdb/table/description/SchemaApiTestV2Impl.java
@@ -32,6 +32,7 @@
 package com.palantir.atlasdb.table.description;
 
 import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyLong;
 import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
@@ -126,6 +127,6 @@ public void testUpdateEntryIfEntryDoesNotExist() {
 
         table.updateColumn1(TEST_ROW_KEY, entry -> entry + 1);
 
-        verify(table, never()).putColumn1(any(), any());
+        verify(table, never()).putColumn1(any(), anyLong());
     }
 }