Tests/FunctionalTests/TestApp/TestSources/GTXTestAnalyticsBaseTest.m

//
// Copyright 2018 Google Inc.
//
// 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.
//

#import "GTXTestAnalyticsBaseTest.h"

#import "GTXAnalytics.h"

@implementation GTXTestAnalyticsBaseTest {
  BOOL _prevAnalyticsEnabled;
  GTXAnalyticsHandlerBlock _prevAnalyticsHandler;
}

+ (void)setUp {
  [super setUp];
  [GTXiLib installOnTestSuite:[GTXTestSuite suiteWithAllTestsInClass:self]
                       checks:@[gCheckFailsIfFailingClass]
            elementBlacklists:@[]];
  [GTXTestViewController addElementToTestArea:
      [[GTXTestFailingClass alloc] initWithFrame:CGRectMake(0, 0, 100, 100)]];
}

+ (void)tearDown {
  [GTXTestViewController clearTestArea];
  [super tearDown];
}

- (void)setUp {
  [super setUp];

  _prevAnalyticsEnabled = GTXAnalytics.enabled;
  _prevAnalyticsHandler = GTXAnalytics.handler;
  _analyticsEventCount = 0;
  GTXAnalytics.handler = ^(GTXAnalyticsEvent event) {
    _analyticsEventCount += 1;
  };
}

- (void)tearDown {
  GTXAnalytics.enabled = _prevAnalyticsEnabled;
  GTXAnalytics.handler = _prevAnalyticsHandler;

  [super tearDown];
}

@end