update integration tests with tags

Author: snopoke

integration_tests/__init__.py

@@ -32,5 +32,8 @@ def _load_config():
         os.environ.get("TASKBADGER_PROJECT", ""),
         os.environ.get("TASKBADGER_API_KEY", ""),
         systems=[CelerySystemIntegration()],
+        tags={
+            "env": "integration",
+        },
     )
     print(f"\nIntegration tests configuration:\n    {badger.mug.Badger.current.settings}\n")

integration_tests/test_basics.py

@@ -9,8 +9,10 @@ def test_basics():
     task = badger.create_task("test basics", data=data)
     task.success(100)
     assert task.status == StatusEnum.SUCCESS
+    assert task.tags == {"env": "integration"}
 
     fresh = badger.get_task(task.id)
     assert fresh.status == StatusEnum.SUCCESS
     assert fresh.value == 100
     assert fresh.data == data
+    assert fresh.tags == {"env": "integration"}

integration_tests/test_celery.py

@@ -34,6 +34,7 @@ def test_celery(celery_session_app, celery_session_worker):
     assert tb_task.status == StatusEnum.SUCCESS
     assert tb_task.value == 100
     assert tb_task.data == {"result": a + b}
+    assert tb_task.tags == {"env": "integration"}
 
 
 def test_celery_auto_track(celery_session_app, celery_session_worker):