def test_get_datasource(self): self.login(username="******") admin_user = self.get_user("admin") tbl = self.get_table(name="birth_names") datasource_post = get_datasource_post() datasource_post["id"] = tbl.id datasource_post["owners"] = [admin_user.id] data = dict(data=json.dumps(datasource_post)) self.get_json_resp("/datasource/save/", data) url = f"/datasource/get/{tbl.type}/{tbl.id}/" resp = self.get_json_resp(url) self.assertEqual(resp.get("type"), "table") col_names = {o.get("column_name") for o in resp["columns"]} self.assertEqual( col_names, { "num_boys", "num", "gender", "name", "ds", "state", "num_girls", "num_california", }, )
def test_save_duplicate_key(self): self.login(username="******") admin_user = self.get_user("admin") tbl_id = self.get_table(name="birth_names").id datasource_post = get_datasource_post() datasource_post["id"] = tbl_id datasource_post["owners"] = [admin_user.id] datasource_post["columns"].extend( [ { "column_name": "<new column>", "filterable": True, "groupby": True, "expression": "<enter SQL expression here>", "id": "somerandomid", }, { "column_name": "<new column>", "filterable": True, "groupby": True, "expression": "<enter SQL expression here>", "id": "somerandomid2", }, ] ) data = dict(data=json.dumps(datasource_post)) resp = self.get_json_resp("/datasource/save/", data, raise_on_error=False) self.assertIn("Duplicate column name(s): <new column>", resp["error"])
def test_save(self): self.login(username="******") tbl_id = self.get_table(name="birth_names").id datasource_post = get_datasource_post() datasource_post["id"] = tbl_id data = dict(data=json.dumps(datasource_post)) resp = self.get_json_resp("/datasource/save/", data) for k in datasource_post: if k == "columns": self.compare_lists(datasource_post[k], resp[k], "column_name") elif k == "metrics": self.compare_lists(datasource_post[k], resp[k], "metric_name") elif k == "database": self.assertEqual(resp[k]["id"], datasource_post[k]["id"]) else: self.assertEqual(resp[k], datasource_post[k])
def test_change_database(self): self.login(username="******") tbl = self.get_table(name="birth_names") tbl_id = tbl.id db_id = tbl.database_id datasource_post = get_datasource_post() datasource_post["id"] = tbl_id new_db = self.create_fake_db() datasource_post["database"]["id"] = new_db.id resp = self.save_datasource_from_dict(datasource_post) self.assertEqual(resp["database"]["id"], new_db.id) datasource_post["database"]["id"] = db_id resp = self.save_datasource_from_dict(datasource_post) self.assertEqual(resp["database"]["id"], db_id) self.delete_fake_db()