createProfileActivity: mock Authenticator

app/src/androidTest/java/ch/sdp/vibester/activity/CreateProfileActivityTest.kt

@@ -11,8 +11,10 @@ import androidx.test.espresso.matcher.ViewMatchers.withId
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import ch.sdp.vibester.R
 import ch.sdp.vibester.TestMode
+import ch.sdp.vibester.auth.FireBaseAuthenticator
 import ch.sdp.vibester.database.DataGetter
 import ch.sdp.vibester.user.User
+import com.google.firebase.auth.FirebaseUser
 import dagger.hilt.android.testing.BindValue
 import dagger.hilt.android.testing.HiltAndroidRule
 import dagger.hilt.android.testing.HiltAndroidTest
@@ -39,21 +41,36 @@ class CreateProfileActivityTest {
     }
 
     @BindValue @JvmField
-    val mockUsersRepo = mockk<DataGetter>()
+    val mockAuthenticator = mockk<FireBaseAuthenticator>()
 
-    private fun createMockInvocation(email: String) {
-        every {mockUsersRepo.createUser(any(), any(), any(), any())} answers {
-            lastArg<(String) -> Unit>().invoke(email)
-        }
+    private fun createMockAuthenticator() {
+        val mockUser = createMockUser()
+        every { mockAuthenticator.getCurrUser() } returns mockUser
+        every { mockAuthenticator.getCurrUID() } returns mockUser.uid
+    }
 
-        every { mockUsersRepo.getUserData(any(), any()) } answers {
-            secondArg<(User) -> Unit>().invoke(User())
-        }
+    private fun createMockUser(): FirebaseUser {
+        val email = "[email protected]"
+        val uid = "mockuseruid"
+        val mockUser = mockk<FirebaseUser>()
+        every { mockUser.email } returns email
+        every { mockUser.uid } returns uid
+        return mockUser
+    }
 
-        every { mockUsersRepo.getCurrentUser() } answers {
-            null
-        }
+    @BindValue @JvmField
+    val mockDataGetter = mockk<DataGetter>()
 
+    private fun createMockDataGetter(email: String) {
+        every {mockDataGetter.createUser(any(), any(), any(), any())} answers {
+            lastArg<(String) -> Unit>().invoke(email)
+        }
+        every { mockDataGetter.getUserData(any(), any()) } answers {secondArg<(User) -> Unit>().invoke(User())}
+        every { mockDataGetter.getCurrentUser() } answers {null }
+        every { mockDataGetter.setSubFieldValue(any(), any(), any(), any()) } answers {}
+        every { mockDataGetter.updateFieldInt(any(), any(), any(), any()) } answers {}
+        every { mockDataGetter.setFieldValue(any(), any(), any()) } answers {}
+        every { mockDataGetter.updateSubFieldInt(any(), any(), any(), any(), any()) } answers {}
     }
 
     @After
@@ -69,9 +86,9 @@ class CreateProfileActivityTest {
 
         val intent = Intent(ApplicationProvider.getApplicationContext(), CreateProfileActivity::class.java)
         intent.putExtra("email", mockEmail)
-        intent.putExtra("isUnitTest", true)
 
-        createMockInvocation(mockEmail)
+        createMockDataGetter(mockEmail)
+        createMockAuthenticator()
 
         val scn: ActivityScenario<CreateProfileActivity> = ActivityScenario.launch(intent)
 
@@ -83,7 +100,6 @@ class CreateProfileActivityTest {
 
        Intents.intended(IntentMatchers.hasComponent(ProfileActivity::class.java.name))
        Intents.intended(IntentMatchers.hasExtra("email", mockEmail))
-       Intents.intended(IntentMatchers.hasExtra("isUnitTest", true))
     }
 
 }

app/src/main/java/ch/sdp/vibester/activity/CreateProfileActivity.kt

@@ -18,6 +18,9 @@ import javax.inject.Inject
 @AndroidEntryPoint
 class CreateProfileActivity : AppCompatActivity() {
 
+    @Inject
+    lateinit var authenticator: FireBaseAuthenticator
+
     @Inject
     lateinit var dataGetter: DataGetter
 
@@ -26,24 +29,18 @@ class CreateProfileActivity : AppCompatActivity() {
 
     private val REQUEST_CODE = 500
 
-    var isUnitTest: Boolean = false
-
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
         setContentView(R.layout.activity_create_profile)
 
-        isUnitTest = intent.getBooleanExtra("isUnitTest", false)
-
         val email = intent.getStringExtra("email").toString()
         val username = findViewById<EditText>(R.id.accountUsername)
 
         val btCreateAcc = findViewById<Button>(R.id.createButton)
         val btnUploadImg = findViewById<Button>(R.id.uploadImg)
 
         btCreateAcc.setOnClickListener {
-            if (!TestMode.isTest()){
-                dataGetter.setFieldValue(FireBaseAuthenticator.getCurrentUID(), "username", username.text.toString())
-            }
+            dataGetter.setFieldValue(authenticator.getCurrUID(), "username", username.text.toString())
             startNewActivity(email)
         }
 
@@ -55,7 +52,6 @@ class CreateProfileActivity : AppCompatActivity() {
     private fun startNewActivity(email: String) {
         val newIntent = Intent(this, ProfileActivity::class.java)
         newIntent.putExtra("email", email)
-        newIntent.putExtra("isUnitTest", isUnitTest)
 
         startActivity(newIntent)
     }
@@ -76,7 +72,7 @@ class CreateProfileActivity : AppCompatActivity() {
     override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?) {
         super.onActivityResult(requestCode, resultCode, data)
         if (resultCode == Activity.RESULT_OK && requestCode == REQUEST_CODE){
-            imageGetter.uploadFile("profileImg/${FireBaseAuthenticator.getCurrentUID()}", data?.data!!) { updateUI() }
+            imageGetter.uploadFile("profileImg/${authenticator.getCurrUID()}", data?.data!!) { updateUI() }
         }
     }
 }

app/src/main/java/ch/sdp/vibester/auth/FireBaseAuthenticator.kt

@@ -84,6 +84,17 @@ class FireBaseAuthenticator @Inject constructor() {
         return Firebase.auth.currentUser
     }
 
+    /**
+     * Getter for the current user ID
+     */
+    fun getCurrUID(): String {
+        var uid = ""
+        if (isLoggedIn()) {
+            uid = FirebaseAuth.getInstance().currentUser!!.uid
+        }
+        return uid
+    }
+
 
     /**
      * A function to log in with email and password