FIRCLSMockFileManager.m 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. // Copyright 2019 Google
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #import "FIRCLSMockFileManager.h"
  15. @implementation FIRCLSMockFileManager
  16. @synthesize pathNamespace;
  17. - (instancetype)init {
  18. self = [super init];
  19. if (!self) {
  20. return nil;
  21. }
  22. // Should be set by the tests when needed
  23. _removeExpectation = nil;
  24. return self;
  25. }
  26. - (BOOL)removeItemAtPath:(NSString *)path {
  27. [super removeItemAtPath:path];
  28. self.removeCount += 1;
  29. // If we set up the expectation, and we went over the expected count or removes, fulfill the
  30. // expectation
  31. if (self.removeExpectation && self.removeCount >= self.expectedRemoveCount) {
  32. [self.removeExpectation fulfill];
  33. }
  34. return YES;
  35. }
  36. @end