FilesExpand file tree

 tmp-generated-samples

/

redact_test.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

45 lines (35 loc) · 1.24 KB

 tmp-generated-samples

/

redact_test.py

Copy path

Top

File metadata and controls

• Code
• Blame

45 lines (35 loc) · 1.24 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

# Copyright 2017 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 os

import shutil

import tempfile

import pytest

import redact

GCLOUD_PROJECT = os.getenv('GCLOUD_PROJECT')

RESOURCE_DIRECTORY = os.path.join(os.path.dirname(__file__), 'resources')

@pytest.fixture(scope='module')

def tempdir():

tempdir = tempfile.mkdtemp()

yield tempdir

shutil.rmtree(tempdir)

def test_redact_image_file(tempdir, capsys):

test_filepath = os.path.join(RESOURCE_DIRECTORY, 'test.png')

output_filepath = os.path.join(tempdir, 'redacted.png')

redact.redact_image(

GCLOUD_PROJECT,

test_filepath,

output_filepath,

['FIRST_NAME', 'EMAIL_ADDRESS'])

out, _ = capsys.readouterr()

assert output_filepath in out