def test_dig_id_validator_invalid(self, input, mi):
mi.side_effect = Exception("")
with self.assertRaises(Exception):
v.dig_id_validator(input)
def test_dig_id_validator(self, input):
assert_equal(input, v.dig_id_validator(input))
def test_dig_id_validator(self, input):
assert_equal(input, v.dig_id_validator(input))
def test_dig_id_validator_invalid(self, input, mi):
mi.side_effect = Exception("")
with self.assertRaises(Exception):
v.dig_id_validator(input)
class TestImport(unittest.TestCase):
@parameterized.expand([
("s", "s"),
("", ""),
(1, ""),
(dt.datetime.now(), ""),
])
def test_strfy(self, data, expected):
result = du.strfy(data)
assert_equal(result, expected)
def test_concat(self):
sheet = {
"A39": MockCell("foo"),
"B39": MockCell("bar"),
"C39": MockCell("baz"),
}
result = du.concat(["A39", "B39", "C39"])(sheet)
assert_equal(result, "foobarbaz")
@parameterized.expand([
("No", "No", " ", ""),
("No", "No", " additional", "additional"),
("No", "Yes", "", "Data Owner approval required"),
("No", "Yes", "additional notes", "Data Owner approval required, additional notes"),
("Yes", "No", " ", "Supervisor approval required"),
("Yes", "Yes", " ", "Supervisor approval required, Data Owner approval required"),
("Yes", "Yes", "additional notes", "Supervisor approval required, Data Owner approval required, additional notes"),
])
def test_access_restrictions(self, sup, owner, addl, expected):
sheet = {
"B16": MockCell(sup),
"D16": MockCell(owner),
"B17": MockCell(addl),
}
result = du.access_restrictions("B16", "D16", "B17")(sheet)
assert_equal(result, expected)
@parameterized.expand([
(dt.datetime(1989, 03, 11), "1989-03-11"),
(dt.date(1989, 03, 11), "1989-03-11"),
("1989-03-11", "1989-03-11"),
("n/a", ""),
])
def test_date(self, cell_val, expected):
sheet = {"A1": MockCell(cell_val)}
result = du.date("A1")(sheet)
assert_equal(result, expected)
@parameterized.expand([
("Yes", "yes"),
("yes", "yes"),
("", ""),
(None, ""),
])
def test_lower(self, cell_val, expected):
sheet = {"A1": MockCell(cell_val)}
result = du.lower("A1")(sheet)
assert_equal(result, expected)
@parameterized.expand([
("", ""),
("21", "21"),
("2.1", "2.1"),
("<1mb", "1"),
("foo", ""),
("12-13mb", "1213"),
("12", "12"),
])
def test_transfer_initial_size(self, cell_val, expected):
sheet = {"B47": MockCell(cell_val)}
result = du.transfer_initial_size(sheet)
assert_equal(result, expected)
@parameterized.expand([
("A39", "foo"),
(lambda x: x["A39"].value.upper(), "FOO"),
])
def test_get_field(self, cell_or_func, expected):
fields = {"test": cell_or_func}
sheet = {"A39": MockCell("foo")}
result = du.get_field(sheet, "test", fields)
assert_equal(result, expected)
@parameterized.expand([
("A39", ({"test": "foo"}, [])),
(lambda ws: v.dig_id_validator(du.strfy(ws["A39"].value)), ({}, ["test: Must be in the format DI#####"])),
])
def test_make_rec_from_sheet(self, cell_or_func, expected):
fields = {"test": cell_or_func}
sheet = {"A39": MockCell("foo")}
result = du.make_rec_from_sheet(sheet, fields)
assert_equal(result, expected)
