Split LabelRepository off the MessageRepository

app/src/main/java/ch/dissem/apps/abit/MainActivity.kt

@@ -28,7 +28,7 @@ import android.widget.RelativeLayout
 import ch.dissem.apps.abit.drawer.ProfileImageListener
 import ch.dissem.apps.abit.drawer.ProfileSelectionListener
 import ch.dissem.apps.abit.listener.ListSelectionListener
-import ch.dissem.apps.abit.repository.AndroidMessageRepository.Companion.LABEL_ARCHIVE
+import ch.dissem.apps.abit.repository.AndroidLabelRepository.Companion.LABEL_ARCHIVE
 import ch.dissem.apps.abit.service.Singleton
 import ch.dissem.apps.abit.synchronization.SyncAdapter
 import ch.dissem.apps.abit.util.Labels
@@ -278,7 +278,7 @@ class MainActivity : AppCompatActivity(), ListSelectionListener<Serializable> {
         }
 
         doAsync {
-            val labels = bmc.messages.getLabels()
+            val labels = bmc.labels.getLabels()
 
             uiThread {
                 if (intent.hasExtra(EXTRA_SHOW_LABEL)) {

app/src/main/java/ch/dissem/apps/abit/adapter/SwipeableMessageAdapter.kt

@@ -28,7 +28,7 @@ import android.widget.ImageView
 import android.widget.TextView
 import ch.dissem.apps.abit.Identicon
 import ch.dissem.apps.abit.R
-import ch.dissem.apps.abit.repository.AndroidMessageRepository.Companion.LABEL_ARCHIVE
+import ch.dissem.apps.abit.repository.AndroidLabelRepository.Companion.LABEL_ARCHIVE
 import ch.dissem.apps.abit.util.Assets
 import ch.dissem.apps.abit.util.Strings.prepareMessageExtract
 import ch.dissem.bitmessage.entity.Plaintext

app/src/main/java/ch/dissem/apps/abit/repository/AndroidLabelRepository.kt

@@ -0,0 +1,113 @@
+/*
+ * Copyright 2015 Christian Basler
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package ch.dissem.apps.abit.repository
+
+import android.content.ContentValues
+import android.content.Context
+import android.database.Cursor
+import android.database.DatabaseUtils
+import ch.dissem.apps.abit.util.Labels
+import ch.dissem.bitmessage.entity.valueobject.Label
+import ch.dissem.bitmessage.ports.AbstractLabelRepository
+import ch.dissem.bitmessage.ports.MessageRepository
+import java.util.*
+
+/**
+ * [MessageRepository] implementation using the Android SQL API.
+ */
+class AndroidLabelRepository(private val sql: SqlHelper, private val context: Context) : AbstractLabelRepository() {
+
+    override fun find(where: String): List<Label> {
+        val result = LinkedList<Label>()
+
+        // Define a projection that specifies which columns from the database
+        // you will actually use after this query.
+        val projection = arrayOf(COLUMN_ID, COLUMN_LABEL, COLUMN_TYPE, COLUMN_COLOR)
+
+        sql.readableDatabase.query(
+            TABLE_NAME, projection,
+                where, null, null, null,
+            COLUMN_ORDER
+        ).use { c ->
+            while (c.moveToNext()) {
+                result.add(getLabel(c, context))
+            }
+        }
+        return result
+    }
+
+    override fun save(label: Label) {
+        val db = sql.writableDatabase
+        if (label.id != null) {
+            val values = ContentValues()
+            values.put(COLUMN_LABEL, label.toString())
+            values.put(COLUMN_TYPE, label.type?.name)
+            values.put(COLUMN_COLOR, label.color)
+            values.put(COLUMN_ORDER, label.ord)
+            db.update(TABLE_NAME, values, "id=?", arrayOf(label.id.toString()))
+        } else {
+            try {
+                db.beginTransaction()
+
+                val exists = DatabaseUtils.queryNumEntries(db, TABLE_NAME, "label=?", arrayOf(label.toString())) > 0
+
+                if (exists) {
+                    val values = ContentValues()
+                    values.put(COLUMN_TYPE, label.type?.name)
+                    values.put(COLUMN_COLOR, label.color)
+                    values.put(COLUMN_ORDER, label.ord)
+                    db.update(TABLE_NAME, values, "label=?", arrayOf(label.toString()))
+                } else {
+                    val values = ContentValues()
+                    values.put(COLUMN_LABEL, label.toString())
+                    values.put(COLUMN_TYPE, label.type?.name)
+                    values.put(COLUMN_COLOR, label.color)
+                    values.put(COLUMN_ORDER, label.ord)
+                    db.insertOrThrow(TABLE_NAME, null, values)
+                }
+                db.setTransactionSuccessful()
+            } finally {
+                db.endTransaction()
+            }
+        }
+    }
+
+    internal fun findLabels(msgId: Any) = find("id IN (SELECT label_id FROM Message_Label WHERE message_id=$msgId)")
+
+    companion object {
+        val LABEL_ARCHIVE = Label("archive", null, 0)
+
+        private const val TABLE_NAME = "Label"
+        private const val COLUMN_ID = "id"
+        private const val COLUMN_LABEL = "label"
+        private const val COLUMN_TYPE = "type"
+        private const val COLUMN_COLOR = "color"
+        private const val COLUMN_ORDER = "ord"
+
+        internal fun getLabel(c: Cursor, context: Context): Label {
+            val typeName = c.getString(c.getColumnIndex(COLUMN_TYPE))
+            val type = if (typeName == null) null else Label.Type.valueOf(typeName)
+            val text: String? = Labels.getText(type, null, context)
+            val label = Label(
+                text ?: c.getString(c.getColumnIndex(COLUMN_LABEL)),
+                type,
+                c.getInt(c.getColumnIndex(COLUMN_COLOR)))
+            label.id = c.getLong(c.getColumnIndex(COLUMN_ID))
+            return label
+        }
+    }
+}

app/src/main/java/ch/dissem/apps/abit/repository/AndroidMessageRepository.kt

@@ -17,12 +17,11 @@
 package ch.dissem.apps.abit.repository
 
 import android.content.ContentValues
-import android.content.Context
 import android.database.Cursor
 import android.database.DatabaseUtils
 import android.database.sqlite.SQLiteConstraintException
 import android.database.sqlite.SQLiteDatabase
-import ch.dissem.apps.abit.util.Labels
+import ch.dissem.apps.abit.repository.AndroidLabelRepository.Companion.LABEL_ARCHIVE
 import ch.dissem.apps.abit.util.UuidUtils
 import ch.dissem.apps.abit.util.UuidUtils.asUuid
 import ch.dissem.bitmessage.entity.BitmessageAddress
@@ -40,7 +39,7 @@ import java.util.*
 /**
  * [MessageRepository] implementation using the Android SQL API.
  */
-class AndroidMessageRepository(private val sql: SqlHelper, private val context: Context) : AbstractMessageRepository() {
+class AndroidMessageRepository(private val sql: SqlHelper) : AbstractMessageRepository() {
 
     override fun findMessages(label: Label?, offset: Int, limit: Int) = if (label === LABEL_ARCHIVE) {
         super.findMessages(null as Label?, offset, limit)
@@ -48,73 +47,6 @@ class AndroidMessageRepository(private val sql: SqlHelper, private val context:
         super.findMessages(label, offset, limit)
     }
 
-    public override fun findLabels(where: String): List<Label> {
-        val result = LinkedList<Label>()
-
-        // Define a projection that specifies which columns from the database
-        // you will actually use after this query.
-        val projection = arrayOf(LBL_COLUMN_ID, LBL_COLUMN_LABEL, LBL_COLUMN_TYPE, LBL_COLUMN_COLOR)
-
-        sql.readableDatabase.query(
-                LBL_TABLE_NAME, projection,
-                where, null, null, null,
-                LBL_COLUMN_ORDER
-        ).use { c ->
-            while (c.moveToNext()) {
-                result.add(getLabel(c))
-            }
-        }
-        return result
-    }
-
-    private fun getLabel(c: Cursor): Label {
-        val typeName = c.getString(c.getColumnIndex(LBL_COLUMN_TYPE))
-        val type = if (typeName == null) null else Label.Type.valueOf(typeName)
-        val text: String? = Labels.getText(type, null, context)
-        val label = Label(
-                text ?: c.getString(c.getColumnIndex(LBL_COLUMN_LABEL)),
-                type,
-                c.getInt(c.getColumnIndex(LBL_COLUMN_COLOR)))
-        label.id = c.getLong(c.getColumnIndex(LBL_COLUMN_ID))
-        return label
-    }
-
-    override fun save(label: Label) {
-        val db = sql.writableDatabase
-        if (label.id != null) {
-            val values = ContentValues()
-            values.put(LBL_COLUMN_LABEL, label.toString())
-            values.put(LBL_COLUMN_TYPE, label.type?.name)
-            values.put(LBL_COLUMN_COLOR, label.color)
-            values.put(LBL_COLUMN_ORDER, label.ord)
-            db.update(LBL_TABLE_NAME, values, "id=?", arrayOf(label.id.toString()))
-        } else {
-            try {
-                db.beginTransaction()
-
-                val exists = DatabaseUtils.queryNumEntries(db, LBL_TABLE_NAME, "label=?", arrayOf(label.toString())) > 0
-
-                if (exists) {
-                    val values = ContentValues()
-                    values.put(LBL_COLUMN_TYPE, label.type?.name)
-                    values.put(LBL_COLUMN_COLOR, label.color)
-                    values.put(LBL_COLUMN_ORDER, label.ord)
-                    db.update(LBL_TABLE_NAME, values, "label=?", arrayOf(label.toString()))
-                } else {
-                    val values = ContentValues()
-                    values.put(LBL_COLUMN_LABEL, label.toString())
-                    values.put(LBL_COLUMN_TYPE, label.type?.name)
-                    values.put(LBL_COLUMN_COLOR, label.color)
-                    values.put(LBL_COLUMN_ORDER, label.ord)
-                    db.insertOrThrow(LBL_TABLE_NAME, null, values)
-                }
-                db.setTransactionSuccessful()
-            } finally {
-                db.endTransaction()
-            }
-        }
-    }
-
     override fun countUnread(label: Label?) = when {
         label === LABEL_ARCHIVE -> 0
         label == null -> DatabaseUtils.queryNumEntries(
@@ -166,11 +98,11 @@ class AndroidMessageRepository(private val sql: SqlHelper, private val context:
 
         // save new parents
         var order = 0
+        val values = ContentValues()
         for (parentIV in message.parents) {
             getMessage(parentIV)?.let { parent ->
                 mergeConversations(db, parent.conversationId, message.conversationId)
                 order++
-                val values = ContentValues()
                 values.put("parent", parentIV.hash)
                 values.put("child", childIV)
                 values.put("pos", order)
@@ -212,51 +144,39 @@ class AndroidMessageRepository(private val sql: SqlHelper, private val context:
             if (limit == 0) null else "$offset, $limit"
         ).use { c ->
             while (c.moveToNext()) {
-                val iv = c.getBlob(c.getColumnIndex(COLUMN_IV))
+                result.add(getMessage(c))
-                val data = c.getBlob(c.getColumnIndex(COLUMN_DATA))
-                val type = Plaintext.Type.valueOf(c.getString(c.getColumnIndex(COLUMN_TYPE)))
-                val builder = Plaintext.readWithoutSignature(type,
-                        ByteArrayInputStream(data))
-                val id = c.getLong(c.getColumnIndex(COLUMN_ID))
-                builder.id(id)
-                builder.IV(InventoryVector.fromHash(iv))
-                val sender = c.getString(c.getColumnIndex(COLUMN_SENDER))
-                if (sender != null) {
-                    val address = ctx.addressRepository.getAddress(sender)
-                    if (address != null) {
-                        builder.from(address)
-                    } else {
-                        builder.from(BitmessageAddress(sender))
-                    }
-                }
-                val recipient = c.getString(c.getColumnIndex(COLUMN_RECIPIENT))
-                if (recipient != null) {
-                    val address = ctx.addressRepository.getAddress(recipient)
-                    if (address != null) {
-                        builder.to(address)
-                    } else {
-                        builder.to(BitmessageAddress(sender))
-                    }
-                }
-                builder.ackData(c.getBlob(c.getColumnIndex(COLUMN_ACK_DATA)))
-                builder.sent(c.getLong(c.getColumnIndex(COLUMN_SENT)))
-                builder.received(c.getLong(c.getColumnIndex(COLUMN_RECEIVED)))
-                builder.status(Plaintext.Status.valueOf(c.getString(c.getColumnIndex(COLUMN_STATUS))))
-                builder.ttl(c.getLong(c.getColumnIndex(COLUMN_TTL)))
-                builder.retries(c.getInt(c.getColumnIndex(COLUMN_RETRIES)))
-                val nextTryColumn = c.getColumnIndex(COLUMN_NEXT_TRY)
-                if (!c.isNull(nextTryColumn)) {
-                    builder.nextTry(c.getLong(nextTryColumn))
-                }
-                builder.conversation(asUuid(c.getBlob(c.getColumnIndex(COLUMN_CONVERSATION))))
-                builder.labels(findLabels(id))
-                result.add(builder.build())
             }
         }
         return result
     }
 
-    private fun findLabels(id: Long) = findLabels("id IN (SELECT label_id FROM Message_Label WHERE message_id=$id)")
+    private fun getMessage(c: Cursor): Plaintext = Plaintext.readWithoutSignature(
+        Plaintext.Type.valueOf(c.getString(c.getColumnIndex(COLUMN_TYPE))),
+        ByteArrayInputStream(c.getBlob(c.getColumnIndex(COLUMN_DATA)))
+    ).build {
+        id = c.getLong(c.getColumnIndex(COLUMN_ID))
+        inventoryVector = InventoryVector.fromHash(c.getBlob(c.getColumnIndex(COLUMN_IV)))
+        c.getString(c.getColumnIndex(COLUMN_SENDER))?.let {
+            from = ctx.addressRepository.getAddress(it) ?: BitmessageAddress(it)
+        }
+        c.getString(c.getColumnIndex(COLUMN_RECIPIENT))?.let {
+            to = ctx.addressRepository.getAddress(it) ?: BitmessageAddress(it)
+        }
+        ackData = c.getBlob(c.getColumnIndex(COLUMN_ACK_DATA))
+        sent = c.getLong(c.getColumnIndex(COLUMN_SENT))
+        received = c.getLong(c.getColumnIndex(COLUMN_RECEIVED))
+        status = Plaintext.Status.valueOf(c.getString(c.getColumnIndex(COLUMN_STATUS)))
+        ttl = c.getLong(c.getColumnIndex(COLUMN_TTL))
+        retries = c.getInt(c.getColumnIndex(COLUMN_RETRIES))
+        val nextTryColumn = c.getColumnIndex(COLUMN_NEXT_TRY)
+        if (!c.isNull(nextTryColumn)) {
+            nextTry = c.getLong(nextTryColumn)
+        }
+        conversation = asUuid(c.getBlob(c.getColumnIndex(COLUMN_CONVERSATION)))
+        labels = findLabels(id!!)
+    }
+
+    private fun findLabels(msgId: Any) = (ctx.labelRepository as AndroidLabelRepository).findLabels(msgId)
 
     override fun save(message: Plaintext) {
         saveContactIfNecessary(message.from)
@@ -326,8 +246,6 @@ class AndroidMessageRepository(private val sql: SqlHelper, private val context:
     }
 
     companion object {
-        val LABEL_ARCHIVE = Label("archive", null, 0)
-
         private const val TABLE_NAME = "Message"
         private const val COLUMN_ID = "id"
         private const val COLUMN_IV = "iv"
@@ -350,12 +268,5 @@ class AndroidMessageRepository(private val sql: SqlHelper, private val context:
         private const val JOIN_TABLE_NAME = "Message_Label"
         private const val JT_COLUMN_MESSAGE = "message_id"
         private const val JT_COLUMN_LABEL = "label_id"
-
-        private const val LBL_TABLE_NAME = "Label"
-        private const val LBL_COLUMN_ID = "id"
-        private const val LBL_COLUMN_LABEL = "label"
-        private const val LBL_COLUMN_TYPE = "type"
-        private const val LBL_COLUMN_COLOR = "color"
-        private const val LBL_COLUMN_ORDER = "ord"
     }
 }

app/src/main/java/ch/dissem/apps/abit/service/Singleton.kt

@@ -52,32 +52,33 @@ object Singleton {
 
     fun getBitmessageContext(context: Context): BitmessageContext =
         init({ bitmessageContext }, { bitmessageContext = it }) {
+            BitmessageContext.build {
+                TTL.pubkey = 2 * DAY
                 val ctx = context.applicationContext
                 val sqlHelper = SqlHelper(ctx)
-            val powRepo = AndroidProofOfWorkRepository(sqlHelper)
+                proofOfWorkEngine = SwitchingProofOfWorkEngine(
-            this.powRepo = powRepo
-            TTL.pubkey = 2 * DAY
-            BitmessageContext.Builder()
-                    .proofOfWorkEngine(SwitchingProofOfWorkEngine(
                     ctx, Constants.PREFERENCE_SERVER_POW,
                     ServerPowEngine(ctx),
                     ServicePowEngine(ctx)
-                    ))
+                )
-                    .cryptography(AndroidCryptography())
+                cryptography = AndroidCryptography()
-                    .nodeRegistry(AndroidNodeRegistry(sqlHelper))
+                nodeRegistry = AndroidNodeRegistry(sqlHelper)
-                    .inventory(AndroidInventory(sqlHelper))
+                inventory = AndroidInventory(sqlHelper)
-                    .addressRepo(AndroidAddressRepository(sqlHelper))
+                addressRepo = AndroidAddressRepository(sqlHelper)
-                    .messageRepo(AndroidMessageRepository(sqlHelper, ctx))
+                labelRepo = AndroidLabelRepository(sqlHelper, ctx)
-                    .powRepo(powRepo)
+                messageRepo = AndroidMessageRepository(sqlHelper)
-                    .networkHandler(NioNetworkHandler())
+                proofOfWorkRepo = AndroidProofOfWorkRepository(sqlHelper).also { powRepo = it }
-                    .listener(getMessageListener(ctx))
+                networkHandler = NioNetworkHandler()
-                    .labeler(labeler)
+                listener = getMessageListener(ctx)
-                    .doNotSendPubkeyOnIdentityCreation()
+                labeler = Singleton.labeler
-                    .build()
+                preferences.sendPubkeyOnIdentityCreation = false
+            }
         }
 
     fun getMessageListener(ctx: Context) = init({ messageListener }, { messageListener = it }) { MessageListener(ctx) }
 
+    fun getLabelRepository(ctx: Context) = getBitmessageContext(ctx).labels as AndroidLabelRepository
+
     fun getMessageRepository(ctx: Context) = getBitmessageContext(ctx).messages as AndroidMessageRepository
 
     fun getAddressRepository(ctx: Context) = getBitmessageContext(ctx).addresses as AndroidAddressRepository

app/src/main/java/ch/dissem/apps/abit/util/Exports.kt

@@ -35,14 +35,15 @@ object Exports {
             )
             zip.closeEntry()
 
-            val messageRepo = Singleton.getMessageRepository(ctx)
+            val labelRepo = Singleton.getLabelRepository(ctx)
             zip.putNextEntry(ZipEntry("labels.json"))
-            val exportLabels = MessageExport.exportLabels(messageRepo.getLabels())
+            val exportLabels = MessageExport.exportLabels(labelRepo.getLabels())
             zip.write(
                 exportLabels.toJsonString(true).toByteArray()
             )
             zip.closeEntry()
             zip.putNextEntry(ZipEntry("messages.json"))
+            val messageRepo = Singleton.getMessageRepository(ctx)
             val exportMessages = MessageExport.exportMessages(messageRepo.getAllMessages())
             zip.write(
                 exportMessages.toJsonString(true).toByteArray()
@@ -62,13 +63,13 @@ object Exports {
                 bmc.addresses.save(contact)
             }
         }
-        bmc.messages.getLabels().forEach { label ->
+        bmc.labels.getLabels().forEach { label ->
             labels[label.toString()] = label
         }
         processEntry(ctx, zipFile, "labels.json") { json ->
             MessageExport.importLabels(json).forEach { label ->
                 if (!labels.contains(label.toString())) {
-                    bmc.messages.save(label)
+                    bmc.labels.save(label)
                     labels[label.toString()] = label
                 }
             }

app/src/test/java/ch/dissem/bitmessage/repository/AndroidLabelRepositoryTest.kt

@@ -0,0 +1,60 @@
+/*
+ * Copyright 2015 Christian Basler
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package ch.dissem.bitmessage.repository
+
+import android.os.Build
+import ch.dissem.apps.abit.repository.AndroidLabelRepository
+import ch.dissem.apps.abit.repository.SqlHelper
+import ch.dissem.bitmessage.entity.valueobject.Label
+import ch.dissem.bitmessage.ports.LabelRepository
+import org.junit.Assert.*
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.RuntimeEnvironment
+import org.robolectric.annotation.Config
+
+@RunWith(RobolectricTestRunner::class)
+@Config(sdk = intArrayOf(Build.VERSION_CODES.LOLLIPOP), packageName = "ch.dissem.apps.abit")
+class AndroidLabelRepositoryTest : TestBase() {
+
+    private lateinit var repo: LabelRepository
+
+
+    @Before
+    fun setUp() {
+        RuntimeEnvironment.application.deleteDatabase(SqlHelper.DATABASE_NAME)
+        val sqlHelper = SqlHelper(RuntimeEnvironment.application)
+
+        repo = AndroidLabelRepository(sqlHelper, RuntimeEnvironment.application)
+    }
+
+    @Test
+    fun `ensure labels are retrieved`() {
+        val labels = repo.getLabels()
+        assertEquals(7, labels.size.toLong())
+    }
+
+    @Test
+    fun `ensure labels can be retrieved by type`() {
+        val labels = repo.getLabels(Label.Type.INBOX)
+        assertEquals(1, labels.size.toLong())
+        assertEquals("Inbox", labels[0].toString())
+    }
+
+}

app/src/test/java/ch/dissem/bitmessage/repository/AndroidMessageRepositoryTest.kt

@@ -17,8 +17,8 @@
 package ch.dissem.bitmessage.repository
 
 import android.os.Build
-import ch.dissem.apps.abit.BuildConfig
 import ch.dissem.apps.abit.repository.AndroidAddressRepository
+import ch.dissem.apps.abit.repository.AndroidLabelRepository
 import ch.dissem.apps.abit.repository.AndroidMessageRepository
 import ch.dissem.apps.abit.repository.SqlHelper
 import ch.dissem.bitmessage.cryptography.sc.SpongyCryptography
@@ -31,12 +31,12 @@ import ch.dissem.bitmessage.entity.valueobject.ExtendedEncoding
 import ch.dissem.bitmessage.entity.valueobject.Label
 import ch.dissem.bitmessage.entity.valueobject.PrivateKey
 import ch.dissem.bitmessage.entity.valueobject.extended.Message
+import ch.dissem.bitmessage.ports.LabelRepository
 import ch.dissem.bitmessage.ports.MessageRepository
 import ch.dissem.bitmessage.utils.UnixTime
 import org.hamcrest.BaseMatcher
 import org.hamcrest.CoreMatchers.`is`
 import org.hamcrest.Description
-import org.hamcrest.Matcher
 import org.hamcrest.Matchers.*
 import org.junit.Assert.*
 import org.junit.Before
@@ -67,10 +67,12 @@ class AndroidMessageRepositoryTest : TestBase() {
         val sqlHelper = SqlHelper(RuntimeEnvironment.application)
 
         val addressRepo = AndroidAddressRepository(sqlHelper)
-        repo = AndroidMessageRepository(sqlHelper, RuntimeEnvironment.application)
+        val labelRepo = AndroidLabelRepository(sqlHelper, RuntimeEnvironment.application)
+        repo = AndroidMessageRepository(sqlHelper)
         mockedInternalContext(
             cryptography = SpongyCryptography(),
             addressRepository = addressRepo,
+            labelRepository = labelRepo,
             messageRepository = repo,
             port = 12345,
             connectionTTL = 10,
@@ -86,29 +88,16 @@ class AndroidMessageRepositoryTest : TestBase() {
         identity = BitmessageAddress(PrivateKey(false, 1, 1000, 1000, DOES_ACK))
         addressRepo.save(identity)
 
-        inbox = repo.getLabels(Label.Type.INBOX)[0]
+        inbox = labelRepo.getLabels(Label.Type.INBOX)[0]
-        sent = repo.getLabels(Label.Type.SENT)[0]
+        sent = labelRepo.getLabels(Label.Type.SENT)[0]
-        drafts = repo.getLabels(Label.Type.DRAFT)[0]
+        drafts = labelRepo.getLabels(Label.Type.DRAFT)[0]
-        unread = repo.getLabels(Label.Type.UNREAD)[0]
+        unread = labelRepo.getLabels(Label.Type.UNREAD)[0]
 
         addMessage(contactA, identity, Plaintext.Status.RECEIVED, inbox, unread)
         addMessage(identity, contactA, Plaintext.Status.DRAFT, drafts)
         addMessage(identity, contactB, Plaintext.Status.DRAFT, unread)
     }
 
-    @Test
-    fun `ensure labels are retrieved`() {
-        val labels = repo.getLabels()
-        assertEquals(7, labels.size.toLong())
-    }
-
-    @Test
-    fun `ensure labels can be retrieved by type`() {
-        val labels = repo.getLabels(Label.Type.INBOX)
-        assertEquals(1, labels.size.toLong())
-        assertEquals("Inbox", labels[0].toString())
-    }
-
     @Test
     fun `ensure messages can be found by label`() {
         val messages = repo.findMessages(inbox)

app/src/test/java/ch/dissem/bitmessage/repository/AndroidProofOfWorkRepositoryTest.kt

@@ -17,10 +17,7 @@
 package ch.dissem.bitmessage.repository
 
 import android.os.Build.VERSION_CODES.LOLLIPOP
-import ch.dissem.apps.abit.repository.AndroidAddressRepository
+import ch.dissem.apps.abit.repository.*
-import ch.dissem.apps.abit.repository.AndroidMessageRepository
-import ch.dissem.apps.abit.repository.AndroidProofOfWorkRepository
-import ch.dissem.apps.abit.repository.SqlHelper
 import ch.dissem.bitmessage.entity.BitmessageAddress
 import ch.dissem.bitmessage.entity.ObjectMessage
 import ch.dissem.bitmessage.entity.Plaintext
@@ -64,10 +61,11 @@ class AndroidProofOfWorkRepositoryTest : TestBase() {
         val sqlHelper = SqlHelper(RuntimeEnvironment.application)
 
         addressRepo = AndroidAddressRepository(sqlHelper)
-        messageRepo = AndroidMessageRepository(sqlHelper, RuntimeEnvironment.application)
+        messageRepo = AndroidMessageRepository(sqlHelper)
         repo = AndroidProofOfWorkRepository(sqlHelper)
         mockedInternalContext(
             addressRepository = addressRepo,
+            labelRepository = AndroidLabelRepository(sqlHelper, RuntimeEnvironment.application),
             messageRepository = messageRepo,
             proofOfWorkRepository = repo,
             cryptography = cryptography()

app/src/test/java/ch/dissem/bitmessage/repository/TestBase.kt

@@ -18,6 +18,7 @@ package ch.dissem.bitmessage.repository
 
 import ch.dissem.bitmessage.BitmessageContext
 import ch.dissem.bitmessage.InternalContext
+import ch.dissem.bitmessage.Preferences
 import ch.dissem.bitmessage.cryptography.sc.SpongyCryptography
 import ch.dissem.bitmessage.entity.BitmessageAddress
 import ch.dissem.bitmessage.entity.ObjectMessage
@@ -51,6 +52,7 @@ open class TestBase {
             nodeRegistry: NodeRegistry = mock {},
             networkHandler: NetworkHandler = mock {},
             addressRepository: AddressRepository = mock {},
+            labelRepository: LabelRepository = mock {},
             messageRepository: MessageRepository = mock {},
             proofOfWorkRepository: ProofOfWorkRepository = mock {},
             proofOfWorkEngine: ProofOfWorkEngine = mock {},
@@ -66,16 +68,18 @@ open class TestBase {
             nodeRegistry,
             networkHandler,
             addressRepository,
+            labelRepository,
             messageRepository,
             proofOfWorkRepository,
             proofOfWorkEngine,
             customCommandHandler,
             listener,
             labeler,
-                "/Jabit:TEST/",
+            Preferences().apply {
-                port,
+                this.port = port
-                connectionTTL,
+                this.connectionTTL = connectionTTL
-                connectionLimit
+                this.connectionLimit = connectionLimit
+            }
         ))
 
         fun randomInventoryVector(): InventoryVector {

build.gradle

@@ -1,5 +1,5 @@
 buildscript {
-    ext.kotlin_version = '1.1.60'
+    ext.kotlin_version = '1.1.61'
     ext.anko_version = '0.10.2'
     repositories {
         jcenter()