FilesExpand file tree

 main

/

test_types.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

161 lines (117 loc) · 5.04 KB

 main

/

test_types.py

Top

File metadata and controls

• Code
• Blame

161 lines (117 loc) · 5.04 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

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

import pytest

from gitlab import exceptions, types

class TestRequiredOptional:

def test_requiredoptional_empty(self):

b = types.RequiredOptional()

assert not b.required

assert not b.optional

assert not b.exclusive

def test_requiredoptional_values_no_keywords(self):

b = types.RequiredOptional(

("required1", "required2"),

("optional1", "optional2"),

("exclusive1", "exclusive2"),

)

assert b.required == ("required1", "required2")

assert b.optional == ("optional1", "optional2")

assert b.exclusive == ("exclusive1", "exclusive2")

def test_requiredoptional_values_keywords(self):

b = types.RequiredOptional(

exclusive=("exclusive1", "exclusive2"),

optional=("optional1", "optional2"),

required=("required1", "required2"),

)

assert b.required == ("required1", "required2")

assert b.optional == ("optional1", "optional2")

assert b.exclusive == ("exclusive1", "exclusive2")

def test_validate_attrs_required(self):

data = {"required1": 1, "optional2": 2}

rq = types.RequiredOptional(required=("required1",))

rq.validate_attrs(data=data)

data = {"optional1": 1, "optional2": 2}

with pytest.raises(AttributeError, match="Missing attributes: required1"):

rq.validate_attrs(data=data)

def test_validate_attrs_exclusive(self):

data = {"exclusive1": 1, "optional1": 1}

rq = types.RequiredOptional(exclusive=("exclusive1", "exclusive2"))

rq.validate_attrs(data=data)

data = {"exclusive1": 1, "exclusive2": 2, "optional1": 1}

with pytest.raises(

AttributeError,

match="Provide only one of these attributes: exclusive1, exclusive2",

):

rq.validate_attrs(data=data)

def test_gitlab_attribute_get():

o = types.GitlabAttribute("whatever")

assert o.get() == "whatever"

o.set_from_cli("whatever2")

assert o.get() == "whatever2"

assert o.get_for_api(key="spam") == ("spam", "whatever2")

o = types.GitlabAttribute()

assert o._value is None

def test_array_attribute_input():

o = types.ArrayAttribute()

o.set_from_cli("foo,bar,baz")

assert o.get() == ["foo", "bar", "baz"]

o.set_from_cli("foo")

assert o.get() == ["foo"]

def test_array_attribute_empty_input():

o = types.ArrayAttribute()

o.set_from_cli("")

assert o.get() == []

o.set_from_cli(" ")

assert o.get() == []

def test_array_attribute_get_for_api_from_cli():

o = types.ArrayAttribute()

o.set_from_cli("foo,bar,baz")

assert o.get_for_api(key="spam") == ("spam[]", ["foo", "bar", "baz"])

def test_array_attribute_get_for_api_from_list():

o = types.ArrayAttribute(["foo", "bar", "baz"])

assert o.get_for_api(key="spam") == ("spam[]", ["foo", "bar", "baz"])

def test_array_attribute_get_for_api_from_int_list():

o = types.ArrayAttribute([1, 9, 7])

assert o.get_for_api(key="spam") == ("spam[]", [1, 9, 7])

def test_array_attribute_does_not_split_string():

o = types.ArrayAttribute("foo")

assert o.get_for_api(key="spam") == ("spam[]", "foo")

# CommaSeparatedListAttribute tests

def test_csv_string_attribute_get_for_api_from_cli():

o = types.CommaSeparatedListAttribute()

o.set_from_cli("foo,bar,baz")

assert o.get_for_api(key="spam") == ("spam", "foo,bar,baz")

def test_csv_string_attribute_get_for_api_from_list():

o = types.CommaSeparatedListAttribute(["foo", "bar", "baz"])

assert o.get_for_api(key="spam") == ("spam", "foo,bar,baz")

def test_csv_string_attribute_get_for_api_from_int_list():

o = types.CommaSeparatedListAttribute([1, 9, 7])

assert o.get_for_api(key="spam") == ("spam", "1,9,7")

# LowercaseStringAttribute tests

def test_lowercase_string_attribute_get_for_api():

o = types.LowercaseStringAttribute("FOO")

assert o.get_for_api(key="spam") == ("spam", "foo")

# JsonAttribute tests

def test_json_attribute() -> None:

attr = types.JsonAttribute()

attr.set_from_cli('{"key": "value"}')

assert attr.get() == {"key": "value"}

with pytest.raises(exceptions.GitlabParsingError):

attr.set_from_cli(" ")

# CommaSeparatedStringAttribute tests

def test_comma_separated_string_attribute() -> None:

# Test with list of integers

attr = types.CommaSeparatedStringAttribute([1, 2, 3])

assert attr.get_for_api(key="ids") == ("ids", "1,2,3")

# Test with list of strings

attr = types.CommaSeparatedStringAttribute(["a", "b"])

assert attr.get_for_api(key="names") == ("names", "a,b")

# Test with string value (should be preserved)

attr = types.CommaSeparatedStringAttribute("1,2,3")

assert attr.get_for_api(key="ids") == ("ids", "1,2,3")

# Test CLI setting

attr = types.CommaSeparatedStringAttribute()

attr.set_from_cli("1, 2, 3")

assert attr.get() == ["1", "2", "3"]

attr.set_from_cli("")

assert attr.get() == []

# Verify transform_in_body is True

assert types.CommaSeparatedStringAttribute.transform_in_body is True