diff --git a/explorer/tests/test_create_sqlite.py b/explorer/tests/test_create_sqlite.py index ed2c53d1..c7849a01 100644 --- a/explorer/tests/test_create_sqlite.py +++ b/explorer/tests/test_create_sqlite.py @@ -30,16 +30,16 @@ class TestCreateSqlite(TestCase): def test_parse_to_sqlite(self): file = SimpleUploadedFile("name.csv", b"name, title\nchris,cto", content_type="text/csv") - sqlite_bytes, name = parse_to_sqlite(file) + sqlite_bytes, name = parse_to_sqlite(file, user_id=1) rows = write_sqlite_and_get_row(sqlite_bytes, "name") - self.assertEqual(name, "name.db") self.assertEqual(rows[0], ("chris", "cto")) + self.assertEqual(name, "name_1.db") def test_parse_to_sqlite_with_no_parser(self): file = SimpleUploadedFile("name.db", SQLITE_BYTES, content_type="application/x-sqlite3") - sqlite_bytes, name = parse_to_sqlite(file) + sqlite_bytes, name = parse_to_sqlite(file, user_id=1) rows = write_sqlite_and_get_row(sqlite_bytes, "data") self.assertEqual(rows[0], ("chris", "cto")) - self.assertEqual(name, "name.db") + self.assertEqual(name, "name_1.db")