content/browser/sms/sms_provider_gms_verification_unittest.cc

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/sms/sms_service.h"

#include <string>

#include "base/android/jni_string.h"
#include "base/android/scoped_java_ref.h"
#include "base/test/scoped_feature_list.h"
#include "content/browser/sms/sms_provider.h"
#include "content/browser/sms/sms_provider_gms_verification.h"
#include "content/public/common/content_features.h"
#include "content/public/test/test_renderer_host.h"
#include "content/test/content_unittests_jni_headers/SmsVerificationFakes_jni.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

using base::android::AttachCurrentThread;
using ::testing::_;
using ::testing::NiceMock;
using url::Origin;

namespace content {

namespace {

class MockObserver : public SmsProvider::Observer {
 public:
  MockObserver() = default;
  ~MockObserver() override = default;

  MOCK_METHOD2(OnReceive,
               bool(const Origin&, const std::string& one_time_code));

 private:
  DISALLOW_COPY_AND_ASSIGN(MockObserver);
};

// SmsProviderGmsVerificationTest tests the JNI bindings to the android
// SmsVerificationReceiver and the handling of the SMS upon retrieval.
class SmsProviderGmsVerificationTest : public RenderViewHostTestHarness {
 protected:
  SmsProviderGmsVerificationTest() = default;
  ~SmsProviderGmsVerificationTest() override = default;

  void SetUp() {
    RenderViewHostTestHarness::SetUp();
    provider_ = std::make_unique<SmsProviderGmsVerification>();
    j_fake_sms_retriever_client_.Reset(
        Java_FakeSmsRetrieverClient_create(AttachCurrentThread()));
    Java_SmsVerificationFakes_setClientForTesting(
        AttachCurrentThread(), provider_->GetSmsReceiverForTesting(),
        j_fake_sms_retriever_client_);
    provider_->AddObserver(&observer_);
  }

  void TriggerSmsVerificationSms(const std::string& sms) {
    JNIEnv* env = base::android::AttachCurrentThread();
    Java_FakeSmsRetrieverClient_triggerSmsVerificationSms(
        env, j_fake_sms_retriever_client_,
        base::android::ConvertUTF8ToJavaString(env, sms));
  }

  void TriggerTimeout() {
    JNIEnv* env = base::android::AttachCurrentThread();
    Java_FakeSmsRetrieverClient_triggerTimeout(env,
                                               j_fake_sms_retriever_client_);
  }

  SmsProviderGmsVerification* provider() { return provider_.get(); }

  NiceMock<MockObserver>* observer() { return &observer_; }

 private:
  std::unique_ptr<SmsProviderGmsVerification> provider_;
  NiceMock<MockObserver> observer_;
  base::android::ScopedJavaGlobalRef<jobject> j_fake_sms_retriever_client_;
  base::test::ScopedFeatureList feature_list_;

  DISALLOW_COPY_AND_ASSIGN(SmsProviderGmsVerificationTest);
};

}  // namespace

TEST_F(SmsProviderGmsVerificationTest, Retrieve) {
  std::string test_url = "https://google.com";

  EXPECT_CALL(*observer(), OnReceive(Origin::Create(GURL(test_url)), "ABC123"));
  provider()->Retrieve(main_rfh());
  TriggerSmsVerificationSms("Hi\n@google.com #ABC123");
}

TEST_F(SmsProviderGmsVerificationTest, IgnoreBadSms) {
  std::string test_url = "https://google.com";
  std::string good_sms = "Hi\n@google.com #ABC123";
  std::string bad_sms = "Hi\n@b.com";

  EXPECT_CALL(*observer(), OnReceive(Origin::Create(GURL(test_url)), "ABC123"));

  provider()->Retrieve(main_rfh());
  TriggerSmsVerificationSms(bad_sms);
  TriggerSmsVerificationSms(good_sms);
}

TEST_F(SmsProviderGmsVerificationTest, TaskTimedOut) {
  EXPECT_CALL(*observer(), OnReceive(_, _)).Times(0);
  provider()->Retrieve(main_rfh());
  TriggerTimeout();
}

TEST_F(SmsProviderGmsVerificationTest, OneObserverTwoTasks) {
  std::string test_url = "https://google.com";

  EXPECT_CALL(*observer(), OnReceive(Origin::Create(GURL(test_url)), "ABC123"));

  // Two tasks for when 1 request gets aborted but the task is still triggered.
  provider()->Retrieve(main_rfh());
  provider()->Retrieve(main_rfh());

  // First timeout should be ignored.
  TriggerTimeout();
  TriggerSmsVerificationSms("Hi\n@google.com #ABC123");
}

}  // namespace content