From 6f7aa0db5f18718c3871e2975d8debb3d3b485d3 Mon Sep 17 00:00:00 2001
From: chrchr-github <christian.riesch@live.com>
Date: Wed, 9 Aug 2023 00:41:06 +0200
Subject: [PATCH] Set ValueType for auto with ternary

---
 lib/symboldatabase.cpp      |  7 +++++++
 test/testsymboldatabase.cpp | 23 +++++++++++++++++++++--
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/lib/symboldatabase.cpp b/lib/symboldatabase.cpp
index 4712e646ec8..5439adee1e0 100644
--- a/lib/symboldatabase.cpp
+++ b/lib/symboldatabase.cpp
@@ -6552,6 +6552,13 @@ void SymbolDatabase::setValueType(Token* tok, const ValueType& valuetype, Source
                     setValueType(parent, *vt1);
                 return;
             }
+
+            if (vt1->pointer == 0U && vt2 && vt2->pointer == 0U) {
+                if (vt1->isTypeEqual(vt2)) {
+                    setValueType(parent, *vt1);
+                    return;
+                }
+            }
         }
 
         if (vt1->pointer != 0U) {
diff --git a/test/testsymboldatabase.cpp b/test/testsymboldatabase.cpp
index 3ca63283168..498a8f33fa1 100644
--- a/test/testsymboldatabase.cpp
+++ b/test/testsymboldatabase.cpp
@@ -511,6 +511,7 @@ class TestSymbolDatabase : public TestFixture {
         TEST_CASE(auto18);
         TEST_CASE(auto19);
         TEST_CASE(auto20);
+        TEST_CASE(auto21);
 
         TEST_CASE(unionWithConstructor);
 
@@ -5716,8 +5717,7 @@ class TestSymbolDatabase : public TestFixture {
         ASSERT_EQUALS(E1->value, 1);
         const Token* const a = Token::findsimplematch(tokenizer.tokens(), "auto");
         ASSERT(a && a->valueType());
-        TODO_ASSERT(E1->scope == a->valueType()->typeScope);
-        ASSERT_EQUALS(a->valueType()->type, ValueType::INT);
+        ASSERT(E1->scope == a->valueType()->typeScope);
     }
 
     void sizeOfType() {
@@ -9454,6 +9454,25 @@ class TestSymbolDatabase : public TestFixture {
         ASSERT_EQUALS(g->function()->tokenDef->linenr(), 4);
     }
 
+    void auto21() {
+        GET_SYMBOL_DB("int f(bool b) {\n"
+                      "    std::vector<int> v1(1), v2(2);\n"
+                      "    auto& v = b ? v1 : v2;\n"
+                      "    v.push_back(1);\n"
+                      "    int& r = v[0];"
+                      "    return v.size();\n"
+                      "}\n");
+        ASSERT_EQUALS("", errout.str());
+        const Token* a = Token::findsimplematch(tokenizer.tokens(), "auto");
+        ASSERT(a && a->valueType());
+        ASSERT_EQUALS(a->valueType()->type, ValueType::CONTAINER);
+        const Token* v = Token::findsimplematch(a, "v . size");
+        ASSERT(v && v->valueType());
+        ASSERT_EQUALS(v->valueType()->type, ValueType::CONTAINER);
+        TODO_ASSERT(v->valueType()->reference == Reference::LValue);
+        ASSERT(v->variable() && v->variable()->isReference());
+    }
+
     void unionWithConstructor() {
         GET_SYMBOL_DB("union Fred {\n"
                       "    Fred(int x) : i(x) { }\n"