Test for password file

spotify · Aug 6, 2018 · decaded · decaded
1 parent d8b28fc
commit decaded
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dbeam-core/src/test/scala/com/spotify/dbeam/JdbcAvroJobTest.scala b/dbeam-core/src/test/scala/com/spotify/dbeam/JdbcAvroJobTest.scala
@@ -36,16 +36,19 @@ class JdbcAvroJobTest extends FlatSpec with Matchers with BeforeAndAfterAll {
     "jdbc:h2:mem:test2;MODE=PostgreSQL;DATABASE_TO_UPPER=false;DB_CLOSE_DELAY=-1"
   private val db: Database = Database.forURL(connectionUrl, driver = "org.h2.Driver")
   private val dir = tmpDir
+  private val passwordFile = new File(dir.getAbsolutePath + ".password")
 
   def tmpDir: File = new File(
     new File(sys.props("java.io.tmpdir")),
     "jdbc-avro-test-" + UUID.randomUUID().toString)
 
   override def beforeAll(): Unit = {
     JdbcTestFixtures.createFixtures(db, Seq(JdbcTestFixtures.record1, JdbcTestFixtures.record2))
+    passwordFile.createNewFile()
   }
 
   override protected def afterAll(): Unit = {
+    passwordFile.delete()
     Files.walk(dir.toPath)
       .sorted(Comparator.reverseOrder())
       .iterator()
@@ -59,7 +62,7 @@ class JdbcAvroJobTest extends FlatSpec with Matchers with BeforeAndAfterAll {
         "--skipPartitionCheck",
         "--connectionUrl=" + connectionUrl,
         "--username=",
-        "--password=",
+        "--passwordFile=" + passwordFile.getAbsolutePath,
         "--table=coffees",
         "--output=" + dir.getAbsolutePath)
     )