ios/chrome/browser/store_kit/store_kit_tab_helper_unittest.mm

// Copyright 2017 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.

#import "ios/chrome/browser/store_kit/store_kit_tab_helper.h"

#include "base/macros.h"
#import "ios/web/public/test/web_test_with_web_state.h"
#import "third_party/ocmock/OCMock/OCMock.h"

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

class StoreKitTabHelperTest : public web::WebTestWithWebState {
 public:
  StoreKitTabHelperTest() = default;
  ~StoreKitTabHelperTest() override = default;

 private:
  DISALLOW_COPY_AND_ASSIGN(StoreKitTabHelperTest);
};

TEST_F(StoreKitTabHelperTest, Constructor) {
  StoreKitTabHelper::CreateForWebState(web_state());
  StoreKitTabHelper* tab_helper = StoreKitTabHelper::FromWebState(web_state());
  ASSERT_TRUE(tab_helper);

  id mock_launcher =
      [OCMockObject niceMockForProtocol:@protocol(StoreKitLauncher)];
  // Verifies that GetLauncher returns the mock launcher object that was set.
  tab_helper->SetLauncher(mock_launcher);
  EXPECT_EQ(mock_launcher, tab_helper->GetLauncher());
}