fix tests that need to know if we're on S2 or not

Author: markus-wa

pkg/demoinfocs/common/common_test.go

@@ -25,7 +25,12 @@ func TestBombPosition(t *testing.T) {
 	plEntity := entityWithID(1)
 	plEntity.On("Position").Return(playerPos)
 
-	bomb.Carrier = &Player{Entity: plEntity}
+	bomb.Carrier = &Player{
+		Entity: plEntity,
+		demoInfoProvider: demoInfoProviderMock{
+			isSource2: false,
+		},
+	}
 	assert.Equal(t, playerPos, bomb.Position(), "Bomb position should be Player.Position")
 }
 

pkg/demoinfocs/game_events_test.go

@@ -17,6 +17,9 @@ import (
 // See #90
 func TestRoundEnd_LoserState_Score(t *testing.T) {
 	p := NewParser(rand.Reader).(*parser)
+	p.header = &common.DemoHeader{
+		Filestamp: "HL2DEMO",
+	}
 
 	p.gameState.tState.Entity = stfake.NewEntityWithProperty("m_scoreTotal", st.PropertyValue{IntVal: 1})
 	p.gameState.ctState.Entity = stfake.NewEntityWithProperty("m_scoreTotal", st.PropertyValue{IntVal: 2})