FilesExpand file tree

 master

/

tableadmin_test.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

66 lines (53 loc) · 2.54 KB

 master

/

tableadmin_test.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

66 lines (53 loc) · 2.54 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

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

#!/usr/bin/env python

# Copyright 2018, Google LLC

# 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 random

from tableadmin import create_table

from tableadmin import delete_table

from tableadmin import run_table_operations

PROJECT = os.environ['GOOGLE_CLOUD_PROJECT']

BIGTABLE_INSTANCE = os.environ['BIGTABLE_INSTANCE']

TABLE_NAME_FORMAT = 'tableadmin-test-{}'

TABLE_NAME_RANGE = 10000

def test_run_table_operations(capsys):

table_name = TABLE_NAME_FORMAT.format(

random.randrange(TABLE_NAME_RANGE))

run_table_operations(PROJECT, BIGTABLE_INSTANCE, table_name)

out, _ = capsys.readouterr()

assert 'Creating the ' + table_name + ' table.' in out

assert 'Listing tables in current project.' in out

assert 'Creating column family cf1 with with MaxAge GC Rule' in out

assert 'Created column family cf1 with MaxAge GC Rule.' in out

assert 'Created column family cf2 with Max Versions GC Rule.' in out

assert 'Created column family cf3 with Union GC rule' in out

assert 'Created column family cf4 with Intersection GC rule.' in out

assert 'Created column family cf5 with a Nested GC rule.' in out

assert 'Printing Column Family and GC Rule for all column families.' in out

assert 'Updating column family cf1 GC rule...' in out

assert 'Updated column family cf1 GC rule' in out

assert 'Print column family cf1 GC rule after update...' in out

assert 'Column Family: cf1' in out

assert 'max_num_versions: 1' in out

assert 'Delete a column family cf2...' in out

assert 'Column family cf2 deleted successfully.' in out

delete_table(PROJECT, BIGTABLE_INSTANCE, table_name)

def test_delete_table(capsys):

table_name = TABLE_NAME_FORMAT.format(

random.randrange(TABLE_NAME_RANGE))

create_table(PROJECT, BIGTABLE_INSTANCE, table_name)

delete_table(PROJECT, BIGTABLE_INSTANCE, table_name)

out, _ = capsys.readouterr()

assert 'Table ' + table_name + ' exists.' in out

assert 'Deleting ' + table_name + ' table.' in out

assert 'Deleted ' + table_name + ' table.' in out